Will Feng 0f45aaf441 Disable autocast when running joint graph passes (#162304) ...

Fixes #159469. See https://github.com/pytorch/pytorch/issues/159469#issuecomment-3221474027 for root-cause analysis.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/162304
Approved by: https://github.com/bdhirsh, https://github.com/zou3519, https://github.com/eellison

2025-09-06 00:57:58 +00:00

..

_activation_checkpointing

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

_aot_autograd

Disable autocast when running joint graph passes (#162304)

2025-09-06 00:57:58 +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 support for param mutation under inference mode (#159661)

2025-08-14 03:34:04 +00:00

apis.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

autograd_function.py

Typo fixes for "overridden" in comments and function names (#155944)

2025-06-14 03:37:38 +00:00

batch_norm_replacement.py

remove allow-untyped-defs for torch/_functorch/batch_norm_replacement.py (#143438)

2024-12-18 09:01:06 +00:00

benchmark_utils.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

compile_utils.py

[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)

2025-07-12 05:47:06 +00:00

compilers.py

[BE][2/16] fix typos in torch/ (torch/_*/) (#156312)

2025-07-12 05:47:06 +00:00

config.py

[MTIA] Allow users who know what they are doing to ignore all device mismatches in tracing and take a preferred device. (#159931)

2025-08-07 22:37:15 +00:00

deprecated.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

eager_transforms.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

functional_call.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

fx_minifier.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

make_functional.py

[BE][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

partitioners.py

[invoke_subgraph][inductor] Thread graphsafe rng input states for hops (#160713)

2025-08-21 20:41:29 +00:00

predispatch.py

Add support for tracing vmap in pre-dispatch export (#154650)

2025-08-20 19:31:07 +00:00

pyfunctorch.py

[dynamo] Guard serialization for FUNCTORCH_STACK_MATCH (#152616)

2025-05-05 18:05:56 +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][PYFMT] migrate PYFMT for torch/_[a-h]*/ to ruff format (#144551)

2025-06-25 06:16:06 +00:00

utils.py

PEP585 update - torch/_functorch (#145139)

2025-01-19 07:06:10 +00:00

vmap.py

Add support for tracing vmap in pre-dispatch export (#154650)

2025-08-20 19:31:07 +00:00