Remove CUDA plugin. (#9597)

This PR removes the CUDA plugin in `plugins/cuda` directory, as well as
explicit mentions and usages. This is in line with the CUDA deprecation
that started on release 2.8.

There are still a few mentions to `torch_xla_cuda_plugin` at
`.github/ci.md` markdown file. But, I will leave those for a future PR,
since we would need to re-write those parts.

Author: ysiraichi

.github/scripts/run_tests.sh

@@ -30,14 +30,7 @@ function run_torch_xla_cpp_tests() {
 
   TORCH_DIR=$(python -c "import pkgutil; import os; print(os.path.dirname(pkgutil.get_loader('torch').get_filename()))")
   export LD_LIBRARY_PATH=${LD_LIBRARY_PATH}:${TORCH_DIR}/lib
-  if [ -x "$(command -v nvidia-smi)" ]; then
-    CUDA_PLUGIN_DIR=$(python -c "import pkgutil; import os; print(os.path.dirname(pkgutil.get_loader('torch_xla_cuda_plugin').get_filename()))")
-    export PJRT_LIBRARY_PATH=$CUDA_PLUGIN_DIR/lib/pjrt_c_api_gpu_plugin.so
-    export PJRT_DEVICE=LIBRARY
-    export PJRT_DYNAMIC_PLUGINS=1
-  else
-    export PJRT_DEVICE=CPU
-  fi
+  export PJRT_DEVICE=CPU
   export XLA_EXPERIMENTAL="nonzero:masked_select:nms"
 
   test_names=("test_aten_xla_tensor_1"

infra/ansible/Dockerfile

@@ -10,7 +10,6 @@ COPY . /ansible
 ARG ansible_vars
 # HACK: install build dependencies only, but skip build step
 RUN ansible-playbook -vvv playbook.yaml -e "stage=build" -e "${ansible_vars}" --tags "bazel,configure_env,install_deps"
-RUN ansible-playbook -vvv playbook.yaml -e "stage=build_plugin" -e "${ansible_vars}"
 RUN ansible-playbook -vvv playbook.yaml -e "stage=build" -e "${ansible_vars}" --skip-tags=fetch_srcs,install_deps
 
 FROM python:${python_version}-${debian_version} AS release

infra/ansible/e2e_tests.Dockerfile

@@ -10,7 +10,6 @@ COPY . /ansible
 # Build PyTorch and PyTorch/XLA wheels.
 ARG ansible_vars
 RUN ansible-playbook -vvv playbook.yaml -e "stage=build" -e "${ansible_vars}"
-RUN ansible-playbook -vvv playbook.yaml -e "stage=build_plugin" -e "${ansible_vars}" --skip-tags=fetch_srcs,install_deps
 
 FROM python:${python_version}-${debian_version}
 

infra/ansible/playbook.yaml

@@ -16,7 +16,7 @@
           "Pass the required variable with: --e \"{{ item.name }}=<value>\""
       loop:
         - name: stage
-          pattern: ^(build|build_plugin|release)$
+          pattern: ^(build|release)$
         - name: arch
           pattern: ^(aarch64|amd64)$
         - name: accelerator
@@ -88,17 +88,6 @@
       when: stage == "build"
       tags: build_srcs
 
-    - role: build_plugin
-      vars:
-        src_root: "/src"
-        env_vars: "{{
-            build_env.common | default({}, true) |
-            combine(build_env[arch] | default({}, true)) |
-            combine(build_env[accelerator] | default({}, true))
-          }}"
-      when: stage == "build_plugin"
-      tags: build_plugin
-
     - role: configure_env
       vars:
         env_vars: "{{