PyTorch Custom Operation

 (leimao.github.io)

14 points | by eigenBasis 5 days ago

2 comments

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