diff --git a/stablehlo/reference/Ops.cpp b/stablehlo/reference/Ops.cpp index 4bc3b080712..ca3f0d81c80 100644 --- a/stablehlo/reference/Ops.cpp +++ b/stablehlo/reference/Ops.cpp @@ -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(