change variable

Author: Xiaofei Han

onnxruntime/core/providers/webgpu/math/gemm.cc

@@ -69,7 +69,7 @@ Status GemmProgram::GenerateShaderCode(ShaderHelper& shader) const {
   }
 
   // calculateBeta
-  if(has_C_input_) {
+  if(need_handle_bias_) {
     const ShaderVariableHelper& C = shader.AddInput("C", ShaderUsage::UseUniform | ShaderUsage::UseIndicesTypeAlias);
       shader.MainFunctionBody() << "  value = value + A_value_t(uniforms.beta) * "
                                 << C.GetByOffset(C.BroadcastedIndicesToOffset("vec2(m, n)", output)) << ";\n";
@@ -117,11 +117,11 @@ Status Gemm::ComputeInternal(ComputeContext& context) const {
   int64_t num_tiles_n = (N + TILE_SIZE - 1) / TILE_SIZE;
   int64_t dispatch_size = num_tiles_m * num_tiles_n;
 
-  GemmProgram program{transA_, transB_, alpha_, beta_, C != nullptr};
+  GemmProgram program{transA_, transB_, alpha_, beta_, C&&beta_};
   program.AddInputs({{A, ProgramTensorMetadataDependency::Type},
                     {B, ProgramTensorMetadataDependency::Type}});
 
-  if(C != nullptr) {
+  if(C && beta_) {
     program.AddInput({C, ProgramTensorMetadataDependency::Rank});
   }
 

onnxruntime/core/providers/webgpu/math/gemm.h

@@ -12,13 +12,13 @@ namespace webgpu {
 
 class GemmProgram final : public Program<GemmProgram> {
  public:
-  GemmProgram(bool transA, bool transB, float alpha, float beta, bool has_C_input)
+  GemmProgram(bool transA, bool transB, float alpha, float beta, bool need_handle_bias)
       : Program{"Gemm"},
         transA_{transA},
         transB_{transB},
         alpha_{alpha},
         beta_{beta},
-        has_C_input_{has_C_input} {}
+        need_handle_bias_{need_handle_bias} {}
 
   Status GenerateShaderCode(ShaderHelper& sh) const override;
 
@@ -35,7 +35,7 @@ class GemmProgram final : public Program<GemmProgram> {
   bool transB_;
   float alpha_;
   float beta_;
-  bool has_C_input_;
+  bool need_handle_bias_;
 };
 
 class Gemm final : public WebGpuKernel {