Commit d58d55e5 authored by chenjun's avatar chenjun
Browse files

edit GemmSpec to MNKPadding

parent 988478d4
...@@ -53,7 +53,7 @@ using AElementOp = PassThrough; ...@@ -53,7 +53,7 @@ using AElementOp = PassThrough;
using BElementOp = PassThrough; using BElementOp = PassThrough;
using CDEElementOp = 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_M = 1;
static constexpr ck::index_t Scale_Block_N = 128; static constexpr ck::index_t Scale_Block_N = 128;
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment