update bazel gpu local workflow

Author: nitins17

.github/workflows/bazel_gpu_local.yml

@@ -27,15 +27,13 @@ jobs:
     steps:
       - uses: actions/checkout@v3
       - name: Build jaxlib
-        run: ./ci/build_artifacts.sh jaxlib
+        run: ./ci/build_artifacts.sh "ci/envs/build_artifacts/jaxlib"
       - name: Build jax-cuda-plugin
-        run: ./ci/build_artifacts.sh jax-cuda-plugin
+        run: ./ci/build_artifacts.sh "ci/envs/build_artifacts/jax-cuda-plugin"
       - name: Build jax-cuda-pjrt
-        run: ./ci/build_artifacts.sh jax-cuda-pjrt
+        run: ./ci/build_artifacts.sh "ci/envs/build_artifacts/jax-cuda-pjrt"
       # Halt for testing
       - name: Wait For Connection
         uses: ./actions/ci_connection/
       - name: Run Bazel GPU tests locally
-        env:
-          ENV_FILE: ci/envs/run_tests/bazel_gpu_local
-        run: ./ci/run_bazel_test_gpu.sh
+        run: ./ci/run_bazel_test.sh "ci/envs/run_tests/bazel_gpu_local"