fix rdna2 variable name: is_rdna2 -> is_rdna2_and_older

This commit is contained in:
resonantsky
2026-04-17 18:44:24 +02:00
parent 576d566429
commit cefad871b9
+3 -3
View File
@@ -13,11 +13,11 @@ import triton
import triton.language as tl
try:
from .common import is_rdna2
from .common import is_rdna2_and_older
except Exception:
is_rdna2 = False
is_rdna2_and_older = False
if is_rdna2:
if is_rdna2_and_older:
matmul_configs = [
triton.Config({'BLOCK_SIZE_M': BM, 'BLOCK_SIZE_N': BN, "BLOCK_SIZE_K": BK, "GROUP_SIZE_M": GM}, num_warps=w, num_stages=s)
for BM in [64, 128]