These kernels are adapted from https://github.com/NetEase-FuXi/EETQ/tree/main/csrc