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☆149Updated last year
- Efficient Top-K implementation on the GPU☆168Updated 5 years ago
- flexible-gemm conv of deepcore☆17Updated 5 years ago
- A warp-oriented dynamic hash table for GPUs☆73Updated last year
- CNNs in Halide☆23Updated 9 years ago
- CUDA Tensor Transpose (cuTT) library☆51Updated 7 years ago
- BGHT: High-performance static GPU hash tables.☆62Updated 6 months ago
- Subpart source code of of deepcore v0.7☆27Updated 4 years ago
- Fast CUDA Kernels for ResNet Inference.☆173Updated 5 years ago
- a c++/cuda template library for tensor lazy evaluation☆163Updated last year
- Connected Component Labeling.☆43Updated 4 years ago
- kmeans clustering with multi-GPU capabilities☆120Updated last year
- Full-speed Array of Structures access☆167Updated last year
- ☆42Updated 7 years ago
- A way to use cuda to accelerate top k algorithm☆29Updated 7 years ago
- how to design cpu gemm on x86 with avx256, that can beat openblas.☆68Updated 5 years ago
- Simple example of implementing a new Tensorflow operation and its gradient in C++.☆56Updated 5 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- Optimized half precision gemm assembly kernels (deprecated due to ROCm)☆47Updated 7 years ago
- support cuda grammars in Visual Studio Code☆34Updated 8 years ago
- Python bindings for NVTX☆66Updated last year
- A CUDNN minimal deep learning training code sample using LeNet.☆264Updated last year
- PyTorch-Based Fast and Efficient Processing for Various Machine Learning Applications with Diverse Sparsity☆107Updated last week
- ☆45Updated 2 years ago
- notes on reading tensorflow source code☆13Updated 6 years ago
- CUDA Data Parallel Primitives Library☆428Updated 6 years ago
- CUDA Templates for Linear Algebra Subroutines☆97Updated 11 months ago
- Implementation of breadth first search on GPU with CUDA Driver API.☆48Updated 3 years ago
- Communication-Minimizing 2D Convolution in GPU Registers☆30Updated 11 years ago
- RDMA Optimization on MXNet☆14Updated 7 years ago