OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 00:20:18 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity
f9fa138a39
pytorch/test/functorch
History
Basil Wong 4941719061 Enable logging for absolute memory estimation (#158799) 
Summary: Update the Auto AC logging so that it also provides the *absolute* memory estimations for each node.

Test Plan:
(aps-gem_omnifm_v2_mwb_dynamic_005_budget-f23a84c3d8): https://fburl.com/ai_infra/0r738h5r

{F1980393481}

* Memory Recorded in bytes

---

```
buck2 test //caffe2/test/functorch:test_ac_logging
```
https://www.internalfb.com/intern/testinfra/testrun/14918173863021573

Rollback Plan:

Differential Revision: D78580107

Pull Request resolved: https://github.com/pytorch/pytorch/pull/158799
Approved by: https://github.com/jansel
2025-09-22 18:36:49 +00:00
..
dim torchdim Python port (#160236) 2025-09-21 03:01:04 +00:00
attn_ft.py torchdim Python port (#160236) 2025-09-21 03:01:04 +00:00
attn_positional.py
common_utils.py [BE][PYFMT] migrate PYFMT for test/[a-h]*/ to ruff format (#144555) 2025-06-24 04:53:54 +00:00
discover_coverage.py Apply ruff fixes to tests (#146140) 2025-02-04 05:41:01 +00:00
functorch_additional_op_db.py fix numpy compatibility for 2d small list indices (#154806) 2025-06-04 01:58:52 +00:00
test_ac_knapsack.py [BE][PYFMT] migrate PYFMT for test/[a-h]*/ to ruff format (#144555) 2025-06-24 04:53:54 +00:00
test_ac_logging.py Enable logging for absolute memory estimation (#158799) 2025-09-22 18:36:49 +00:00
test_ac.py rename-HAS_CUDA-to-HAS_CUDA_AND_TRITON (#159883) 2025-08-08 15:44:52 +00:00
test_aot_joint_with_descriptors.py Ensure export joint with descriptors + compile works (#159337) 2025-07-29 17:43:52 +00:00
test_aotdispatch.py [Opitmus] fix fp8 activation quatization for duplicates forward output (#163364) 2025-09-20 06:33:20 +00:00
test_control_flow.py Turn on capture_scalar_outputs when fullgraph=True (#163121) 2025-09-18 21:24:15 +00:00
test_dims.py torchdim Python port (#160236) 2025-09-21 03:01:04 +00:00
test_eager_transforms.py [ROCm] Enabling several UTs (#161715) 2025-09-09 15:49:21 +00:00
test_logging.py
test_memory_efficient_fusion.py [BE][PYFMT] migrate PYFMT for test/[a-h]*/ to ruff format (#144555) 2025-06-24 04:53:54 +00:00
test_minifier.py Add None return type to init -- tests (#132352) 2024-08-01 15:44:51 +00:00
test_ops.py [ROCm] fix miopen batchnorm changing output format (#162112) 2025-09-11 19:37:48 +00:00
test_parsing.py [BE][PYFMT] migrate PYFMT for test/[a-h]*/ to ruff format (#144555) 2025-06-24 04:53:54 +00:00
test_rearrange.py [BE][PYFMT] migrate PYFMT for test/[a-h]*/ to ruff format (#144555) 2025-06-24 04:53:54 +00:00
test_vmap_registrations.py Remove guard_size_oblivious from default contiguity python check, and add aten.sym_is_contiguous. [attempt2] (#160869) 2025-09-08 22:59:13 +00:00
test_vmap.py [BE][3/6] fix typos in test/ (#157637) 2025-07-17 12:08:33 +00:00
xfail_suggester.py [BE][Easy][9/19] enforce style for empty lines in import segments in test/[e-h]*/ (#129760) 2024-07-17 14:25:29 +00:00