edit GemmSpec to MNKPadding

d58d55e5 · chenjun · 988478d4 · d58d55e5
Commit d58d55e5 authored Dec 26, 2024 by chenjun
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

example/65_gemm_multiply_multiply/gemm_multiply_multiply_xdl_fp8_ab_scale.cpp ...iply_multiply/gemm_multiply_multiply_xdl_fp8_ab_scale.cpp +1 -1

No files found.
--- a/example/65_gemm_multiply_multiply/gemm_multiply_multiply_xdl_fp8_ab_scale.cpp
+++ b/example/65_gemm_multiply_multiply/gemm_multiply_multiply_xdl_fp8_ab_scale.cpp
@@ -53,7 +53,7 @@ using AElementOp   = PassThrough;
 using BElementOp   = PassThrough;
 using CDEElementOp = PassThrough;

-static constexpr auto GemmSpec = ck::tensor_operation::device::GemmSpecialization::Default;
+static constexpr auto GemmSpec = ck::tensor_operation::device::GemmSpecialization::MNKPadding;

 static constexpr ck::index_t Scale_Block_M = 1;
 static constexpr ck::index_t Scale_Block_N = 128;