krulis-martin / cuda-kmeans
A novell, highly-optimized CUDA implementation of k-means algorithm.
☆29Updated 2 years ago
Related projects: ⓘ
- PyTorch compilation tutorial covering TorchScript, torch.fx, and Slapo☆18Updated last year
- Code for paper "Design Principles for Sparse Matrix Multiplication on the GPU" accepted to Euro-Par 2018☆70Updated 3 years ago
- IMPACT GPU Algorithms Teaching Labs☆55Updated last year
- Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS☆17Updated 2 years ago
- Implementation of TSM2L and TSM2R -- High-Performance Tall-and-Skinny Matrix-Matrix Multiplication Algorithms for CUDA☆31Updated 4 years ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆52Updated 6 years ago
- GVProf: A Value Profiler for GPU-based Clusters☆46Updated 5 months ago
- MAGIS: Memory Optimization via Coordinated Graph Transformation and Scheduling for DNN (ASPLOS'24)☆35Updated 3 months ago
- ☆38Updated 4 years ago
- Implementation of breadth first search on GPU with CUDA Driver API.☆46Updated 3 years ago
- CUDA templates for tile-sparse matrix multiplication based on CUTLASS.☆48Updated 6 years ago
- ☆17Updated this week
- A Winograd Minimal Filter Implementation in CUDA☆20Updated 3 years ago
- study of Ampere' Sparse Matmul☆13Updated 3 years ago
- An optimizing compiler for decision tree ensemble inference.☆15Updated last week
- End to End steps for adding custom ops in PyTorch.☆18Updated 4 years ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆129Updated last year
- A Data-Centric Compiler for Machine Learning☆81Updated 8 months ago
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆79Updated last year
- Optimize tensor program fast with Felix, a gradient descent autotuner.☆15Updated 4 months ago
- Triton to TVM transpiler.☆15Updated last week
- ☆14Updated 5 months ago
- A GPU algorithm for sparse matrix-matrix multiplication☆65Updated 3 years ago
- A warp-oriented dynamic hash table for GPUs☆70Updated 8 months ago
- ☆14Updated 5 years ago
- ☆20Updated 3 years ago
- ☆39Updated 3 years ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- ☆17Updated 3 years ago
- cuDNN sample codes provided by Nvidia☆42Updated 5 years ago