enable gfx12

50530c17 · Jing Zhang · 1bf4f6ae · 50530c17
Commit 50530c17 authored Apr 07, 2024 by Jing Zhang
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

include/ck/tensor_operation/gpu/device/impl/device_multi_query_attention_forward_wmma.hpp ...device/impl/device_multi_query_attention_forward_wmma.hpp +2 -1

No files found.
--- a/include/ck/tensor_operation/gpu/device/impl/device_multi_query_attention_forward_wmma.hpp
+++ b/include/ck/tensor_operation/gpu/device/impl/device_multi_query_attention_forward_wmma.hpp
@@ -60,7 +60,8 @@ __global__ void
                                          bool input_permute,
                                          bool output_permute)
 {
-#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx11__) || defined(__gfx11__))
+#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx11__) || defined(__gfx11__) || \
+    defined(__gfx12__))

    // clang-format off
 // ***************************************************