Skip to content

Commit a8c023d

Browse files
committed
quick test rocm
1 parent 833a778 commit a8c023d

File tree

1 file changed

+5
-5
lines changed

1 file changed

+5
-5
lines changed

ggml/src/ggml-cuda/fattn.cu

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -216,11 +216,11 @@ static best_fattn_kernel ggml_cuda_get_best_fattn_kernel(const int device, const
216216

217217
const int cc = ggml_cuda_info().devices[device].cc;
218218

219-
#if defined(GGML_HIP_ROCWMMA_FATTN)
220-
if (GGML_CUDA_CC_IS_AMD(cc) && ggml_cuda_should_use_wmma_fattn(cc)) { //kcpp: fix for rocwmma
221-
return BEST_FATTN_KERNEL_WMMA_F16;
222-
}
223-
#endif // defined(GGML_HIP_ROCWMMA_FATTN)
219+
// #if defined(GGML_HIP_ROCWMMA_FATTN)
220+
// if (GGML_CUDA_CC_IS_AMD(cc) && ggml_cuda_should_use_wmma_fattn(cc)) { //kcpp: fix for rocwmma
221+
// return BEST_FATTN_KERNEL_WMMA_F16;
222+
// }
223+
// #endif // defined(GGML_HIP_ROCWMMA_FATTN)
224224

225225
switch (K->ne[0]) {
226226
case 40:

0 commit comments

Comments
 (0)