| .. |
|
mock_modules
|
|
|
|
__init__.py
|
|
|
|
test_activation_checkpointing.py
|
[dynamo][higher order op] Fix minor bug in error msgs (#110099)
|
2023-09-27 20:28:17 +00:00 |
|
test_after_aot.py
|
|
|
|
test_aot_autograd.py
|
Reland attempt 2 of "Update AOTAutograd to use FunctionalTensorMode instead of C++ functionalization (#106406)" (#109906)" (#110079)
|
2023-10-03 18:50:25 +00:00 |
|
test_autograd_function.py
|
Back out "Dynamo support for autograd.Function w/ once_differentiable (#108686)" (#109199)
|
2023-09-13 15:43:20 +00:00 |
|
test_backends.py
|
|
|
|
test_backward_higher_order_ops.py
|
Higher order op for preserving leaf functions through trace, particularly for getting user defined hooks to compiled autograd (#109690)
|
2023-09-27 20:47:15 +00:00 |
|
test_compile.py
|
|
|
|
test_comptime.py
|
[dynamo] Add BACKEND_MATCH guard to detect and recompile when backend changes (#107337)
|
2023-09-14 15:49:30 +00:00 |
|
test_config.py
|
|
|
|
test_ctx_manager.py
|
Flash Attention v2 (#105602)
|
2023-09-13 13:59:05 +00:00 |
|
test_cudagraphs.py
|
|
|
|
test_debug_utils.py
|
[dynamo] Convert dtype arguments as well as inputs in cast_to_fp64 (#110232)
|
2023-09-29 12:42:14 +00:00 |
|
test_decorators.py
|
Add API to mark input tensors static for cudagraphs (#107154)
|
2023-08-16 04:38:19 +00:00 |
|
test_dynamic_shapes.py
|
Record and replay for ShapeEnv. (#107989)
|
2023-09-13 00:22:38 +00:00 |
|
test_exc.py
|
[New][1/N] Dynamo skipfiles refactor (#110330)
|
2023-10-03 16:50:33 +00:00 |
|
test_export_mutations.py
|
|
|
|
test_export.py
|
Move export.constrain_as_* to torch._constrain_as_* (#110757)
|
2023-10-11 02:37:55 +00:00 |
|
test_functions.py
|
Add a side table for triton kernels to avoid using itertools.partial (#110633)
|
2023-10-08 02:01:59 +00:00 |
|
test_global.py
|
|
|
|
test_higher_order_ops.py
|
[HigherOrderOp] fallthrough some keys by default. (#110478)
|
2023-10-05 16:25:42 +00:00 |
|
test_hooks.py
|
Support a simple subset of functions as backward hooks on intermediate tensors (#109537)
|
2023-10-11 01:35:37 +00:00 |
|
test_input_attr_tracking.py
|
Fix typo under test directory (#110826)
|
2023-10-08 20:52:38 +00:00 |
|
test_interop.py
|
|
|
|
test_logging.py
|
[dynamo, logging] Report name of defining class along side function name in Dynamo logs (#110190)
|
2023-10-05 20:41:38 +00:00 |
|
test_minifier.py
|
Use user directed names for variables where possible (#109092)
|
2023-09-13 07:44:04 +00:00 |
|
test_misc.py
|
Move export.constrain_as_* to torch._constrain_as_* (#110757)
|
2023-10-11 02:37:55 +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
|
Revert "Reland 3rd try [finishing colesbury's PR 100642] Guard on nn.Module dicts and type (#109323)" + Forward fixes + test (#110964)
|
2023-10-11 05:16:47 +00:00 |
|
test_nops.py
|
|
|
|
test_optimizers.py
|
Cudagraphs support for compiled optimizers (#107504)
|
2023-08-31 20:47:18 +00:00 |
|
test_pre_dispatch.py
|
|
|
|
test_profiler.py
|
[dynamo] Add "Torch-Compiled Region" profiler event (#108462)
|
2023-09-08 02:10:17 +00:00 |
|
test_python_autograd.py
|
|
|
|
test_recompile_ux.py
|
|
|
|
test_recompiles.py
|
Revert "Reland 3rd try [finishing colesbury's PR 100642] Guard on nn.Module dicts and type (#109323)" + Forward fixes + test (#110964)
|
2023-10-11 05:16:47 +00:00 |
|
test_replay_record.py
|
|
|
|
test_repros.py
|
[dynamo] list index: add more list types to testing, support namedtuple, improve error handling (#110919)
|
2023-10-11 00:16:39 +00:00 |
|
test_skip_non_tensor.py
|
|
|
|
test_sources.py
|
|
|
|
test_subclasses.py
|
Make NestedTensor compilable with eager backend (#109171)
|
2023-10-11 04:47:10 +00:00 |
|
test_subgraphs.py
|
|
|
|
test_torchrec.py
|
[RELAND] Force synced KJT to trace unbacked SymInt (#108960) (#109216)
|
2023-09-18 14:39:44 +00:00 |
|
test_unspec.py
|
Implement traceable torch.tensor when you have SymInt/SymFloat inputs (#109515)
|
2023-09-19 13:19:57 +00:00 |
|
test_verify_correctness.py
|
|
|
|
utils.py
|
|
|
