Skip to content
Merged
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
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