PyTorch Custom Operation

 (leimao.github.io)

16 points | by eigenBasis 5 days ago

2 comments

  • Evidlo 1 hour ago
    Aren't there other options for custom PyTorch operators before going so low-level as CUDA C++, like using Jax or CuPy?
  • dayeye2006 2 hours ago
    neat -- how to add pt2 support and how to allow fusion of your custom ops with normal ops by inductor?