From 954232d2a001e44b1a473f593acd072adcf430d2 Mon Sep 17 00:00:00 2001 From: Mateusz Ozga Date: Mon, 15 Sep 2025 14:52:27 +0000 Subject: [PATCH] Add _VectorSize to Traits --- include/ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp | 3 +++ 1 file changed, 3 insertions(+) diff --git a/include/ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp b/include/ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp index 64900c9a97..f6d5781d67 100644 --- a/include/ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp +++ b/include/ck_tile/ops/gemm/pipeline/tile_gemm_traits.hpp @@ -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_;