File tree Expand file tree Collapse file tree 2 files changed +5
-5
lines changed
onnxruntime/contrib_ops/webgpu/quantization Expand file tree Collapse file tree 2 files changed +5
-5
lines changed Original file line number Diff line number Diff line change @@ -533,7 +533,7 @@ Status MatMulNBitsProgram::GenerateShaderCode(ShaderHelper& shader) const {
533533 return Status::OK ();
534534}
535535
536- Status MatMulNBitsBlockWideTileProgram ::GenerateShaderCode (ShaderHelper& shader) const {
536+ Status MatMulNBitsWideTileProgram ::GenerateShaderCode (ShaderHelper& shader) const {
537537 const auto & a = shader.AddInput (" input_a" , ShaderUsage::UseUniform | ShaderUsage::UseIndicesTypeAlias | ShaderUsage::UseValueTypeAlias);
538538 const auto & b = shader.AddInput (" input_b" , ShaderUsage::UseUniform | ShaderUsage::UseIndicesTypeAlias | ShaderUsage::UseValueTypeAlias);
539539 const auto & scales = shader.AddInput (" scales" , ShaderUsage::UseUniform);
@@ -712,7 +712,7 @@ Status MatMulNBits::ComputeInternal(onnxruntime::webgpu::ComputeContext& context
712712 constexpr uint32_t tile_m = workgroup_size / 8 ;
713713 constexpr uint32_t tile_n = workgroup_size;
714714
715- MatMulNBitsBlockWideTileProgram program{tile_m, tile_n};
715+ MatMulNBitsWideTileProgram program{tile_m, tile_n};
716716 program.SetWorkgroupSize (workgroup_size);
717717 program.SetDispatchGroupSize ((N + tile_n - 1 ) / tile_n,
718718 (M + tile_m - 1 ) / tile_m,
Original file line number Diff line number Diff line change @@ -35,10 +35,10 @@ class MatMulNBitsProgram final : public Program<MatMulNBitsProgram> {
3535 bool use_subgroup_;
3636};
3737
38- class MatMulNBitsBlockWideTileProgram final : public Program<MatMulNBitsBlockWideTileProgram > {
38+ class MatMulNBitsWideTileProgram final : public Program<MatMulNBitsWideTileProgram > {
3939 public:
40- MatMulNBitsBlockWideTileProgram (uint32_t tile_m, uint32_t tile_n)
41- : Program{" MatMulNBitsBlockWideTileProgram " }, tile_m_(tile_m), tile_n_(tile_n) {}
40+ MatMulNBitsWideTileProgram (uint32_t tile_m, uint32_t tile_n)
41+ : Program{" MatMulNBitsWideTileProgram " }, tile_m_(tile_m), tile_n_(tile_n) {}
4242
4343 Status GenerateShaderCode (ShaderHelper& sh) const override ;
4444 WEBGPU_PROGRAM_DEFINE_UNIFORM_VARIABLES ({" block_size" , ProgramUniformVariableDataType::Uint32});
You can’t perform that action at this time.
0 commit comments