Files
pytorch/requirements.txt
Aaron Orenstein 241df7e7f8 Add multi-cache autotune test (#133868)
Summary:
The existing tests didn't cover a case where we had multiple autotunes in a single graph.  Add a test to demonstrate that case.

Also added a test dependency on redis and removed the "fake redis" from the previous PR (#133579)

Test Plan: unit tests

Reviewed By: oulgen

Differential Revision: D61178861

Pull Request resolved: https://github.com/pytorch/pytorch/pull/133868
Approved by: https://github.com/oulgen
2024-08-20 10:26:45 +00:00

24 lines
442 B
Plaintext

# Python dependencies required for development
astunparse
expecttest!=0.2.0
hypothesis
numpy
psutil
pyyaml
requests
setuptools
types-dataclasses
typing-extensions>=4.8.0
sympy==1.12.1 ; python_version == "3.8"
sympy==1.13.1 ; python_version >= "3.9"
filelock
networkx
jinja2
fsspec
lintrunner
ninja
# setuptools was removed from default python install
setuptools ; python_version >= "3.12"
packaging
optree>=0.12.0 ; python_version <= "3.12"