Custom backward pass?

[vgoklani]

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!

[haijieg]

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

[vgoklani]

thanks @haijieg do you have one for flash attention?

[haijieg]

@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.