lightning-thunder
lightning-thunder copied to clipboard
Add cutlass python dsl executor for `quack-kernels`
What does this PR do?
As per title, this adds cutlass python dsl executor. In this PR, the kernels defined in https://github.com/Dao-AILab/quack, except matmul, are registered. Also, backward is not integrated.