| .. |
|
mock_modules
|
|
|
|
__init__.py
|
|
|
|
test_activation_checkpointing.py
|
[cuDNN][SDPA] Remove TORCH_CUDNN_SDPA_ENABLED=1, enable cuDNN SDPA by default on H100 and 2nd on other archs >= sm80 (#125343)
|
2024-06-30 19:22:16 +00:00 |
|
test_after_aot.py
|
|
|
|
test_aot_autograd.py
|
|
|
|
test_aot_autograd_cache.py
|
Reduce all tensors to their metadata in AOTAutogradCache; add tests (#128583)
|
2024-07-11 15:39:09 +00:00 |
|
test_autograd_function.py
|
Revert "Fix names conflict when lifting (#129817)"
|
2024-07-15 22:08:45 +00:00 |
|
test_backends.py
|
|
|
|
test_backward_higher_order_ops.py
|
Stop immediately specializing common constants 0/1 for plain int (#128327)
|
2024-07-03 16:41:51 +00:00 |
|
test_base_output.py
|
|
|
|
test_bytecode_utils.py
|
[3.13, dynamo] codegen PUSH_NULL when callable is codegen'd (#129172)
|
2024-06-22 17:25:23 +00:00 |
|
test_compile.py
|
|
|
|
test_comptime.py
|
|
|
|
test_config.py
|
|
|
|
test_ctx_manager.py
|
|
|
|
test_cudagraphs.py
|
|
|
|
test_cudagraphs_expandable_segments.py
|
Support for expandable segments with cuda graph trees (#128068)
|
2024-07-15 23:23:23 +00:00 |
|
test_debug_utils.py
|
|
|
|
test_decorators.py
|
Update mark_static_address for inlining NN modules (#130392)
|
2024-07-16 00:25:29 +00:00 |
|
test_deviceguard.py
|
|
|
|
test_dynamic_shapes.py
|
|
|
|
test_exc.py
|
Fix typo in floordiv solver code that affects flipped relation (#129888)
|
2024-07-03 04:47:32 +00:00 |
|
test_exceptions.py
|
|
|
|
test_export.py
|
Revert "Fix names conflict when lifting (#129817)"
|
2024-07-15 22:08:45 +00:00 |
|
test_export_mutations.py
|
|
|
|
test_frame_init.py
|
Enhancements to recompiles logs (#130043)
|
2024-07-09 03:40:56 +00:00 |
|
test_functions.py
|
dynamo add support for defaultdict(set) (#130745)
|
2024-07-15 22:23:33 +00:00 |
|
test_fx_passes_pre_grad.py
|
|
|
|
test_global.py
|
|
|
|
test_guard_manager.py
|
[BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199)
|
2024-07-11 17:30:28 +00:00 |
|
test_higher_order_ops.py
|
Revert "Fix names conflict when lifting (#129817)"
|
2024-07-15 22:08:45 +00:00 |
|
test_hooks.py
|
|
|
|
test_inline_inbuilt_nn_modules.py
|
[Reland][dynamo] Enable some inlining inbuilt nn module tests (#128440)
|
2024-06-13 22:39:22 +00:00 |
|
test_input_attr_tracking.py
|
|
|
|
test_interop.py
|
|
|
|
test_logging.py
|
Add trace_shape_events artifact tracing for ShapeEnv events (#130473)
|
2024-07-12 13:50:25 +00:00 |
|
test_minifier.py
|
|
|
|
test_misc.py
|
[export] clean up allow_complex_guards_as_runtime_asserts flag (#130596)
|
2024-07-12 17:17:11 +00:00 |
|
test_model_output.py
|
[3.13, dynamo] codegen PUSH_NULL when callable is codegen'd (#129172)
|
2024-06-22 17:25:23 +00:00 |
|
test_modules.py
|
Mark nn_module params and buffers as static in dynamo (#130391)
|
2024-07-16 00:25:23 +00:00 |
|
test_nops.py
|
|
|
|
test_optimizers.py
|
|
|
|
test_pre_dispatch.py
|
|
|
|
test_profiler.py
|
|
|
|
test_python_autograd.py
|
|
|
|
test_recompile_ux.py
|
|
|
|
test_recompiles.py
|
|
|
|
test_reorder_logs.py
|
|
|
|
test_repros.py
|
[dynamo] update call map to allow multiple input parameters (#130748)
|
2024-07-15 22:16:49 +00:00 |
|
test_resume.py
|
|
|
|
test_sdpa.py
|
|
|
|
test_skip_non_tensor.py
|
|
|
|
test_sources.py
|
|
|
|
test_structured_trace.py
|
[easy][inline-inbuilt-nn-module] Update test output (#130681)
|
2024-07-15 06:19:53 +00:00 |
|
test_subclasses.py
|
Propagate buffer and parameter indices through AOT (#130393)
|
2024-07-16 00:25:38 +00:00 |
|
test_subgraphs.py
|
[runtime asserts] deduplicate runtime asserts & CSE (#128599) (#130380)
|
2024-07-10 19:23:37 +00:00 |
|
test_torchrec.py
|
|
|
|
test_trace_rules.py
|
[BE][Easy] apply autofix for ruff rules unnecessary-collection-call (C408): list() / tuple() / dict() (#130199)
|
2024-07-11 17:30:28 +00:00 |
|
test_unspec.py
|
Do not set hints for mark_unbacked quantities (#130483)
|
2024-07-11 15:51:00 +00:00 |
|
test_utils.py
|
[pt2-bench] fix accuracy failure for a few models (#129941)
|
2024-07-05 10:26:39 +00:00 |
|
test_verify_correctness.py
|
|
|
|
test_view.py
|
|
|
|
utils.py
|
|
|