dmlc / cub
☆22Updated 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☆150Updated last year
- flexible-gemm conv of deepcore☆17Updated 5 years ago
- Efficient Top-K implementation on the GPU☆175Updated 6 years ago
- Optimized half precision gemm assembly kernels (deprecated due to ROCm)☆47Updated 7 years ago
- kmeans clustering with multi-GPU capabilities☆120Updated 2 years ago
- A warp-oriented dynamic hash table for GPUs☆73Updated last year
- a heterogeneous multiGPU level-3 BLAS library☆45Updated 5 years ago
- a c++/cuda template library for tensor lazy evaluation☆163Updated last year
- A communication library for deep learning☆50Updated 8 months ago
- CNNs in Halide☆23Updated 9 years ago
- Connected Component Labeling.☆43Updated 4 years ago
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆84Updated last year
- ☆45Updated 2 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- Python bindings for NVTX☆66Updated last year
- CUDA Data Parallel Primitives Library☆429Updated 6 years ago
- ☆22Updated 5 years ago
- Subpart source code of of deepcore v0.7☆27Updated 4 years ago
- Full-speed Array of Structures access☆169Updated last year
- Fast CUDA Kernels for ResNet Inference.☆173Updated 5 years ago
- A simple memory manager for CUDA designed to help Deep Learning frameworks manage memory☆297Updated 6 years ago
- TensorFlow and TVM integration☆37Updated 4 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 6 years ago
- ☆67Updated 11 years ago
- Symbolic Expression and Statement Module for new DSLs☆205Updated 4 years ago
- CUDA Tensor Transpose (cuTT) library☆51Updated 7 years ago
- BGHT: High-performance static GPU hash tables.☆63Updated last week
- Codebase associated with the PyTorch compiler tutorial☆45Updated 5 years ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆65Updated 3 years ago
- An GPU/CUDA implementation of the Hungarian algorithm☆109Updated 6 years ago