Custom backward pass?

Open vgoklani opened this issue 2 weeks ago • 3 comments

Does cuTile support a custom backward pass? It wasn't clear from the examples. This is obviously needed to training and something that triton supports. thanks!

Dec 09 '25 03:12 vgoklani

Hi @vgoklani, you can find a backward kernel example here.

Dec 09 '25 18:12 haijieg

thanks @haijieg do you have one for flash attention?

Dec 09 '25 18:12 vgoklani

@vgoklani no, we don't. Please feel free to file a feature request and this will be a good one to get community contribution on.

Dec 10 '25 03:12 haijieg