| .. |
|
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
|
|
|
|
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
|
Change default torch_function behavior to be disabled when torch_dispatch is defined (take 2) (#120632)
|
2024-03-09 01:08:37 +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
|
Improve Dynamo support for torch function and class methods in general (#121365)
|
2024-03-08 20:03:49 +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] improve binary op fast path broadcast check (#121546)
|
2024-03-09 01:49:42 +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
|
Add operator length hint support (#121495)
|
2024-03-08 19:08:33 +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] Simplify DYNAMIC_INDICES guard (#121614)
|
2024-03-12 17:08:56 +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
|
feat: Add min, max ranges to mark_dynamic API (#119737)
|
2024-03-07 23:26:03 +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
|
|
|
|
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
|
Improve Dynamo support for torch function and class methods in general (#121365)
|
2024-03-08 20:03:49 +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
|
Disable test_torch_name_rule_map_updated in code (#120627)
|
2024-03-08 21:00:51 +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 |
4e63d9065a