Skip to content

Commit 13ba06f

Browse files
fix the blockscale 2d case (#3148)
Co-authored-by: Aviral Goel <[email protected]>
1 parent 0be0288 commit 13ba06f

File tree

1 file changed

+5
-6
lines changed

1 file changed

+5
-6
lines changed

example/ck_tile/38_block_scale_gemm/gemm_utils.hpp

Lines changed: 5 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,12 +12,11 @@
1212
#include "ck_tile/ops/gemm_quant.hpp"
1313

1414
#define CK_TILE_SUPPORTED_QUANT_GROUPS(X) \
15-
X(1, 1, 64) /* 1D */ \
16-
X(1, 1, 128) /* 1D */ \
17-
X(1, 8, 128) /* 2D N=8 */ \
18-
X(1, 32, 128) /* 2D N=32 */ \
19-
X(1, 64, 128) /* 2D N=64 */ \
20-
X(1, 128, 128) /* 2D N=128 */
15+
X(1, 1, 64) /* 1D */ \
16+
X(1, 1, 128) /* 1D */ \
17+
X(1, 8, 128) /* 2D N=8 */ \
18+
X(1, 32, 128) /* 2D N=32 */ \
19+
X(1, 64, 128) /* 2D N=64 */
2120

2221
template <typename PrecType, ck_tile::index_t M_Warp_Tile>
2322
constexpr ck_tile::index_t get_k_warp_tile()

0 commit comments

Comments
 (0)