dmlc / cub
☆21Updated 7 years ago
Alternatives and similar repositories for cub:
Users that are interested in cub are comparing it to the libraries listed below
- Some CUDA design patterns and a bit of template magic for CUDA☆148Updated last year
- flexible-gemm conv of deepcore☆17Updated 5 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 5 years ago
- ☆45Updated 2 years ago
- Optimized half precision gemm assembly kernels (deprecated due to ROCm)☆47Updated 7 years ago
- Efficient Top-K implementation on the GPU☆151Updated 5 years ago
- kmeans clustering with multi-GPU capabilities☆117Updated last year
- BGHT: High-performance static GPU hash tables.☆57Updated 5 months ago
- a heterogeneous multiGPU level-3 BLAS library☆45Updated 5 years ago
- CNNs in Halide☆23Updated 9 years ago
- Example of how to use CUDA with CMake >= 3.8☆69Updated last year
- CUDA Tensor Transpose (cuTT) library☆51Updated 7 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- Connected Component Labeling.☆43Updated 4 years ago
- ☆22Updated 5 years ago
- A way to use cuda to accelerate top k algorithm☆29Updated 7 years ago
- ☆10Updated 4 years ago
- a c++/cuda template library for tensor lazy evaluation☆163Updated last year
- Fast CUDA Kernels for ResNet Inference.☆171Updated 5 years ago
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆83Updated 11 months ago
- CNN accelerated by cuda. Test on mnist and finilly get 99.76%☆185Updated 7 years ago
- Test winograd convolution written in TVM for CUDA and AMDGPU☆40Updated 6 years ago
- A warp-oriented dynamic hash table for GPUs☆72Updated last year
- An GPU/CUDA implementation of the Hungarian algorithm☆109Updated 5 years ago
- CVFusion is an open-source deep learning compiler to fuse the OpenCV operators.☆29Updated 2 years ago
- ☆42Updated 7 years ago
- Full-speed Array of Structures access☆164Updated last year
- TensorFlow and TVM integration☆37Updated 4 years ago
- ☆12Updated 6 years ago
- This is a tuned sparse matrix dense vector multiplication(SpMV) library☆21Updated 8 years ago