-
Notifications
You must be signed in to change notification settings - Fork 24.8k
Open
Labels
module: higher order operatorstorch.cond and similartorch.cond and similarmodule: pt2-dispatcherPT2 dispatcher-related issues (e.g., aotdispatch, functionalization, faketensor, custom-op,PT2 dispatcher-related issues (e.g., aotdispatch, functionalization, faketensor, custom-op,oncall: pt2triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module
Description
- No dict proxies
- No dataclass proxies
The main motivation is that fx passes are really difficult to write when there are dict or dataclass operations in the graph. E.g. the dict lookup below

For dicts, this means we would rewrite the triton_kernel_wrapper* HOPs to not use dicts, or shepherd them to use flat_apply HOPs.
For dataclass, we would need to fix invoke_quant somehow.
Thoughts @oulgen, @aakhundov, @davidberard98, @eellison?
Metadata
Metadata
Assignees
Labels
module: higher order operatorstorch.cond and similartorch.cond and similarmodule: pt2-dispatcherPT2 dispatcher-related issues (e.g., aotdispatch, functionalization, faketensor, custom-op,PT2 dispatcher-related issues (e.g., aotdispatch, functionalization, faketensor, custom-op,oncall: pt2triagedThis issue has been looked at a team member, and triaged and prioritized into an appropriate moduleThis issue has been looked at a team member, and triaged and prioritized into an appropriate module