Skip to content

Commit

Permalink
Update pre_build_script.sh (#390)
Browse files Browse the repository at this point in the history
* Update pre_build_script.sh

* Update smoke_test.py

* Update post_build_script.sh

* Update post_build_script.sh
  • Loading branch information
msaroufim authored Jun 17, 2024
1 parent 1b78dda commit d043d23
Show file tree
Hide file tree
Showing 3 changed files with 3 additions and 7 deletions.
4 changes: 1 addition & 3 deletions packaging/post_build_script.sh
Original file line number Diff line number Diff line change
Expand Up @@ -30,6 +30,4 @@ popd
MANYWHEEL_NAME=$(ls dist/)
# Try to install the new wheel
pip install "dist/${MANYWHEEL_NAME}"
# and validating it by running the unit tests. Some tests are failing here and
# there, so let's add more of them later
pytest -v test/test_ops.py
python -c "import torchao"
2 changes: 0 additions & 2 deletions packaging/pre_build_script.sh
Original file line number Diff line number Diff line change
Expand Up @@ -10,5 +10,3 @@ set -eux
echo "This script is run before building torchao binaries"

pip install setuptools wheel twine auditwheel
pip install -r requirements.txt
pip install -r dev-requirements.txt
4 changes: 2 additions & 2 deletions packaging/smoke_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -6,14 +6,14 @@
# LICENSE file in the root directory of this source tree.

import subprocess
import torchao.ops
import torchao


def main():
"""
Run torchao binary smoke tests like importing and performing simple ops
"""
print(dir(torchao.ops))
print(dir(torchao))


if __name__ == "__main__":
Expand Down

0 comments on commit d043d23

Please sign in to comment.