fix a test break

feich-ms · feich-ms · commit 62da03be7830 · 2025-04-11T09:27:10.000+08:00
diff --git a/onnxruntime/core/providers/webgpu/math/einsum.cc b/onnxruntime/core/providers/webgpu/math/einsum.cc
@@ -68,8 +68,8 @@ EinsumEquation::EinsumEquation(const std::vector<const Tensor*>& inputs, const s
   // Initialize RHS if not specified.
   if (rhs.empty()) {
     for (const auto& pair : symbol_to_info_) {
-      if (pair.second.count == 1 || pair.first == "...") {
-        rhs += pair.first;
+      if (pair.second.count == 1 || pair.first == "..." || pair.first.starts_with("_")) {
+        rhs += pair.first.starts_with("_") ? "..." : pair.first;
       }
     }
   } else {

Original file line number	Diff line number	Diff line change
`@@ -68,8 +68,8 @@ EinsumEquation::EinsumEquation(const std::vector<const Tensor*>& inputs, const s`
`68`	`68`	`// Initialize RHS if not specified.`
`69`	`69`	`if (rhs.empty()) {`
`70`	`70`	`for (const auto& pair : symbol_to_info_) {`
`71`		`- if (pair.second.count == 1 \|\| pair.first == "...") {`
`72`		`- rhs += pair.first;`
	`71`	`+ if (pair.second.count == 1 \|\| pair.first == "..." \|\| pair.first.starts_with("_")) {`
	`72`	`+ rhs += pair.first.starts_with("_") ? "..." : pair.first;`
`73`	`73`	`}`
`74`	`74`	`}`
`75`	`75`	`} else {`