fix bug

jingguo-st · jingguo-st · commit f593ec99cd62 · 2024-07-19T12:29:32.000+08:00
diff --git a/dipu/scripts/autogen_diopi_wrapper/diopi_functions.yaml b/dipu/scripts/autogen_diopi_wrapper/diopi_functions.yaml
@@ -84,6 +84,8 @@
 - schema: "sub.Tensor(Tensor self, Tensor other, *, Scalar alpha=1) -> Tensor"
   ins: [selfTmp]
   custom_code_at_the_beginning: |
+    at::native::sub_check(self, other);
+
     if (is_scalar_on_cpu(other)) {
       return dipu_sub_scalar(self, other.item(), alpha);
     }
@@ -94,7 +96,6 @@
       dipu_fill__scalar(selfTmp, self.item());
     }
 
-    at::native::sub_check(selfTmp, other);
     at::Tensor out = BinaryOpInferrer().infer_out(selfTmp, other);
 
   interface: diopiSub(ctx, out, selfTmp, other, alpha)

Original file line number	Diff line number	Diff line change
`@@ -84,6 +84,8 @@`
`84`	`84`	`- schema: "sub.Tensor(Tensor self, Tensor other, *, Scalar alpha=1) -> Tensor"`
`85`	`85`	`ins: [selfTmp]`
`86`	`86`	`custom_code_at_the_beginning: \|`
	`87`	`+ at::native::sub_check(self, other);`
	`88`	`+`
`87`	`89`	`if (is_scalar_on_cpu(other)) {`
`88`	`90`	`return dipu_sub_scalar(self, other.item(), alpha);`
`89`	`91`	`}`
`@@ -94,7 +96,6 @@`
`94`	`96`	`dipu_fill__scalar(selfTmp, self.item());`
`95`	`97`	`}`
`96`	`98`
`97`		`- at::native::sub_check(selfTmp, other);`
`98`	`99`	`at::Tensor out = BinaryOpInferrer().infer_out(selfTmp, other);`
`99`	`100`
`100`	`101`	`interface: diopiSub(ctx, out, selfTmp, other, alpha)`