ziplab / PTQD

The official implementation of PTQD: Accurate Post-Training Quantization for Diffusion Models

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

CUTLASS Implementation of Convolution

ugonfor opened this issue · comments

Hi,
Thank you for releasing an awesome work!

I'm trying to reproduce the below table. But, I'm facing a problem to implement the conv2d with CUTLASS. Could you please release the code or guide about conv2d CUTLASS implementation?
image