fix gemm_streamk example on mi300 (#875)

bf1912ed · Illia Silin · GitHub · 9e86ebd6 · bf1912ed
Unverified Commit bf1912ed authored Aug 30, 2023 by Illia Silin Committed by GitHub Aug 30, 2023
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_streamk.hpp ...ensor_operation/gpu/grid/gridwise_gemm_xdlops_streamk.hpp +2 -1

No files found.
--- a/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_streamk.hpp
+++ b/include/ck/tensor_operation/gpu/grid/gridwise_gemm_xdlops_streamk.hpp
@@ -37,7 +37,8 @@ __global__ void
                                   index_t StrideC,
                                   typename GridwiseGemm::Block2CTileMap block_mapping)
 {
-#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx908__) || defined(__gfx90a__))
+#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx908__) || defined(__gfx90a__) || \
+    defined(__gfx940__) || defined(__gfx941__) || defined(__gfx942__))
    constexpr index_t shared_size = GridwiseGemm::GetSharedMemoryNumberOfByte();

    __shared__ uint8_t p_shared[shared_size];