Skip to content

Commit

Permalink
Use new constant function instead of deleted makeScalar function
Browse files Browse the repository at this point in the history
This seems to be from #1987. Not sure how presubmit failed to catch this.
  • Loading branch information
Michael Levesque-Dion committed Feb 7, 2024
1 parent 6520b58 commit 974db2c
Showing 1 changed file with 2 additions and 3 deletions.
5 changes: 2 additions & 3 deletions stablehlo/reference/Ops.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -1115,9 +1115,8 @@ Tensor evalCollectiveBroadcastOp(
return process->rendezvous(*processGroup, channelId, operand)
.lookup((*processGroup)[0]);

return evalBroadcastInDimOp(
makeScalar(convert(operand.getElementType(), 0.0)), {},
operand.getType());
return evalBroadcastInDimOp(constant(0.0, operand.getElementType()), {},
operand.getType());
}

Tensor evalCollectivePermuteOp(
Expand Down

0 comments on commit 974db2c

Please sign in to comment.