William Wen d14d62b7aa [dynamo] add more refleak tests (#120657) ...

Pull Request resolved: https://github.com/pytorch/pytorch/pull/120657
Approved by: https://github.com/jansel

2024-03-07 22:25:43 +00:00

..

mock_modules

…

__init__.py

…

test_activation_checkpointing.py

Add linux cpu test for 3.12 (#117853)

2024-02-14 20:52:23 +00:00

test_after_aot.py

Preserve leaf-ness and requires_grad-ness in minified repros (#102899)

2023-06-05 19:56:00 +00:00

test_aot_autograd.py

Min-cut partitioner always saves tensors that are returned as-is in backward (#114970)

2024-02-13 00:04:41 +00:00

test_autograd_function.py

dynamo: respect autograd.Function + multiple save_for_backward calls (#117667)

2024-02-16 21:16:07 +00:00

test_backends.py

LazyGraphModule (#117911)

2024-01-27 04:10:18 +00:00

test_backward_higher_order_ops.py

[Compiled Autograd] Introduce BackwardState capture (#120382)

2024-02-28 20:36:47 +00:00

test_base_output.py

Support 'BaseOutput' and subclasses from 'diffusers' in dynamo (#111978)

2023-10-30 19:53:31 +00:00

test_bytecode_hook.py

add testcase when bytecode hook changes the bytecode; fix code map (#114487)

2023-11-28 22:14:57 +00:00

test_compile.py

[torch.compile] Provide capability to register callback on compile start/stop (#120764)

2024-02-29 07:37:52 +00:00

test_comptime.py

Revert "[dynamo] guarded config (#111299)" (#115386)

2023-12-11 19:35:42 +00:00

test_config.py

Enable local_partial_types (#118467)

2024-01-28 13:38:22 +00:00

test_ctx_manager.py

[dynamo] support comparing stream with constant (#119199)

2024-02-06 19:26:03 +00:00

test_cudagraphs.py

[ROCm] Enabling additional UTs on ROCm (#115738)

2024-01-09 08:36:07 +00:00

test_debug_utils.py

Extend aot_graph_input_parser to sym shapes (#120246)

2024-02-20 23:24:45 +00:00

test_decorators.py

Update to TorchFix 0.4.0 (#119424)

2024-02-12 23:30:12 +00:00

test_dynamic_shapes.py

[dynamo] add more refleak tests (#120657)

2024-03-07 22:25:43 +00:00

test_exc.py

Avoid performing replacements when it would unrefine ranges (#117356)

2024-02-13 15:56:59 +00:00

test_export_mutations.py

stop using torch.Tensor in dynamo/test_export_mutations.py (#118287)

2024-01-25 22:21:41 +00:00

test_export.py

[export][reland] Disable exported_program.__call__ (#120019)

2024-03-05 05:29:46 +00:00

test_frame_init.py

[Dynamo] Look up variables from old frame, rather than copy variables to new frame; skip some copy to save time. (#115062)

2023-12-16 00:02:59 +00:00

test_functions.py

Do not wrap output with input device inside _to_copy (#119868)

2024-02-28 01:51:43 +00:00

test_global.py

Revert "[dynamo] Refactor test cross importing (#113242)"

2023-11-15 01:53:07 +00:00

test_guard_manager.py

[dynamo][guards-cpp-refactor] Pass source name for debug ease (#121154)

2024-03-06 08:36:45 +00:00

test_higher_order_ops.py

Add jacrev support in torch.compile (#121146)

2024-03-06 16:05:33 +00:00

test_hooks.py

[Compiled Autograd] Introduce BackwardState capture (#120382)

2024-02-28 20:36:47 +00:00

test_input_attr_tracking.py

[dynamo] handle setting .data on a tensor (#113080)

2023-12-02 00:35:44 +00:00

test_interop.py

Set _dynamo.config.capture_func_transforms=False (#115267)

2023-12-07 18:42:15 +00:00

test_logging.py

Change default TORCH_LOGS format to match Meta/glog standard (#119869)

2024-02-14 18:56:35 +00:00

test_minifier.py

Convert requires_cuda to full decorator (#118281)

2024-01-25 15:50:21 +00:00

test_misc.py

[dynamo] add more refleak tests (#120657)

2024-03-07 22:25:43 +00:00

test_model_output.py

[dynamo] implement custom dict variable as a general solution for HF's ModelOutput class (#105044)

2023-09-14 17:15:50 +00:00

test_modules.py

[dynamo][refactor] Use existing helper functions for CLOSURE_MATCH (#120145)

2024-02-18 00:31:36 +00:00

test_nops.py

Add hooks structure for passing around user provided hooks, add a new guard_failure_fn (#90371)

2022-12-07 17:51:53 +00:00

test_optimizers.py

Remove old optimizer tests (#120257)

2024-02-21 05:11:23 +00:00

test_pre_dispatch.py

pre_dispatch tracing: support autocast and no_grad/enable_grad ctx managers, add a pre_dispatch_eager dynamo backend (#103024)

2023-06-29 14:17:42 +00:00

test_profiler.py

[execution trace] ignore some properties when symbolic size/strides exist (#112458)

2023-10-31 18:13:03 +00:00

test_python_autograd.py

Fix typo under test directory (#112346)

2023-11-03 07:53:33 +00:00

test_recompile_ux.py

[dynamo] Extend LazyVariableTracker to tuples (#117426)

2024-01-18 15:51:28 +00:00

test_recompiles.py

Revert "[dynamo] guarded config (#111299)" (#115386)

2023-12-11 19:35:42 +00:00

test_reorder_logs.py

[dynamo] Reorder logs (#116106)

2024-03-01 17:04:24 +00:00

test_replay_record.py

[dynamo] Test for using torch.nn when replay_records are enabled (#116215)

2024-01-24 23:42:35 +00:00

test_repros.py

Don't aggressively rewrite asserts for symbolic expressions (#120564)

2024-03-01 17:46:36 +00:00

test_sdpa.py

Add support for compiling SDPAParams (#117207)

2024-01-19 05:51:15 +00:00

test_skip_non_tensor.py

feat(dynamo): remove inconsistent tracing histories by acknowledging possibility of inconsistent side-effects (#110804)

2023-10-25 19:27:11 +00:00

test_sources.py

Fix guard for SUPPORTED_NODES (#120798)

2024-03-01 16:03:21 +00:00

test_structured_trace.py

Test that tlparse understands the structured logs we output (#120658)

2024-02-28 21:58:39 +00:00

test_subclasses.py

Subclass view fake-ification via reified ViewFuncs (#118405)

2024-03-07 19:56:16 +00:00

test_subgraphs.py

Remove replace_all and make VTs mutable (#113725)

2023-12-10 09:31:21 +00:00

test_torchrec.py

[RELAND] Force synced KJT to trace unbacked SymInt (#108960) (#109216)

2023-09-18 14:39:44 +00:00

test_trace_rules.py

Avoid COW materialization in at::parallel_for/parallel_reduce (#120455)

2024-03-01 05:05:28 +00:00

test_unspec.py

SymIntify prod_backward (#120776)

2024-02-29 20:05:22 +00:00

test_verify_correctness.py

LazyGraphModule (#117911)

2024-01-27 04:10:18 +00:00

utils.py

[2/N] Dynamo supports skip by function & removes skipfiles circular import (#110835)

2023-10-12 00:44:41 +00:00