include/ck/wrapper/traits/blockwise_gemm_xdl_traits.hpp Source File

include/ck/wrapper/traits/blockwise_gemm_xdl_traits.hpp Source File#

Composable Kernel: include/ck/wrapper/traits/blockwise_gemm_xdl_traits.hpp Source File
 // SPDX-License-Identifier: MIT
 // Copyright (c) 2024, Advanced Micro Devices, Inc. All rights reserved.
  
 #pragma once
  
 #include "ck/ck.hpp"
  
 // Disable from doxygen docs generation
 namespace ck {
 namespace wrapper {
  
 template <typename MPerXDLValue,
           typename NPerXDLValue,
           typename MXdlPerWaveValue,
           typename NXdlPerWaveValue,
           typename K1Value>
 struct BlockwisGemmXdlTraits
 {
     static constexpr auto MPerXDL     = MPerXDLValue{};
     static constexpr auto NPerXDL     = NPerXDLValue{};
     static constexpr auto MXdlPerWave = MXdlPerWaveValue{};
     static constexpr auto NXdlPerWave = NXdlPerWaveValue{};
     static constexpr auto K1          = K1Value{};
 };
  
 // K1 = 4
 struct BlockwisGemmXdlTraits_32x32Xdl_4x2XdlPerWave_4K1
     : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<4>, Number<2>, Number<4>>
 {
 };
 struct BlockwisGemmXdlTraits_32x32Xdl_2x4XdlPerWave_4K1
     : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<4>, Number<4>>
 {
 };
 struct BlockwisGemmXdlTraits_32x32Xdl_2x2XdlPerWave_4K1
     : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<2>, Number<4>>
 {
 };
 // K1 = 8
 struct BlockwisGemmXdlTraits_32x32Xdl_4x2XdlPerWave_8K1
     : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<4>, Number<2>, Number<8>>
 {
 };
 struct BlockwisGemmXdlTraits_32x32Xdl_2x4XdlPerWave_8K1
     : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<4>, Number<8>>
 {
 };
 struct BlockwisGemmXdlTraits_32x32Xdl_2x2XdlPerWave_8K1
     : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<2>, Number<8>>
 {
 };
 // K1 = 16
 struct BlockwisGemmXdlTraits_32x32Xdl_4x2XdlPerWave_16K1
     : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<4>, Number<2>, Number<16>>
 {
 };
 struct BlockwisGemmXdlTraits_32x32Xdl_2x4XdlPerWave_16K1
     : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<4>, Number<16>>
 {
 };
 struct BlockwisGemmXdlTraits_32x32Xdl_2x2XdlPerWave_16K1
     : BlockwisGemmXdlTraits<Number<32>, Number<32>, Number<2>, Number<2>, Number<16>>
 {
 };
  
 } // namespace wrapper
 } // namespace ck