Skip to content

Commit 00fac78

Browse files
authored
Build torch_xla wheel in build script (#8782)
1 parent cee0820 commit 00fac78

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

scripts/build_developer.sh

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -21,12 +21,16 @@ cd ..
2121
if [ -d "vision" ]; then
2222
cd vision
2323
python3 setup.py develop
24+
cd ..
2425
fi
2526

2627
# Install torch_xla
27-
cd ..
2828
cd pytorch/xla
2929
pip uninstall torch_xla -y
30+
31+
# Optional: build the wheel.
32+
python3 setup.py bdist_wheel
33+
3034
python3 setup.py develop
3135

3236
# libtpu is needed to talk to the TPUs. If TPUs are not present,

0 commit comments

Comments
 (0)