Add static_assert

798a9867 · Paul · ce364aae · 798a9867
Commit 798a9867 authored Oct 20, 2022 by Paul
Show whitespace changes
Inline Side-by-side

Showing with 6 additions and 0 deletions

src/targets/gpu/kernels/include/migraphx/kernels/ck_gemm.hpp src/targets/gpu/kernels/include/migraphx/kernels/ck_gemm.hpp +6 -0

No files found.
--- a/src/targets/gpu/kernels/include/migraphx/kernels/ck_gemm.hpp
+++ b/src/targets/gpu/kernels/include/migraphx/kernels/ck_gemm.hpp
@@ -59,6 +59,12 @@ __device__ void ck_gemm(E e, A a, B b, Ds... ds)
    constexpr auto e_grid_desc_mblock_mperblock_nblock_nperblock =
        GridwiseGemm::MakeEGridDescriptor_MBlock_MPerBlock_NBlock_NPerBlock(e_grid_desc_m_n);

+    static_assert(GridwiseGemm::CheckValidity(a_grid_desc_m_k,
+                                    b_grid_desc_n_k,
+                                    ds_grid_desc_m_n,
+                                    e_grid_desc_m_n,
+                                    block_2_etile_map));
+
    __shared__ char p_shared_block[GridwiseGemm::GetSharedMemoryNumberOfByte()];

    constexpr const bool HasMainKBlockLoop =