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
This commit is contained in:
Aaron Orenstein
2024-08-20 10:26:45 +00:00
committed by PyTorch MergeBot
parent 11af423eca
commit 241df7e7f8
4 changed files with 62 additions and 17 deletions

View File

@ -17,7 +17,6 @@ jinja2
fsspec
lintrunner
ninja
redis
# setuptools was removed from default python install
setuptools ; python_version >= "3.12"
packaging