Added (optional) debugging info to the partitioner (AOT_PARTITIONER_DEBUG=1) and add a bunch of ops that inductor supports

pytorch / functorch

functorch is JAX-like composable function transforms for PyTorch.

https://pytorch.org/functorch/

BSD 3-Clause "New" or "Revised" License

1.38k stars 102 forks source link

Open Chillee opened 1 year ago

Chillee commented 1 year ago

Will need to guard the new ops behind a flag before landing.