cutlass
cutlass copied to clipboard
[QST] custom kernel integrated in Pytorch
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.