[XLA:MSA] Remove unnecessary Extend() call in memory space assignment…

…. This Extend() call would also lead to a memory assignment issue since it wasn't accompanied by the necessary chunk commit requests. We also add a VerifyAllocations() function that uses a BufferIntervalTree to check for overlapping Allocations before scheduling the asynchronous copies. This is an extra check for the correctness of MsaAlgorithm allocations, and is only applied if options_.verify is enabled in MSA options. options_.verify is disabled by default.

PiperOrigin-RevId: 693110290

xla/service/memory_space_assignment/BUILD

@@ -79,6 +79,7 @@ xla_cc_test(
     deps = [
         ":algorithm",
         ":allocation",
+        ":allocation_value",
         ":buffer_interval_comparator",
         ":cost_analysis",
         ":memory_space_assignment",
@@ -107,7 +108,6 @@ xla_cc_test(
         "//xla/service/heap_simulator:allocation_block",
         "//xla/tests:hlo_test_base",
         "//xla/tests:test_utils",
-        "//xla/tests:verified_hlo_module",
         "//xla/tests:xla_internal_test_main",
         "//xla/tsl/lib/core:status_test_util",
         "@com_google_absl//absl/algorithm:container",
@@ -273,6 +273,7 @@ cc_library(
     srcs = [],
     hdrs = ["options.h"],
     deps = [
+        ":allocation_value",
         ":buffer_interval_comparator",
         ":cost_analysis",
         ":memory_space_assignment_proto_cc",
@@ -508,6 +509,7 @@ cc_library(
     hdrs = ["algorithm.h"],
     deps = [
         ":allocation",
+        ":allocation_value",
         ":buffer_interval_comparator",
         ":cost_analysis",
         ":memory_bound_loop_optimizer",
@@ -577,6 +579,18 @@ cc_library(
     ],
 )
 
+cc_library(
+    name = "allocation_value",
+    hdrs = ["allocation_value.h"],
+    deps = [
+        ":allocation",
+        "//xla/hlo/ir:hlo",
+        "//xla/service:hlo_value",
+        "@com_google_absl//absl/container:flat_hash_set",
+        "@com_google_absl//absl/types:span",
+    ],
+)
+
 xla_cc_test(
     name = "prefetch_interval_picker_test",
     srcs = ["prefetch_interval_picker_test.cc"],