diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml index 09d873ff428..2ba6b025bdc 100644 --- a/.github/workflows/build_and_test.yml +++ b/.github/workflows/build_and_test.yml @@ -30,9 +30,11 @@ jobs: build-cuda-plugin: name: "Build XLA CUDA plugin" + uses: ./.github/workflows/_build_plugin.yml with: - uses: ./.github/workflows/_build_plugin.yml dev-image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/development:3.10_cuda_12.3 + secrets: + gcloud-service-key: ${{ secrets.GCLOUD_SERVICE_KEY }} test-cpu: name: "CPU tests"