ViDiT-Q Extension CUDA Kernel Installation run python setup.py develop. Benchmark run ./bench/bench_xxx.py Acknowlegement Thanks @JasonHuang03 for the cuda kernel development