From d58d55e5404b1cb7e9c382d0ebfa65cf94d59154 Mon Sep 17 00:00:00 2001 From: chenjun Date: Thu, 26 Dec 2024 07:38:14 +0000 Subject: [PATCH] edit GemmSpec to MNKPadding --- .../gemm_multiply_multiply_xdl_fp8_ab_scale.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git 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 index f4fa6ab2ab..6e103bfedd 100644 --- 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;