xla
Update documentations for scan cache
#9394
Merged
Overview
Commits
13
Changes
View On GitHub
Commits
  • Add cache to `value_and_grad_partitioned`, which perform AOTAutograd tracing that is time consuming.
    iwknow committed 315 days ago
  • 1. Parameterize some of the existing tests with `is_fn_pure`. 2. add `setUp()` to clear cache. 3. update `test_scan_computation_cache` test to count how many times the fn is called
    iwknow committed 294 days ago
  • Merge branch 'pytorch:master' into scan-cache
    iwknow committed 294 days ago
  • Update the default value of `is_fn_pure` to `True`
    iwknow committed 294 days ago
  • Update test cases.
    iwknow committed 294 days ago
  • format
    iwknow committed 294 days ago
  • Add cache support for scan_layers
    iwknow committed 293 days ago
  • Merge branch 'pytorch:master' into scan-cache
    iwknow committed 293 days ago
  • Merge branch 'pytorch:master' into scan-cache
    iwknow committed 286 days ago
  • Merge branch 'master' into scan-cache
    iwknow committed 280 days ago
  • correct format
    iwknow committed 280 days ago
  • Merge branch 'pytorch:master' into scan-cache
    iwknow committed 280 days ago
  • Update documentation and examples for scan cache.
    iwknow committed 280 days ago
FAQ Terms Privacy Refunds Impressum
Loading