PyTorch Custom Operation

 (leimao.github.io)

18 points | by eigenBasis 5 days ago

3 comments

  • Evidlo 2 hours ago
    Aren't there other options for custom PyTorch operators before going so low-level as CUDA C++, like using Jax or CuPy?
  • fooblaster 1 hour ago
    hmm, dynamo did not support torchbind last time I looked. when did support for torchbind get added? I need to know what version of torch to upgrade to.
  • dayeye2006 3 hours ago
    neat -- how to add pt2 support and how to allow fusion of your custom ops with normal ops by inductor?