diff --git a/openmp/libomptarget/deviceRTLs/nvptx/CMakeLists.txt b/openmp/libomptarget/deviceRTLs/nvptx/CMakeLists.txt --- a/openmp/libomptarget/deviceRTLs/nvptx/CMakeLists.txt +++ b/openmp/libomptarget/deviceRTLs/nvptx/CMakeLists.txt @@ -172,7 +172,15 @@ list(GET ptx_feature_list ${itr} ptx_num) set(cuda_flags ${sm_flags}) list(APPEND cuda_flags -Xclang -target-feature -Xclang +ptx${ptx_num}) - list(APPEND cuda_flags "-DCUDA_VERSION=${cuda_version}00") + if("${cuda_version}" MATCHES "^([0-9]+)([0-9])$") + set(cuda_version_major ${CMAKE_MATCH_1}) + set(cuda_version_minor ${CMAKE_MATCH_2}) + else() + libomptarget_error_say( + "Unrecognized CUDA version format: ${cuda_version}") + endif() + list(APPEND cuda_flags + "-DCUDA_VERSION=${cuda_version_major}0${cuda_version_minor}0") set(bc_files "") foreach(src ${cuda_src_files})