Basil Wong 2f1dbfea02 Logging Refactor - Remove Print Statements (#139782) ...

Summary:
Removes print statements and implements logging via the logging library.

Hopefully this will allow more control on the level of logging when running models.

Test Plan:
```
AOT_PARTITIONER_DEBUG=1 buck2 run @mode/opt //aps_models/ads/icvr:icvr_launcher -- mode=local_fb_fm_v4 launcher.num_workers=2
```

Resulting output paste: P1674535630
* Full logs paste: P1674535621

```
pastry P1674535621 | grep "functorch/partitioners.py" | pastry
```

Logging results: P1674549514

Differential Revision: D61678215

Pull Request resolved: https://github.com/pytorch/pytorch/pull/139782
Approved by: https://github.com/paryxyt, https://github.com/jansel

2024-11-13 23:09:18 +00:00

..

_aot_autograd

Revert "[logging] Overhaul dynamo_timed and CompilationMetrics logging. (#139849)"

2024-11-13 18:47:51 +00:00

__init__.py

Revert "Replace _dynamo.config with an object instead of module (#96455)"

2023-06-05 07:06:14 +00:00

aot_autograd.py

Add option to dynamo_timed and chromium_event_logger for logging pt2 compile events (#139309)

2024-11-01 02:40:25 +00:00

apis.py

Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"

2024-11-05 23:10:38 +00:00

autograd_function.py

Fix torch.library.register_vmap (#137306)

2024-10-04 03:46:35 +00:00

batch_norm_replacement.py

Revert "[BE] typing for decorators - _library/custom_ops (#131578)"

2024-07-28 03:29:32 +00:00

benchmark_utils.py

[Functorch] Fix devices Parameter Type in benchmark_utilization Function (#138774)

2024-10-25 19:25:18 +00:00

compile_utils.py

Fix unintentional deduplication of returned tensors (#134726)

2024-09-04 23:42:56 +00:00

compilers.py

[BE] UFMT directory torch/_functorch (#123723)

2024-04-12 08:04:51 +00:00

config.py

[AOTAutograd] Handle edge cases for donated buffer & enable in oss (#139669)

2024-11-05 18:38:20 +00:00

deprecated.py

Flip default value for mypy disallow_untyped_defs [2/11] (#127839)

2024-06-08 18:23:08 +00:00

eager_transforms.py

Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"

2024-11-05 23:10:38 +00:00

functional_call.py

Allow optional positional arguments for torch.func.functional_call (#134643)

2024-09-12 15:22:06 +00:00

fx_minifier.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)

2024-07-31 10:42:50 +00:00

make_functional.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)

2024-07-31 10:42:50 +00:00

partitioners.py

Logging Refactor - Remove Print Statements (#139782)

2024-11-13 23:09:18 +00:00

pyfunctorch.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)

2024-07-31 10:42:50 +00:00

python_key.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)

2024-07-31 10:42:50 +00:00

pytree_hacks.py

[BE] wrap deprecated function/class with typing_extensions.deprecated (#127689)

2024-06-02 12:30:43 +00:00

top_operators_github_usage.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)

2024-07-31 10:42:50 +00:00

utils.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)

2024-07-31 10:42:50 +00:00

vmap.py

[BE][Easy][14/19] enforce style for empty lines in import segments in torch/_[a-c]*/ and torch/_[e-h]*/ and torch/_[j-z]*/ (#129765)

2024-07-31 10:42:50 +00:00