mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
6096c0fc74
This diff moves export run_decompositions to use aot_export_joint_with_descriptors instead of aot_export_module. Doing so, i ran into 2 main bugs: 1) aot_export_joint_with_descriptors don't correctly pass in record_nn_module_stack flag that is needed to populate nn_module_stack by switching the internal tracer. 2) When creating symint with negative inputs, we need to pass in positive=False. This didn't matter before because aot_autograd directly returns integer inputs instead of creating symint. Pull Request resolved: https://github.com/pytorch/pytorch/pull/165931 Approved by: https://github.com/zhxchen17 |
||
|---|---|---|
| .. | ||
| _activation_checkpointing | ||
| _aot_autograd | ||
| __init__.py | ||
| aot_autograd.py | ||
| apis.py | ||
| autograd_function.py | ||
| batch_norm_replacement.py | ||
| benchmark_utils.py | ||
| compile_utils.py | ||
| compilers.py | ||
| config.py | ||
| deprecated.py | ||
| eager_transforms.py | ||
| functional_call.py | ||
| fx_minifier.py | ||
| make_functional.py | ||
| partitioners.py | ||
| predispatch.py | ||
| pyfunctorch.py | ||
| python_key.py | ||
| pytree_hacks.py | ||
| top_operators_github_usage.py | ||
| utils.py | ||
| vmap.py | ||