anilshanbhag / gpu-topkLinks
Efficient Top-K implementation on the GPU
☆179Updated 6 years ago
Alternatives and similar repositories for gpu-topk
Users that are interested in gpu-topk are comparing it to the libraries listed below
Sorting:
- this is the release repository of superneurons☆52Updated 4 years ago
- Assembler for NVIDIA Volta and Turing GPUs☆218Updated 3 years ago
- Subpart source code of of deepcore v0.7☆27Updated 4 years ago
- Implementation of TSM2L and TSM2R -- High-Performance Tall-and-Skinny Matrix-Matrix Multiplication Algorithms for CUDA☆32Updated 4 years ago
- ☆112Updated last year
- Dissecting NVIDIA GPU Architecture☆95Updated 2 years ago
- Kernel Fusion and Runtime Compilation Based on NNVM☆70Updated 8 years ago
- [MLSys 2021] IOS: Inter-Operator Scheduler for CNN Acceleration☆200Updated 3 years ago
- HierarchicalKV is a part of NVIDIA Merlin and provides hierarchical key-value storage to meet RecSys requirements. The key capability of…☆147Updated 2 weeks ago
- ☆144Updated 5 months ago
- A warp-oriented dynamic hash table for GPUs☆73Updated last year
- Yinghan's Code Sample☆328Updated 2 years ago
- ☆121Updated 6 months ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆121Updated 2 years ago
- ☆97Updated last year
- ☆194Updated 2 years ago
- ☆93Updated 2 months ago
- THIS REPOSITORY HAS MOVED TO github.com/nvidia/cub, WHICH IS AUTOMATICALLY MIRRORED HERE.☆84Updated last year
- ☆51Updated 5 years ago
- Third party assembler and GEMM library for NVIDIA Kepler GPU☆81Updated 5 years ago
- A library of GPU kernels for sparse matrix operations.☆264Updated 4 years ago
- An extension library of WMMA API (Tensor Core API)☆97Updated 10 months ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆182Updated 4 months ago
- An Efficient Pipelined Data Parallel Approach for Training Large Model☆76Updated 4 years ago
- An unofficial cuda assembler, for all generations of SASS, hopefully ：）☆83Updated 2 years ago
- A tool for examining GPU scheduling behavior.☆84Updated 9 months ago
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆88Updated 2 years ago
- Code for paper "Design Principles for Sparse Matrix Multiplication on the GPU" accepted to Euro-Par 2018☆71Updated 4 years ago
- Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616☆132Updated last year
- ☆79Updated 2 years ago