From 8b13e0771bdd945f4f2412ca9b791817e75f0020 Mon Sep 17 00:00:00 2001 From: taoboyang Date: Wed, 13 Dec 2023 17:25:07 +0800 Subject: [PATCH] Hot fix for gpu build wheel v2 --- bmf/hml/src/CMakeLists.txt | 4 ++++ 1 file changed, 4 insertions(+) diff --git a/bmf/hml/src/CMakeLists.txt b/bmf/hml/src/CMakeLists.txt index ed02d3ce..81bd053f 100644 --- 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()