Aaron Gokaslan 29cc293725 [BE]: FURB142 - Remove set mutations. Use set update (#124551) ...

Uses set mutation methods instead of manually reimplementing (update, set_difference etc).

Pull Request resolved: https://github.com/pytorch/pytorch/pull/124551
Approved by: https://github.com/ezyang

2024-04-21 14:12:33 +00:00

..

mock_modules

…

__init__.py

…

test_activation_checkpointing.py

[dynamo][easy] Add AC test and improve graph break message (#121394)

2024-04-06 01:02:45 +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

[dynamo][guards] Delay DUPLICATE_INPUT guard because of incorrect ordering (#123605)

2024-04-10 07:30:02 +00:00

test_autograd_function.py

[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)

2024-04-17 19:29:34 +00:00

test_backends.py

LazyGraphModule (#117911)

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

test_backward_higher_order_ops.py

[compiled autograd][dynamo] Make compiled graph take in boxed inputs (#122353)

2024-04-12 10:29:09 +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

[dynamo][guards] Move backend match to eval_frame (#121954)

2024-03-17 06:52:10 +00:00

test_config.py

Enable local_partial_types (#118467)

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

test_cpp_guard_manager.py

[dynamo][cpp-guards] Enable a few tests to prevent frequent regressions (#123059)

2024-04-01 21:09:41 +00:00

test_ctx_manager.py

[Compile FSDP2][2/n] Support streams created outside of compile region (#123487)

2024-04-06 08:42:42 +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

[dynamo][nn_module] Enable torch.compile/disable as decorators on the class (#124187)

2024-04-18 02:51:30 +00:00

test_deviceguard.py

[Inductor] Add a device agnostic DeviceGuard class to inductor (#123338)

2024-04-12 18:21:27 +00:00

test_dynamic_shapes.py

[dynamo] use proxies to nn.Module in dynamo generated GraphModules (#120756)

2024-03-21 21:23:12 +00:00

test_exc.py

[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)

2024-04-17 19:29:34 +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] allow Dim(1,2) for export dynamic shapes (v2 after revert) (#121910)

2024-03-19 19:08:05 +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

[Dynamo] Fix NamedTuple hasattr bug (#124531)

2024-04-21 04:36:22 +00:00

test_fx_passes_pre_grad.py

Add inductor fx pass unit test for shape propagation (#122897)

2024-03-29 16:44:22 +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][cpp-guards] Improve when to use Dict vs DictSubclassGuardManager (#124237)

2024-04-18 03:33:37 +00:00

test_higher_order_ops.py

[BE]: FURB142 - Remove set mutations. Use set update (#124551)

2024-04-21 14:12:33 +00:00

test_hooks.py

[dynamo] Optimize BuiltinVariable (#122055)

2024-03-19 04:23:20 +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

Enable capture_func_transforms by default (#122211)

2024-04-05 03:29:11 +00:00

test_logging.py

[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)

2024-04-17 19:29:34 +00:00

test_minifier.py

Convert requires_cuda to full decorator (#118281)

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

test_misc.py

[dynamo, 3.8-3.9] support dataclass with frozen=True in Python 3.8/3.9 (#124393)

2024-04-19 05:10:33 +00:00

test_model_output.py

[dynamo][model_output] Do not include none for CustomizedDictVariable (#122005)

2024-03-19 02:35:48 +00:00

test_modules.py

[Compile FSDP2][1/n] Support using user-defined object instance method as hook (#123399)

2024-04-09 17:29:08 +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_repros.py

[dynamo] Support list.reverse (#124210)

2024-04-17 23:33:32 +00:00

test_sdpa.py

Add support for compiling SDPAParams (#117207)

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

test_skip_non_tensor.py

[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)

2024-04-17 19:29:34 +00:00

test_sources.py

Fix guard for SUPPORTED_NODES (#120798)

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

test_structured_trace.py

[BE] enable ruff rule RSE and remove useless parentheses in raise statements (#124261)

2024-04-17 19:29:34 +00:00

test_subclasses.py

[NJT] Inline through torch.nested.nested_tensor_from_jagged instead of graph break (#124343)

2024-04-19 23:13:59 +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

[dynamo][pt2d] avoid skipping modules from torch/testing/_internal (#122851)

2024-03-29 06:42:06 +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