Skip to content

Commit

Permalink
test removing explicit broadcast
Browse files Browse the repository at this point in the history
  • Loading branch information
qihqi committed Nov 29, 2023
1 parent b9475d9 commit 6eef388
Showing 1 changed file with 1 addition and 0 deletions.
1 change: 1 addition & 0 deletions torch_xla/csrc/helpers.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -614,6 +614,7 @@ xla::Shape XlaHelpers::GetPromotedDynamicShape(const xla::Shape& shape1,

std::pair<xla::XlaOp, xla::XlaOp> XlaHelpers::PromoteShapes(xla::XlaOp op1,
xla::XlaOp op2) {
return std::make_pair(op1, op2);
const xla::Shape& shape1 = ShapeHelper::ShapeOfXlaOp(op1);
const xla::Shape& shape2 = ShapeHelper::ShapeOfXlaOp(op2);
if (xla::ShapeUtil::Compatible(shape1, shape2)) {
Expand Down

0 comments on commit 6eef388

Please sign in to comment.