| .. |
|
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
|
|
|
|
test_aot_autograd.py
|
[dynamo][guards] Delay DUPLICATE_INPUT guard because of incorrect ordering (#123605)
|
2024-04-10 07:30:02 +00:00 |
|
test_aot_autograd_cache.py
|
Setup initial testing harness and cache key generation for AOTAutograd Cache (#124642)
|
2024-04-30 18:17:38 +00:00 |
|
test_autograd_function.py
|
Dynamo x autograd.Function supports setup_context (#124802)
|
2024-04-27 04:57:13 +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
|
[dynamo] handle inactive nullcontexts across graph breaks (#125518)
|
2024-05-04 12:52:20 +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.py
|
switch tests from constrain_as* to torch._check* (#125253)
|
2024-05-01 21:01:27 +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_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] Bug fix for LOAD_GLOBAL and STORE_GLOBAL (#125002)
|
2024-04-28 15:24:17 +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] Differentiate dict guards wrt to guarding on key order (#124779)
|
2024-04-25 08:20:35 +00:00 |
|
test_higher_order_ops.py
|
[Dynamo] Add dynamo support to torch.func.linearize (#123118)
|
2024-04-23 21:31:49 +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
|
[dynamo] add trace_bytecode logging artifact (#125360)
|
2024-05-02 22:01:00 +00:00 |
|
test_minifier.py
|
Convert requires_cuda to full decorator (#118281)
|
2024-01-25 15:50:21 +00:00 |
|
test_misc.py
|
Revert "Don't call item() into torch.scalar_tensor uselessly (#125373)"
|
2024-05-04 22:22:36 +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
|
[guards][cpp-guards] Optimize NN module getattr guards (#124522)
|
2024-05-04 22:08:56 +00:00 |
|
test_nops.py
|
|
|
|
test_optimizers.py
|
Remove old optimizer tests (#120257)
|
2024-02-21 05:11:23 +00:00 |
|
test_pre_dispatch.py
|
|
|
|
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] Add support for DELETE_SUBSCR (#123526)
|
2024-04-25 22:07:24 +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]: TRY002 - Ban raising vanilla exceptions (#124570)
|
2024-04-21 22:26:40 +00:00 |
|
test_sources.py
|
Fix guard for SUPPORTED_NODES (#120798)
|
2024-03-01 16:03:21 +00:00 |
|
test_structured_trace.py
|
Hotfix: restore CPP guard string in structured trace (#125303)
|
2024-05-02 03:57:19 +00:00 |
|
test_subclasses.py
|
[Meta Tensor] fix meta inplace set storage (#123880)
|
2024-05-01 06:53:49 +00:00 |
|
test_subgraphs.py
|
Remove replace_all and make VTs mutable (#113725)
|
2023-12-10 09:31:21 +00:00 |
|
test_torchrec.py
|
|
|
|
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
|
|
|
a32ad828dc