Add cache support for scan_layers #9297
tengyifei
merged
11 commits into
pytorch:master
from
iwknow:scan-cache
Add cache to `value_and_grad_partitioned`, which perform AOTAutograd …
d17b4f98
1. Parameterize some of the existing tests with `is_fn_pure`. 2. add …
edd3290f
Merge branch 'pytorch:master' into scan-cache
7313adcd
Update the default value of `is_fn_pure` to `True`
21525c54
Update test cases.
385600e7
format
f90a36dd
Add cache support for scan_layers
12d0dfdd
Merge branch 'pytorch:master' into scan-cache
720af9de
qihqi
approved these changes
on 2025-06-12
Merge branch 'pytorch:master' into scan-cache
51aa6a18
tengyifei
approved these changes
on 2025-06-13
Merge branch 'master' into scan-cache
47892dfd
correct format
f5c2ded7
disabled auto-merge 266 days ago
Head branch was pushed to by a user without write access
Head branch was pushed to by a user without write access
disabled auto-merge 266 days ago
Manually disabled by user
Manually disabled by user
Login to write a write a comment.
Login via GitHub