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

Enable UFMT on test/functorch (#123541)

2024-04-15 06:21:52 +00:00

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

Enable UFMT on test/functorch (#123541)

2024-04-15 06:21:52 +00:00

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