add more navi devices

ROCm · Sep 13, 2024 · 5f153c5 · 5f153c5
1 parent fe2aea8
commit 5f153c5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/perf-kernels/flash-attention.py b/python/perf-kernels/flash-attention.py
@@ -326,7 +326,7 @@ def get_NAVI_autotune_configs():
 
 def is_navi():
     target = triton.runtime.driver.active.get_current_target()
-    return target.backend == 'hip' and target.arch in ["gfx1030"]
+    return target.backend == 'hip' and target.arch in ["gfx1030", "gfx1100", "gfx1101", "gfx1102", "gfx1200", "gfx1201"]
 
 
 def get_autotune_configs():