IvanKobzarev 370c1c4297 [aotd] Fix rrelu compilation (#136008) ... Issues: https://github.com/pytorch/pytorch/issues/135083 https://github.com/pytorch/pytorch/issues/120292 rrelu decomposition contains mutation, copy_. Decompositions are executed below Functionalization, as a result AOT produces non-functional graph. Also that decomposition is registered as python_dispatch kernel for AutogradCUDA. Autograd dispatch happens above Functionalization, so registering it for Autograd to handle all backends makes functionalization running after this. Testing: ``` python test/functorch/test_aotdispatch.py -k test_rrelu ``` Pull Request resolved: https://github.com/pytorch/pytorch/pull/136008 Approved by: https://github.com/bdhirsh		2024-09-25 11:26:19 +00:00
..
attn_ft.py
attn_positional.py
common_utils.py
discover_coverage.py
functorch_additional_op_db.py
test_ac.py	[BE]: Simplify some list comps to generators C419 (#132578)	2024-08-04 17:46:26 +00:00
test_aotdispatch.py	[aotd] Fix rrelu compilation (#136008)	2024-09-25 11:26:19 +00:00
test_control_flow.py	[Dynamo] Trace torch function modes entered outside of torch.compile (#133137)	2024-09-14 18:52:22 +00:00
test_dims.py
test_eager_transforms.py	Revert expectFailureIf condition on tests with torch.compile on Windows (#134759)	2024-09-06 20:51:55 +00:00
test_logging.py
test_memory_efficient_fusion.py	Revert "Deprecate torch._utils.is_compiling() and torch._dynamo.external_utils.is_compiling() (#127690)"	2024-08-07 00:05:20 +00:00
test_minifier.py
test_ops.py	[cuDNN][conv][A100] Bump tolerances for vmap_autograd_grad conv2d on A100 (#136178)	2024-09-18 19:42:13 +00:00
test_parsing.py
test_rearrange.py
test_vmap.py	[TF32] Account for TF32 in test_conv_double_backward (#135716)	2024-09-21 01:06:22 +00:00
test_vmap_registrations.py	Add randomness checking for sdpa vmap (#135176)	2024-09-06 04:50:49 +00:00
xfail_suggester.py