Skip to content

Commit 7cad767

Browse files
committed
mtgpu: add mp_21 support
Signed-off-by: Xiaodong Ye <[email protected]>
1 parent faf67b3 commit 7cad767

File tree

2 files changed

+2
-2
lines changed

2 files changed

+2
-2
lines changed

Makefile

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -611,7 +611,7 @@ ifdef GGML_CUDA
611611

612612
MK_CPPFLAGS += -DGGML_USE_CUDA -I$(CUDA_PATH)/include
613613
MK_LDFLAGS += -lmusa -lmublas -lmusart -lpthread -ldl -lrt -L$(CUDA_PATH)/lib -L/usr/lib64
614-
MK_NVCCFLAGS += -x musa -mtgpu --cuda-gpu-arch=mp_22
614+
MK_NVCCFLAGS += -x musa -mtgpu --cuda-gpu-arch=mp_21 --cuda-gpu-arch=mp_22
615615
else
616616
ifneq ('', '$(wildcard /opt/cuda)')
617617
CUDA_PATH ?= /opt/cuda

ggml/src/CMakeLists.txt

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -364,7 +364,7 @@ if (GGML_CUDA)
364364
if (GGML_MUSA)
365365
set_source_files_properties(${GGML_SOURCES_CUDA} PROPERTIES LANGUAGE CXX)
366366
foreach(SOURCE ${GGML_SOURCES_CUDA})
367-
set_property(SOURCE ${SOURCE} PROPERTY COMPILE_FLAGS "-x musa -mtgpu --cuda-gpu-arch=mp_22")
367+
set_property(SOURCE ${SOURCE} PROPERTY COMPILE_FLAGS "-x musa -mtgpu --cuda-gpu-arch=mp_21 --cuda-gpu-arch=mp_22")
368368
endforeach()
369369
endif()
370370

0 commit comments

Comments
 (0)