Split out MemorySpaceAssignmentTest class for re-use.

PiperOrigin-RevId: 707258018

Author: Google-ML-Automation

xla/service/memory_space_assignment/BUILD

@@ -84,6 +84,7 @@ xla_cc_test(
         ":cost_analysis",
         ":memory_space_assignment",
         ":memory_space_assignment_proto_cc",
+        ":memory_space_assignment_test_base",
         ":options",
         ":prefetch_interval_picker",
         ":repacking",
@@ -160,6 +161,34 @@ cc_library(
     ],
 )
 
+cc_library(
+    name = "memory_space_assignment_test_base",
+    testonly = True,
+    hdrs = ["memory_space_assignment_test_base.h"],
+    deps = [
+        ":buffer_interval_comparator",
+        ":cost_analysis",
+        ":memory_space_assignment",
+        ":options",
+        ":prefetch_interval_picker",
+        "//xla:shape_util",
+        "//xla/hlo/analysis:hlo_alias_analysis",
+        "//xla/hlo/ir:hlo",
+        "//xla/hlo/transforms:instruction_hoister",
+        "//xla/hlo/utils:hlo_live_range",
+        "//xla/service:buffer_value",
+        "//xla/service:hlo_buffer",
+        "//xla/service:hlo_cost_analysis",
+        "//xla/service:hlo_value",
+        "//xla/tests:hlo_test_base",
+        "//xla/tsl/lib/core:status_test_util",
+        "@com_google_absl//absl/status:statusor",
+        "@tsl//tsl/platform:status",
+        "@tsl//tsl/platform:statusor",
+        "@tsl//tsl/platform:test",
+    ],
+)
+
 cc_library(
     name = "utils",
     srcs = ["utils.cc"],