UDC-GAC / venomLinks
A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
☆54Updated last year
Alternatives and similar repositories for venom
Users that are interested in venom are comparing it to the libraries listed below
Sorting:
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆89Updated 2 years ago
- ☆158Updated last year
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆140Updated 2 years ago
- Artifacts of EVT ASPLOS'24☆28Updated last year
- SpInfer: Leveraging Low-Level Sparsity for Efficient Large Language Model Inference on GPUs☆59Updated 7 months ago
- Distributed MoE in a Single Kernel [NeurIPS '25]☆91Updated last month
- ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch☆38Updated 7 months ago
- ☆32Updated 3 years ago
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆51Updated last year
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆120Updated last year
- ☆109Updated last year
- play gemm with tvm☆92Updated 2 years ago
- DietCode Code Release☆65Updated 3 years ago
- A GPU-optimized system for efficient long-context LLMs decoding with low-bit KV cache.☆61Updated 3 weeks ago
- A lightweight design for computation-communication overlap.☆183Updated last month
- Dissecting NVIDIA GPU Architecture☆109Updated 3 years ago
- ☆38Updated last year
- ☆83Updated 2 years ago
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆116Updated 3 years ago
- ☆16Updated 2 years ago
- Flash-LLM: Enabling Cost-Effective and Highly-Efficient Large Generative Model Inference with Unstructured Sparsity☆222Updated 2 years ago
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆121Updated 3 years ago
- study of Ampere' Sparse Matmul☆18Updated 4 years ago
- ☆243Updated last year
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆100Updated 4 months ago
- MAGIS: Memory Optimization via Coordinated Graph Transformation and Scheduling for DNN (ASPLOS'24)☆55Updated last year
- Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)☆145Updated 5 years ago
- ☆154Updated 6 months ago
- ☆156Updated 10 months ago
- PyTorch-Based Fast and Efficient Processing for Various Machine Learning Applications with Diverse Sparsity☆118Updated 3 weeks ago