GitHub repository: lxxue/prefix_sum
Primary language: Cuda
Description: A PyTorch wrapper of parallel exclusive scan in CUDA