wy17646051/tensorrt_scatter

This is stars and forks stats for /wy17646051/tensorrt_scatter repository. As of 28 Apr, 2024 this repository has 88 stars and 17 forks.

TensorRT Scatter TensorRT Plugin of corresponding PyTorch Scatter operators. At present, the project is only tested on TensorRT 8.5.x and CUDA 11.6, this does not mean that other versions cannot run, but it should be used with caution. Supporting Operators TensorRT Version CUDA Version scatter (sum, add, mean, mul, min, max) 8.5.x 11.6 segment_coo (sum, add, mean, min, max) 8.5.x 11.6 gather_coo 8.5.x 11.6 segment_csr (sum, add, mean, min, max) 8.5.x 11.6 gather_csr 8.5.x 11.6 Installation Before...