Convert scalar to the right type

JackCaoG · JackCaoG · commit 5de4435a5e48 · 2022-08-11T23:47:56.000Z
diff --git a/torch_xla/csrc/xla_lower_util.cpp b/torch_xla/csrc/xla_lower_util.cpp
@@ -1020,6 +1020,7 @@ xla::XlaOp BuildAddcdiv(xla::XlaOp input, xla::XlaOp t1, xla::XlaOp t2,
 
 xla::XlaOp BuildAddcmul(xla::XlaOp input, xla::XlaOp t1, xla::XlaOp t2,
                         xla::XlaOp val) {
+  val = MaybeConvertTo(val, XlaHelpers::ShapeOfXlaOp(t1).element_type());
   return XlaHelpers::PromotedAdd(
       input, XlaHelpers::PromotedMul(XlaHelpers::PromotedMul(t1, t2), val));
 }

Original file line number	Diff line number	Diff line change
`@@ -1020,6 +1020,7 @@ xla::XlaOp BuildAddcdiv(xla::XlaOp input, xla::XlaOp t1, xla::XlaOp t2,`
`1020`	`1020`
`1021`	`1021`	`xla::XlaOp BuildAddcmul(xla::XlaOp input, xla::XlaOp t1, xla::XlaOp t2,`
`1022`	`1022`	`xla::XlaOp val) {`
	`1023`	`+ val = MaybeConvertTo(val, XlaHelpers::ShapeOfXlaOp(t1).element_type());`
`1023`	`1024`	`return XlaHelpers::PromotedAdd(`
`1024`	`1025`	`input, XlaHelpers::PromotedMul(XlaHelpers::PromotedMul(t1, t2), val));`
`1025`	`1026`	`}`