pytorch_grouped_gemm icon indicating copy to clipboard operation
pytorch_grouped_gemm copied to clipboard
verified publisher icon Karbo123

How to build pytorch with cutlass?

Open Qiyu8 opened this issue 3 years ago • 1 comments

I noticed that pytorch 1.13.1 is support cutlass, Do you know how to enable cutlass in pytorch?

Qiyu8 avatar Apr 07 '23 12:04 Qiyu8

Please checkout our new Python interface

We have an example creating a PyTorch extension

mnicely avatar Apr 27 '23 17:04 mnicely