Skip to content

Commit 34acbf0

Browse files
[XLA:TPU] Disable memory space assignment pass via exec_time_optimization_effort.
PiperOrigin-RevId: 715353785
1 parent 941ee63 commit 34acbf0

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

xla/service/memory_space_assignment/memory_space_assignment.h

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -323,6 +323,8 @@ class MemorySpaceAssignment {
323323
const HloAliasAnalysis& alias_analysis,
324324
std::vector<int64_t>* alt_mem_bytes_occupied = nullptr);
325325

326+
static constexpr absl::string_view kName = "memory-space-assignment";
327+
326328
protected:
327329
// Main driver of the memory space assignment pass.
328330
virtual absl::StatusOr<std::unique_ptr<PresetAssignments>>

0 commit comments

Comments
 (0)