cutlass icon indicating copy to clipboard operation
cutlass copied to clipboard

[QST] custom kernel integrated in Pytorch

Open IzanCatalan opened this issue 2 months ago • 2 comments

What is your question? Hello, I would like to know if, in addition to importing a custom kernel as detailed in this example, it is possible to import it into PyTorch and then reuse it not for a specific operation but to perform inference on a neural network model that uses these operations. This would be similar to what is described here, where there are pre-trained models in PyTorch, such as ResNet50, which includes up to 50 convolutions among other operations.

IzanCatalan avatar Dec 16 '24 17:12 IzanCatalan