Merge pull request #25739 from siju-samuel:patch-28
PiperOrigin-RevId: 234067633
This commit is contained in:
commit
4b7ceaee0f
@ -502,7 +502,7 @@ class HloInstruction {
|
||||
// Data is sent/received according to the (source_replica_id,
|
||||
// target_replica_id) pairs in `source_target_pairs`. If a replica id is not a
|
||||
// target_replica_id in any pair, the output on that replica is a tensor
|
||||
// conssits of 0(s) in `shape`.
|
||||
// consists of 0(s) in `shape`.
|
||||
static std::unique_ptr<HloInstruction> CreateCollectivePermute(
|
||||
const Shape& shape, HloInstruction* operand,
|
||||
const std::vector<std::pair<int64, int64>>& source_target_pairs);
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user