PyTorch Custom Operation

18 pointsposted 5 days ago
by eigenBasis

3 Comments

Evidlo

3 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

an 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?