Delete ARM64 Kokoro build now that Github Actions based build is bloc…

…king

PiperOrigin-RevId: 722798426

third_party/xla/build_tools/ci/build.py

@@ -89,7 +89,6 @@ def _write_to_sponge_config(key, value) -> None:
 class BuildType(enum.Enum):
   """Enum representing all types of builds."""
   CPU_X86_SELF_HOSTED = enum.auto()
-  CPU_ARM64 = enum.auto()
   CPU_ARM64_SELF_HOSTED = enum.auto()
   GPU = enum.auto()
   GPU_CONTINUOUS = enum.auto()
@@ -184,11 +183,7 @@ def commands(self) -> List[List[str]]:
 
     # pyformat:disable
 
-    if self.type_ == BuildType.CPU_ARM64 and using_docker:
-      # We would need to install parallel, but `apt` hangs regularly on Kokoro
-      # VMs due to yaqs/eng/q/4506961933928235008
-      cmds.append(["docker", "pull", self.image_url])
-    elif using_docker:
+    if using_docker:
       cmds.append(retry(["docker", "pull", self.image_url]))
 
     container_name = "xla_ci"
@@ -309,16 +304,6 @@ def nvidia_gpu_build_with_compute_capability(
     "-requires-gpu-amd",
     "-not_run:arm",
 )
-_CPU_ARM64_BUILD = Build(
-    type_=BuildType.CPU_ARM64,
-    repo="openxla/xla",
-    image_url=_ML_BUILD_ARM64_IMAGE,
-    configs=("warnings", "rbe_cross_compile_linux_arm64", "nonccl"),
-    target_patterns=_XLA_DEFAULT_TARGET_PATTERNS,
-    options={**_DEFAULT_BAZEL_OPTIONS, "build_tests_only": True},
-    build_tag_filters=cpu_arm_tag_filter,
-    test_tag_filters=cpu_arm_tag_filter,
-)
 _CPU_ARM64_SELF_HOSTED_BUILD = Build(
     type_=BuildType.CPU_ARM64_SELF_HOSTED,
     repo="openxla/xla",
@@ -488,9 +473,7 @@ def nvidia_gpu_build_with_compute_capability(
 )
 
 _KOKORO_JOB_NAME_TO_BUILD_MAP = {
-    "tensorflow/xla/linux/arm64/build_cpu": _CPU_ARM64_BUILD,
     "tensorflow/xla/linux/gpu/build_gpu": _GPU_BUILD,
-    "tensorflow/xla/linux/github_continuous/arm64/build_cpu": _CPU_ARM64_BUILD,
     "tensorflow/xla/linux/github_continuous/build_gpu": _GPU_BUILD,
     "tensorflow/xla/macos/github_continuous/cpu_py39_full": _MACOS_X86_BUILD,
     "tensorflow/xla/jax/cpu/build_cpu": _JAX_CPU_BUILD,

third_party/xla/build_tools/ci/golden_commands.txt

@@ -1,12 +1,3 @@
-# BEGIN BuildType.CPU_ARM64
-$KOKORO_ARTIFACTS_DIR/github/xla/.kokoro/generate_index_html.sh index.html
-docker pull us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build-arm64:latest
-docker run --detach --name=xla_ci --rm --interactive --tty --volume=./github:/github --workdir=/github/xla us-central1-docker.pkg.dev/tensorflow-sigs/tensorflow/ml-build-arm64:latest bash
-docker exec xla_ci parallel --ungroup --retries 3 --delay 15 --nonall -- bazel build --build_tag_filters=-no_oss,-gpu,-requires-gpu-nvidia,-requires-gpu-amd,-not_run:arm --test_tag_filters=-no_oss,-gpu,-requires-gpu-nvidia,-requires-gpu-amd,-not_run:arm --config=warnings --config=rbe_cross_compile_linux_arm64 --config=nonccl --test_output=errors --verbose_failures --keep_going --nobuild_tests_only --profile=profile.json.gz --flaky_test_attempts=3 --jobs=150 --bes_upload_mode=fully_async --build_tests_only --nobuild -- //xla/... //build_tools/... @local_tsl//tsl/...
-docker exec xla_ci bazel test --build_tag_filters=-no_oss,-gpu,-requires-gpu-nvidia,-requires-gpu-amd,-not_run:arm --test_tag_filters=-no_oss,-gpu,-requires-gpu-nvidia,-requires-gpu-amd,-not_run:arm --config=warnings --config=rbe_cross_compile_linux_arm64 --config=nonccl --test_output=errors --verbose_failures --keep_going --nobuild_tests_only --profile=profile.json.gz --flaky_test_attempts=3 --jobs=150 --bes_upload_mode=fully_async --build_tests_only -- //xla/... //build_tools/... @local_tsl//tsl/...
-docker exec xla_ci bazel analyze-profile profile.json.gz
-docker stop xla_ci
-# END BuildType.CPU_ARM64
 # BEGIN BuildType.CPU_ARM64_SELF_HOSTED
 parallel --ungroup --retries 3 --delay 15 --nonall -- bazel build --build_tag_filters=-no_oss,-gpu,-requires-gpu-nvidia,-requires-gpu-amd,-not_run:arm --test_tag_filters=-no_oss,-gpu,-requires-gpu-nvidia,-requires-gpu-amd,-not_run:arm --config=warnings --config=rbe_cross_compile_linux_arm64 --config=nonccl --test_output=errors --verbose_failures --keep_going --nobuild_tests_only --profile=profile.json.gz --flaky_test_attempts=3 --jobs=150 --bes_upload_mode=fully_async --build_tests_only --nobuild -- //xla/... //build_tools/... @local_tsl//tsl/...
 bazel test --build_tag_filters=-no_oss,-gpu,-requires-gpu-nvidia,-requires-gpu-amd,-not_run:arm --test_tag_filters=-no_oss,-gpu,-requires-gpu-nvidia,-requires-gpu-amd,-not_run:arm --config=warnings --config=rbe_cross_compile_linux_arm64 --config=nonccl --test_output=errors --verbose_failures --keep_going --nobuild_tests_only --profile=profile.json.gz --flaky_test_attempts=3 --jobs=150 --bes_upload_mode=fully_async --build_tests_only -- //xla/... //build_tools/... @local_tsl//tsl/...