fix sub check for alpha

jingguo-st · jingguo-st · commit fad4275dd072 · 2024-07-19T16:29:40.000+08:00
diff --git a/dipu/scripts/autogen_diopi_wrapper/diopi_functions.yaml b/dipu/scripts/autogen_diopi_wrapper/diopi_functions.yaml
@@ -87,6 +87,7 @@
     at::native::sub_check(self, other);
 
     if (is_scalar_on_cpu(other)) {
+      at::native::alpha_check(self.scalar_type(), alpha);
       return dipu_sub_scalar(self, other.item(), alpha);
     }
 
@@ -96,6 +97,7 @@
       dipu_fill__scalar(selfTmp, self.item());
     }
 
+    at::native::alpha_check(selfTmp.scalar_type(), alpha);
     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
`@@ -87,6 +87,7 @@`
`87`	`87`	`at::native::sub_check(self, other);`
`88`	`88`
`89`	`89`	`if (is_scalar_on_cpu(other)) {`
	`90`	`+ at::native::alpha_check(self.scalar_type(), alpha);`
`90`	`91`	`return dipu_sub_scalar(self, other.item(), alpha);`
`91`	`92`	`}`
`92`	`93`
`@@ -96,6 +97,7 @@`
`96`	`97`	`dipu_fill__scalar(selfTmp, self.item());`
`97`	`98`	`}`
`98`	`99`
	`100`	`+ at::native::alpha_check(selfTmp.scalar_type(), alpha);`
`99`	`101`	`at::Tensor out = BinaryOpInferrer().infer_out(selfTmp, other);`
`100`	`102`
`101`	`103`	`interface: diopiSub(ctx, out, selfTmp, other, alpha)`