pytorch

mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00

Files

Sherlock Huang e532f62e0d Introduce joint_custom_pass callback (#164981 )

```
        def joint_custom_pass(joint_gm: torch.fx.GraphModule, joint_inputs):
           # apply your pass for joint graph here

            return joint_gm

        class M(torch.nn.Module):
            def forward(self, x):
                return x.sin()

        x = torch.randn(10, requires_grad=False)
        compiled_fn = torch.compile(M(), backend="aot_eager")

        with torch._functorch.config.patch("joint_custom_pass", joint_custom_pass):
            out = compiled_fn(x)
```

Pull Request resolved: https://github.com/pytorch/pytorch/pull/164981
Approved by: https://github.com/ezyang, https://github.com/anijain2305

2025-10-09 04:40:54 +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

Introduce joint_custom_pass callback (#164981 )

2025-10-09 04:40:54 +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

[reland] Allow setting grad_dtype on leaf tensors (#164751 )

2025-10-08 20:23:13 +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

Use torch.testing.test_close instead of torch.testing.test_allclose (#164539 )

2025-10-03 14:39:10 +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, nested graph breaks] fix nested step graph break related issues (#162737 )

2025-10-08 22:02:52 +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

[export] Turn on install_free_tensors flag (#164691 )

2025-10-09 03:25:15 +00:00

test_export.py

[export] Turn on install_free_tensors flag (#164691 )

2025-10-09 03:25:15 +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_annotate.py

[dynamo] Support torch.fx.traceback.annotate (#164678 )

2025-10-08 22:41:00 +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

More ruff SIM fixes (#164695 )

2025-10-09 03:24:50 +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

[export] Turn on install_free_tensors flag (#164691 )

2025-10-09 03:25:15 +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

Fix refine_ranges corner case (#164075 ) (#164846 )

2025-10-08 18:42:37 +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] fix nested step graph break related issues (#162737 )

2025-10-08 22:02:52 +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

Support partial _DynamoCacheEntries when not all backends available (#163521 )

2025-10-03 16:14:32 +00:00

test_pgo.py

[PGO] log missing sources in allowlist (#164881 )

2025-10-09 04:39:09 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

More ruff SIM fixes (#164695 )

2025-10-09 03:24:50 +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

[export][dynamo] Fallback to slowpath for MultiHeadAttention for strict export (#164721 )

2025-10-09 03:25:15 +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

Revert "Enable all SIM rules except disabled ones (#164645 )"

2025-10-05 19:32:21 +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

use check_size instead of check_is_size in ops.py (#164668 )

2025-10-08 14:23:38 +00:00

utils.py

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

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