replace sync call in sub

jingguo-st · jingguo-st · commit 44d5c852f7ea · 2024-07-19T11:10:53.000+08:00
diff --git a/dipu/scripts/autogen_diopi_wrapper/diopi_functions.yaml b/dipu/scripts/autogen_diopi_wrapper/diopi_functions.yaml
@@ -90,7 +90,8 @@
 
     at::Tensor selfTmp = self;
     if (is_scalar_on_cpu(selfTmp)) {
-      selfTmp = selfTmp.to(other.device());
+      selfTmp = nodispatch::empty({1}, self.options().device(other.device()));
+      dipu_fill__scalar(selfTmp, self.item());
     }
 
     at::native::sub_check(selfTmp, other);

Original file line number	Diff line number	Diff line change
`@@ -90,7 +90,8 @@`
`90`	`90`
`91`	`91`	`at::Tensor selfTmp = self;`
`92`	`92`	`if (is_scalar_on_cpu(selfTmp)) {`
`93`		`- selfTmp = selfTmp.to(other.device());`
	`93`	`+ selfTmp = nodispatch::empty({1}, self.options().device(other.device()));`
	`94`	`+ dipu_fill__scalar(selfTmp, self.item());`
`94`	`95`	`}`
`95`	`96`
`96`	`97`	`at::native::sub_check(selfTmp, other);`