Update documentations for scan cache #9394
tengyifei
merged
13 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
Merge branch 'pytorch:master' into scan-cache
51aa6a18
Merge branch 'master' into scan-cache
47892dfd
correct format
f5c2ded7
Merge branch 'pytorch:master' into scan-cache
03ac3dde
Update documentation and examples for scan cache.
453e6b69
tengyifei
approved these changes
on 2025-06-19
Login to write a write a comment.
Login via GitHub