mirror of https://github.com/llvm/torch-mlir
c39a90e6e9
* It is optional to link against Python3_LIBRARIES. Check that and don't do it if they don't exist for this config. * Clean and auditwheel need to operate on sanitized package names. So "torch_mlir" vs "torch-mlir". * Adds a pyproject.toml file that pins the build dependencies needed to detect both Torch and Python (the MLIR Python build was failing to detect because Numpy wasn't in the pip venv). * Commented out auditwheel: These wheels are not PyPi compliant since they weak link to libtorch at runtime. However, they should be fine to deploy to users. * Adds the --extra-index-url to the pip wheel command, allowing PyTorch to be found. * Hack setup.py to remove the _mlir_libs dir before building. This keeps back-to-back versions from accumulating in the wheels for subsequent versions. IREE has a more principled way of doing this, but what I have here should work. |
||
|---|---|---|
| .. | ||
| python_deploy | ||
| torchscript_e2e_heavydep_tests | ||
| build_python_wheels.sh | ||
| build_standalone.sh | ||
| update_shape_lib.sh | ||
| update_torch_ods.sh | ||
| write_env_file.sh | ||