Hot fix for gpu build wheel v2

BabitMF · Dec 13, 2023 · 8b13e07 · 8b13e07
1 parent f5c0f2b
commit 8b13e07
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/bmf/hml/src/CMakeLists.txt b/bmf/hml/src/CMakeLists.txt
@@ -69,6 +69,10 @@ if(HMP_ENABLE_CUDA)
     list(APPEND HMP_SRCS ${CUDA_SRCS} ${CUDA_KERN_SRCS})
     list(APPEND HMP_HDRS ${CUDA_HDRS})
     add_library(hmp ${HMP_SRCS} ${HMP_HDRS})
+    set_target_properties(hmp PROPERTIES
+        CUDA_STANDARD 17
+        CUDA_STANDARD_REQUIRED ON
+    )
 else()
     add_library(hmp ${HMP_SRCS} ${HMP_HDRS})
 endif()