ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-07-01 12:17:00 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add _VectorSize to Traits

Browse Source
This commit is contained in:
Mateusz Ozga
2025-09-15 14:52:27 +00:00
parent 0b9a638f26
commit 954232d2a0
1 changed files with 3 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
include/ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp
View File

@@ -56,6 +56,9 @@ struct TileGemmUniversalTraits
using BLayout = BLayout_;
using CLayout = CLayout_;
// TODO this can't be hardcoded here! Should be in policy!
static constexpr int _VectorSize = 16;
static constexpr bool TransposeC = TransposeC_;
static constexpr bool UseStructuredSparsity = UseStructuredSparsity_;
static constexpr bool UsePersistentKernel = UsePersistentKernel_;