MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / Run

Method Run

tensorflow/compiler/xla/service/buffer_assignment.cc:820–833  ·  view source on GitHub ↗

static */

Source from the content-addressed store, hash-verified

818
819/* static */
820StatusOr<std::unique_ptr<BufferAssignment>> BufferAssigner::Run(
821 const HloModule* module, std::unique_ptr<HloOrdering> hlo_ordering,
822 BufferValue::SizeFunction buffer_size,
823 LogicalBuffer::AlignmentFunction color_alignment,
824 bool allocate_buffers_for_constants, BufferAssigner::Colorer colorer,
825 const absl::flat_hash_set<HloOpcode>& reuse_checker,
826 HloDataflowAnalysis::CanShareBuffer can_share_buffer,
827 std::unique_ptr<PresetAssignments> preset_assignments) {
828 BufferAssigner assigner(allocate_buffers_for_constants, std::move(colorer),
829 reuse_checker, std::move(preset_assignments));
830 return assigner.CreateAssignment(
831 module, std::move(hlo_ordering), std::move(buffer_size),
832 std::move(color_alignment), std::move(can_share_buffer));
833}
834
835bool BufferAssigner::LiveRangeInterferes(const HloValue* buffer1,
836 const HloValue* buffer2,

Callers

nothing calls this directly

Calls 1

CreateAssignmentMethod · 0.80

Tested by

no test coverage detected