jax
f600caa2
- Use @register_lowering to register Pallas GPU lowering rules
Go
Login via GitHub
Home
Pricing
FAQ
Install
Login
via GitHub
Commit
View On
GitHub
Commit
1 year ago
Use @register_lowering to register Pallas GPU lowering rules This leads to slightly more compact code, but should otherwise be identical. PiperOrigin-RevId: 635442002
Author
superbobry
Committer
a-googler
Parents
c3d0b0d1
Loading