Skip to content

GitLab

"...resnet50_tensorflow.git" did not exist on "3d62e8959f5beb34da990358843d76c50ea3d47a"
Commit 9b69d57f authored by illsilin's avatar illsilin
Browse files

enable batched_gemm_softmax_gemm_perm_wmma for gfx12

parent 3639ceaa
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
include/ck/tensor_operation/gpu/device/impl/device_batched_gemm_softmax_gemm_permute_wmma_cshuffle.hpp
View file @ 9b69d57f
...@@ -322,7 +322,7 @@ __global__ void ...@@ -322,7 +322,7 @@ __global__ void
index_t head_size, index_t head_size,
float alpha) float alpha)
{ {
#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx11__)) #if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx11__) || defined(__gfx12__))
// clang-format off // clang-format off
// *************************************************** // ***************************************************
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment