add gfx1201 macro in amd_wmma header

0b7addbd · illsilin · 8bee1e81 · 0b7addbd
Commit 0b7addbd authored Apr 16, 2024 by illsilin
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

include/ck/utility/amd_wmma.hpp include/ck/utility/amd_wmma.hpp +1 -1

No files found.
--- a/include/ck/utility/amd_wmma.hpp
+++ b/include/ck/utility/amd_wmma.hpp
@@ -260,7 +260,7 @@ struct intrin_wmma_i32_16x16x16_iu8_w64<16, 16, neg_a, neg_b, clamp>
 // gfx12
 /********************************WAVE32 MODE***********************************************/
-#if defined(__gfx1200__)
+#if defined(__gfx1200__) || defined(__gfx1201__)
 #define __gfx12__
 #endif