expose number of components

microsoft · Sep 13, 2024 · b8c369d · b8c369d
1 parent eae4c3f
commit b8c369d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/onnxruntime/core/providers/webgpu/shader_variable.h b/onnxruntime/core/providers/webgpu/shader_variable.h
@@ -63,6 +63,9 @@ class ShaderVariable {
   // get the name of the variable.
   inline std::string_view Name() const { return name_; }
 
+  // get the number of components of the variable.
+  inline int NumComponents() const { return num_components_; }
+
   // create a WGSL expression ({varname}_indices_t) for getting indices from offset.
   // \param offset: a WGSL expression (u32) representing the offset.
   inline std::string OffsetToIndices(std::string_view offset_expr) const;