Skip to content

Commit 6afb592

Browse files
author
Huaishun Hu
committed
musa: ggml_cuda_get_physical_warp_size should return 128 for musa
1 parent 86d33ac commit 6afb592

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

ggml/src/ggml-cuda/common.cuh

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -183,6 +183,8 @@ static constexpr bool new_mma_available(const int cc) {
183183
static constexpr __device__ int ggml_cuda_get_physical_warp_size() {
184184
#if defined(GGML_USE_HIP) && defined(__HIP_PLATFORM_AMD__)
185185
return __AMDGCN_WAVEFRONT_SIZE;
186+
#elif defined(GGML_USE_MUSA)
187+
return 128;
186188
#else
187189
return 32;
188190
#endif // defined(GGML_USE_HIP) && defined(__HIP_PLATFORM_AMD__)

0 commit comments

Comments
 (0)