Skip to content

Commit fa7ea97

Browse files
authored
quick dirty hack to compile rocm
1 parent 066b909 commit fa7ea97

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

ggml/src/ggml-cuda/fattn-wmma-f16.cu

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -56,7 +56,7 @@ static __global__ void flash_attn_ext_f16(
5656
const int ne1,
5757
const int ne2,
5858
const int ne3) {
59-
#if defined(FLASH_ATTN_AVAILABLE) && ((__CUDA_ARCH__ == GGML_CUDA_CC_VOLTA || __CUDA_ARCH__ == GGML_CUDA_CC_TURING) || (defined(GGML_HIP_ROCWMMA_FATTN) && defined(FP16_MMA_AVAILABLE)))
59+
#if 0
6060
// Skip unused kernel variants for faster compilation:
6161
if (use_logit_softcap && !(D == 128 || D == 256)) {
6262
NO_DEVICE_CODE;

0 commit comments

Comments
 (0)