PyTorch Custom Operation

2 comments

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