xla
1bbe333a - [Pallas] Introduce make_kernel_from_pallas (#6713)

Commit
1 year ago
[Pallas] Introduce make_kernel_from_pallas (#6713) Summary: This pull request introduces make_kernel_from_pallas API which is the top level API to interact with the Pallas integration. It takes a pallas_call wrapper and than make it a custom pytorch op. Test Plan: python test/test_pallas.py
Author
Parents
Loading