llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity
torch-mlir/test-requirements.txt

6 lines
52 B
Plaintext
Raw Normal View History

python: separate build- and test-related pip dependencies (#1874) We want to ensure that pip packages required for building torch-mlir should be included in the dependencies of torch-mlir, but we don't want the pip packages required for _testing_ of torch-mlir to be included among the dependencies. To be able to specify and install one set of dependencies and not the other, this patch separates the pip packages into two files: build-requirements.txt and test-requirements.txt. This patch also updates references to the requirements.txt file so that CI builds that run end-to-end tests install test-related pip dependencies while everything else (including WHL builds) sticks to just the build-related pip dependencies. Despite this change, this patch should not affect a torch-mlir developer's workflow. More precisely, since this patch makes the top-level requirements.txt file refer to both build-requirements.txt and test-requirements.txt files, a torch-mlir developer should be able to continue referring to the requirements.txt file without any impact.
2023-02-14 11:22:09 +08:00
pillow
dill
multiprocess
Add support for bfloat16 in fximporter (#2896) this introduces an additional soft dependency on the python ml_dtypes python packages in order to support bfloat16 Addresses #2843
2024-02-15 06:24:25 +08:00
onnx==1.15.0
[ci] Fix mpmath 1.4.0 error by forcing 1.3.0 (#2946) `mpmath 1.4.0` changes some import locations breaking `torch`. Changing to `1.3.0` to avoid breaking on `python 3.11`
2024-02-24 05:13:54 +08:00
mpmath==1.3.0