build CUDA plugin in CI

pytorch · Jan 18, 2024 · baf70be · baf70be
1 parent cfb60fa
commit baf70be
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/.circleci/common.sh b/.circleci/common.sh
@@ -104,6 +104,7 @@ function build_torch_xla() {
   XLA_DIR=$1
   pushd "$XLA_DIR"
   python setup.py install
+  pip install plugins/cuda --no-build-isolation -v
   popd
 }