We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent db3e7f1 commit 914789eCopy full SHA for 914789e
1 file changed
backends/metax_gpu/CMakeLists.txt
@@ -809,6 +809,11 @@ file(
809
set(CUSTOM_DEVICE_SRCS ${CUDA_SRCS} ${CC_SRCS} ${ERNIE_CORE_SRCS})
810
811
set_source_files_properties(${CUSTOM_DEVICE_SRCS} PROPERTIES LANGUAGE CUDA)
812
+# 为weight_only_linear_kernel.cu添加Metax GPU架构支持
813
+set_source_files_properties(
814
+ ${CMAKE_CURRENT_SOURCE_DIR}/kernels/custom_kernel/weight_only_linear_kernel.cu
815
+ PROPERTIES COMPILE_OPTIONS "--offload-arch=xcore1000;--offload-arch=xcore1500")
816
+
817
set(MACA_PATH $ENV{MACA_PATH})
818
set(CMAKE_CUCC_COMPILER "cucc")
819
set(CMAKE_CUCC_FLAGS "-I ${MACA_PATH}/tools/cu-bridge/include/")
0 commit comments