fix workflow

pytorch · Apr 10, 2024 · e46768e · e46768e
1 parent 4d7b5bc
commit e46768e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.github/workflows/build_and_test.yml b/.github/workflows/build_and_test.yml
@@ -31,6 +31,7 @@ jobs:
   build-cuda-plugin:
     name: "Build XLA CUDA plugin"
     with:
+      uses: ./.github/workflows/_build_plugin.yml
       dev-image: us-central1-docker.pkg.dev/tpu-pytorch-releases/docker/development:3.10_cuda_12.3
 
   test-cpu: