Skip to content

Commit 2e49a97

Browse files
committed
Rename to MatMulNBitsWideTileProgram
1 parent f3cc4d4 commit 2e49a97

File tree

2 files changed

+5
-5
lines changed

2 files changed

+5
-5
lines changed

onnxruntime/contrib_ops/webgpu/quantization/matmul_nbits.cc

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff 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,

onnxruntime/contrib_ops/webgpu/quantization/matmul_nbits.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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});

0 commit comments

Comments
 (0)