Yep, that’s the key idea!
We do want to support selective “takeover” of entire operations that haven’t been converted to prims as an option, but by supporting just the prims you should enable all of PyTorch’s functionality.
Yep, that’s the key idea!
We do want to support selective “takeover” of entire operations that haven’t been converted to prims as an option, but by supporting just the prims you should enable all of PyTorch’s functionality.