wzsh / wmma_tensorcore_sample
Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)
☆122Updated 4 years ago
Alternatives and similar repositories for wmma_tensorcore_sample:
Users that are interested in wmma_tensorcore_sample are comparing it to the libraries listed below
- ☆84Updated 9 months ago
- CUDA Matrix Multiplication Optimization☆155Updated 6 months ago
- Dissecting NVIDIA GPU Architecture☆83Updated 2 years ago
- An extension library of WMMA API (Tensor Core API)☆87Updated 6 months ago
- Examples of CUDA implementations by Cutlass CuTe☆132Updated 2 months ago
- ☆108Updated 9 months ago
- Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.☆54Updated 4 months ago
- play gemm with tvm☆85Updated last year
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆85Updated 2 years ago
- Training material for Nsight developer tools☆143Updated 5 months ago
- ☆180Updated 6 months ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆175Updated this week
- A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores☆48Updated last year
- ☆128Updated last month
- ☆95Updated last month
- collection of benchmarks to measure basic GPU capabilities☆287Updated 3 weeks ago
- ☆67Updated last month
- ☆38Updated 4 years ago
- Assembler for NVIDIA Volta and Turing GPUs☆204Updated 3 years ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆47Updated this week
- ☆46Updated 5 years ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆88Updated 11 months ago
- Sample examples of how to call collective operation functions on multi-GPU environments. A simple example of using broadcast, reduce, all…☆29Updated last year
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆334Updated 4 months ago
- Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.☆316Updated 3 weeks ago
- Implementation of TSM2L and TSM2R -- High-Performance Tall-and-Skinny Matrix-Matrix Multiplication Algorithms for CUDA☆32Updated 4 years ago
- ☆58Updated 3 weeks ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆64Updated 6 years ago
- ☆32Updated 2 years ago
- Some source code about matrix multiplication implementation on CUDA☆35Updated 6 years ago