pytorch/test/dynamo
2024-07-16 00:25:38 +00:00
..
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