Simon Fan 124dd364e9 [hop] support local_map + SAC (#163322) ...

Some ops like local_map hop's deferred mode are not desugared by make_fx, this means that when we apply SAC tags, we will need to define dispatch rules for the SAC torch dispatch modes as pointed out here: https://github.com/pytorch/pytorch/issues/162246#issuecomment-3259176721. This PR adds those rules.

Additionally it fixes a pre-existing issue where we weren't coercing tangent layout (that AOTAutograd typically does) when partitioning the HOP joint.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/163322
Approved by: https://github.com/ezyang

2025-09-24 04:57:40 +00:00

..

test_invoke_quant.py

[BE][3/6] fix typos in test/ (#157637)

2025-07-17 12:08:33 +00:00

test_invoke_subgraph.py

[dynamo][hops] Remove const outputs from the speculated subgraph (#161355)

2025-09-04 18:52:01 +00:00

test_local_map.py

[hop] support local_map + SAC (#163322)

2025-09-24 04:57:40 +00:00

test_with_effects.py

Using the latest torch.library.register_fake API instead of torch.library.impl_abstract (#158839)

2025-07-25 02:37:30 +00:00