Animesh Jain 0e5773b7fa [dynamo][export] Do not graph break on torch.autograd._profiler_enabled for export (#164418) ...

Actually we would like to not graph break even in the case of Dynamo. But there is a weird-unsolved bug with Kineto + Dynamo when there are distributed jobs that lead to NCCL timeouts. This bug is a rare edege case, but we have not been able to root cause it yet.

But for export, we do not anticipate JIT tracing in distributed job training and therefore this PR is safe for export.

Fixes #ISSUE_NUMBER

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164418
Approved by: https://github.com/StrongerXi, https://github.com/williamwen42

2025-10-02 09:00:00 +00:00

..

cpython/3_13

Fix range.__getitem__() (#161804)

2025-09-04 02:33:03 +00:00

mock_modules

…

__init__.py

…

_test_nested_graph_breaks_helper.py

[dynamo, nested graph breaks] add nested graph break tests (#144516)

2025-08-27 03:00:56 +00:00

imports_non_circular_repro.py

[Bugfix] Fix circular import between export and dynamo from tensor fn map (#158931)

2025-07-24 22:24:57 +00:00

mock_store_global_crossfile_inline.py

[Reland] Fix inlining module-scoped store global (#132439)

2024-08-02 09:13:52 +00:00

test_activation_checkpointing.py

[dynamo][sac] Support functools partial context_fn for sac (#164308)

2025-10-01 02:47:55 +00:00

test_after_aot.py

[BE][Easy][11/19] enforce style for empty lines in import segments in test/dy*/ (#129762)

2024-07-27 17:43:53 +00:00

test_aot_autograd_cache.py

Reapply "Make functionalization ViewMeta serializable with pickle. (#143712)" (#163769)

2025-09-25 10:27:37 +00:00

test_aot_autograd.py

Graph break with error message (#158800)

2025-07-29 17:34:49 +00:00

test_aot_compile.py

[precompile] Add option to disable guard check on aot-compiled function. (#163432)

2025-09-30 16:10:15 +00:00

test_autograd_function.py

Enable XPU for test_autograd_function.py (#160309)

2025-08-13 06:38:34 +00:00

test_backends.py

[Code Clean] Remove deadcodes about Python3.9 [2/N] (#163627)

2025-09-24 07:30:50 +00:00

test_backward_higher_order_ops.py

[ca] Functionalize AccumulateGrad (#155521)

2025-06-16 18:45:02 +00:00

test_base_hop.py

Fix requires_cuda to requires_cuda_and_triton (#160222)

2025-08-10 07:05:52 +00:00

test_base_output.py

[BE][Easy][11/19] enforce style for empty lines in import segments in test/dy*/ (#129762)

2024-07-27 17:43:53 +00:00

test_buffers_override.py

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

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

test_bytecode_utils.py

[Code Clean] Remove deadcodes about Python3.9 [8/N] (#163728)

2025-09-25 05:12:46 +00:00

test_callback.py

Use dataclass features in two classes (#164221)

2025-10-01 03:20:39 +00:00

test_compile.py

dynamo: Handle objects in graph that do not support weakref (#163168)

2025-09-22 22:11:09 +00:00

test_compiler_bisector.py

[inductor] teach bisector to look at pre_grad passes (#164250)

2025-10-02 05:42:18 +00:00

test_comptime.py

[aotd] Support saved tensors hooks in aot_autograd (#150032)

2025-05-22 14:09:38 +00:00

test_config.py

Rename cache limit to recompile limit in configs (#143709)

2024-12-22 10:03:57 +00:00

test_ctx_manager.py

[dynamo] Fix graph break on calling functions decorated with special context manager (#160703)

2025-08-18 20:33:45 +00:00

test_cudagraphs_expandable_segments.py

Revert "Use absolute path path.resolve() -> path.absolute() (#129409)"

2025-01-04 14:17:20 +00:00

test_cudagraphs.py

Fix unused Python variables in test/[a-d]* (#134665)

2024-12-13 22:13:12 +00:00

test_debug_utils.py

Fix requires_cuda to requires_cuda_and_triton (#160222)

2025-08-10 07:05:52 +00:00

test_decorators.py

[dynamo] change error_on_graph_break/fullgraph semantics (#161747)

2025-09-04 17:10:17 +00:00

test_deque_reconstruct.py

[dynamo] allow global import from collections import deque in user code (#148676)

2025-03-10 13:14:05 +00:00

test_deviceguard.py

[BE][Easy][11/19] enforce style for empty lines in import segments in test/dy*/ (#129762)

2024-07-27 17:43:53 +00:00

test_dicts.py

[dynamo] Correctly track mutation class source for MutableMappingVariable (#161568)

2025-08-27 21:47:17 +00:00

test_dynamic_shapes.py

fix dynamic float when dynamic=True (#149564)

2025-03-22 05:58:59 +00:00

test_einops.py

Add einops x torch.compile testing in PyTorch CI (#157416)

2025-07-03 17:36:39 +00:00

test_error_messages.py

[dynamo, 3.14] fix context managers (#163109)

2025-09-30 17:42:03 +00:00

test_exc.py

[dynamo] update pytorch-labs -> meta-pytorch in graph break URLs (#159975)

2025-08-06 23:57:31 +00:00

test_exceptions.py

[dynamo] Fix handling of kwargs in exception constructor (#163390)

2025-09-24 22:44:14 +00:00

test_export_mutations.py

[Dynamo] Replace torch._dynamo.optimize() with torch.compile() [6/N] (#140688)

2024-11-18 04:09:09 +00:00

test_export.py

[BE][5/6] fix typos in test/ (test/dynamo/) (#157639)

2025-07-06 06:34:25 +00:00

test_fake_distributed.py

[dynamo][device_mesh] Support mesh_dim_names (#164200)

2025-09-30 07:16:28 +00:00

test_flat_apply.py

[export] Add meta[val] to getattr nodes (#154934)

2025-06-13 05:48:21 +00:00

test_frame_init.py

Use dataclass features in two classes (#164221)

2025-10-01 03:20:39 +00:00

test_functions.py

[Code Clean] Remove deadcodes about Python3.9 [8/N] (#163728)

2025-09-25 05:12:46 +00:00

test_fx_graph_runnable.py

Make user defined Triton kernels serializable for fx_graph_runnable (#160002)

2025-08-11 20:54:33 +00:00

test_fx_passes_pre_grad.py

[Codemod][AddExplicitStrictExportArg] caffe2/test (#143688)

2024-12-27 07:58:44 +00:00

test_generator.py

[generator] Raise StopIteration(value) with value from the return stmt (#157152)

2025-08-15 01:42:40 +00:00

test_global.py

[Dynamo] Replace torch._dynamo.optimize() with torch.compile() [2/N] (#140238)

2024-11-13 05:13:39 +00:00

test_graph_deduplication.py

[Dynamo][Hierarchical Compile] Flatten tuple inputs for regions (#158812)

2025-08-18 18:03:11 +00:00

test_graph_region_tracker.py

[ROCm] Remove HIPBLASLT_ALLOW_TF32 from codebase (#162998)

2025-09-18 13:53:48 +00:00

test_guard_manager.py

Revert "[dynamo][guards] Do not construct entire framelocals dict for LAMBDA_GUARD (#162525)"

2025-09-19 06:15:28 +00:00

test_guard_serialization.py

Use dataclass features in two classes (#164221)

2025-10-01 03:20:39 +00:00

test_higher_order_ops.py

graph break on tolist if capture_scalar_outputs is false (#163807)

2025-09-28 04:02:52 +00:00

test_hooks.py

[BE][5/6] fix typos in test/ (test/dynamo/) (#157639)

2025-07-06 06:34:25 +00:00

test_inline_and_install.py

[dynamo][export] Make the source_stack and fqn info same between dynamo and export (#164085)

2025-09-29 04:35:13 +00:00

test_input_attr_tracking.py

[BE]: Update ruff to 0.11.8 (#153249)

2025-05-12 18:30:52 +00:00

test_install_free_tensors.py

Add functionality for installing free variables (#151134)

2025-04-24 17:57:54 +00:00

test_interop.py

[dynamo] Properly handle torch.script.jit under @staticmethod (#153984)

2025-05-21 19:45:06 +00:00

test_list.py

[list] Add list.__delitem__ (#156339)

2025-07-07 14:51:32 +00:00

test_logging.py

Adjust ...mark_unbacked() -> ...decorators.mark_unbacked() in logs. (#164131)

2025-09-29 17:44:00 +00:00

test_metrics_context.py

[BE][5/6] fix typos in test/ (test/dynamo/) (#157639)

2025-07-06 06:34:25 +00:00

test_minifier.py

Port two dynamo test cases for Intel GPU (#156056)

2025-06-19 12:49:04 +00:00

test_misc.py

[dynamo][export] Make the source_stack and fqn info same between dynamo and export (#164085)

2025-09-29 04:35:13 +00:00

test_model_output.py

Port three dynamo test to Intel GPU (#156575)

2025-06-27 05:56:22 +00:00

test_modes.py

[dynamo] Fix TorchFunctionMode handling with get_rng_state (#163412)

2025-09-23 15:37:45 +00:00

test_modules.py

[dynamo] Trace nn.Module __delattr__ (#159969)

2025-08-06 23:43:19 +00:00

test_nested_graph_breaks.py

[dynamo, nested graph breaks] support nested graph breaks that cause skipped frames (#160470)

2025-09-03 02:47:07 +00:00

test_nops.py

[BE][Easy][11/19] enforce style for empty lines in import segments in test/dy*/ (#129762)

2024-07-27 17:43:53 +00:00

test_optimizers.py

[BE][PYFMT] migrate PYFMT for test/[a-h]*/ to ruff format (#144555)

2025-06-24 04:53:54 +00:00

test_package.py

Simplify PrecompileContext to no longer be a CacheArtifactManager (#162886)

2025-09-20 01:24:37 +00:00

test_pgo.py

[PGO] ignore extra PGO key if warm/cold cache present (#163810)

2025-09-25 07:16:05 +00:00

test_pre_dispatch.py

Replace clone.detach with detach.clone (#140264)

2024-11-13 07:01:02 +00:00

test_precompile_context.py

Simplify PrecompileContext to no longer be a CacheArtifactManager (#162886)

2025-09-20 01:24:37 +00:00

test_profiler.py

[dynamo][export] Do not graph break on torch.autograd._profiler_enabled for export (#164418)

2025-10-02 09:00:00 +00:00

test_python_autograd.py

Apply ruff fixes to tests (#146140)

2025-02-04 05:41:01 +00:00

test_python_dispatcher.py

port 4 dynamo test files to Intel GPU (#157779)

2025-07-11 10:11:49 +00:00

test_recompile_ux.py

port 4 dynamo test files to Intel GPU (#157779)

2025-07-11 10:11:49 +00:00

test_recompiles.py

Add logging for when inbuilt_inline_nn_modules will help with ID_MATCH guard triggered recompiles (#160592)

2025-08-15 17:09:39 +00:00

test_reconstruct.py

[Linter] Expanding the scope of detecting device-bias code. (#159949)

2025-08-09 09:41:16 +00:00

test_reorder_logs.py

[dynamo] update pytorch-labs -> meta-pytorch in graph break URLs (#159975)

2025-08-06 23:57:31 +00:00

test_repros.py

[Bugfix] Match eager stride semantics for cloned tensors with preserve_format in compile (#163017)

2025-09-19 19:41:33 +00:00

test_resume.py

Fix unused Python variables in test/[a-d]* (#134665)

2024-12-13 22:13:12 +00:00

test_sdpa.py

SDP Backend function fix (#161169)

2025-09-19 20:19:59 +00:00

test_sets.py

Capture TypeError in CONTAINS_OP (#161069)

2025-09-04 04:49:09 +00:00

test_skip_guard_eval_unsafe.py

[dynamo][guards] Make class members go through obj.__class__.__dict__ (#159534)

2025-08-04 05:12:44 +00:00

test_skip_non_tensor.py

Deprecate torch._utils.is_compiling() (#127690)

2024-12-08 22:55:36 +00:00

test_sources.py

[Codemod][AddExplicitStrictExportArg] caffe2/test (#143688)

2024-12-27 07:58:44 +00:00

test_structured_trace.py

Add local file path to inductor_output_code trace metadata (#160920)

2025-09-18 18:39:46 +00:00

test_subclasses.py

[1/N] Apply UP035 rule in tests (#163947)

2025-09-29 01:42:01 +00:00

test_subgraphs.py

[BE][5/6] fix typos in test/ (test/dynamo/) (#157639)

2025-07-06 06:34:25 +00:00

test_torchrec.py

PEP585 update - test (#145176)

2025-01-22 04:48:28 +00:00

test_trace_rules.py

[BE][5/6] fix typos in test/ (test/dynamo/) (#157639)

2025-07-06 06:34:25 +00:00

test_unittest.py

Add infra to run CPython tests under Dynamo (#150787)

2025-05-07 04:03:14 +00:00

test_unspec.py

[fx][pass] Support converting a float32 tensor to a scalar in FX trace. (#158216)

2025-08-09 15:13:13 +00:00

test_utils.py

Log exception_stack_trace to dynamo_compile (#161096)

2025-08-22 03:29:15 +00:00

test_verify_correctness.py

[Dynamo] Replace torch._dynamo.optimize() with torch.compile() [2/N] (#140238)

2024-11-13 05:13:39 +00:00

test_view.py

Better fix for __index__ SymInt issue (#157201)

2025-07-01 07:06:46 +00:00

utils.py

[Dynamo] Introduce hook receiving list of traced code objects (#153622)

2025-05-28 15:40:09 +00:00