wmmae / wmma_extension
An extension library of WMMA API (Tensor Core API)
☆96Updated 9 months ago
Alternatives and similar repositories for wmma_extension:
Users that are interested in wmma_extension are comparing it to the libraries listed below
- ☆95Updated last year
- Dissecting NVIDIA GPU Architecture☆91Updated 2 years ago
- ☆101Updated last month
- ☆60Updated 4 months ago
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆181Updated 2 months ago
- Assembler for NVIDIA Volta and Turing GPUs☆217Updated 3 years ago
- ☆141Updated this week
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆130Updated last year
- A GPU benchmark suite for assessing on-chip GPU memory bandwidth☆104Updated 7 years ago
- ☆38Updated 5 years ago
- Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)☆130Updated 4 years ago
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆85Updated 2 years ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆81Updated last week
- CUDA Matrix Multiplication Optimization☆179Updated 9 months ago
- Several optimization methods of half-precision general matrix vector multiplication (HGEMV) using CUDA core.☆61Updated 7 months ago
- A language and compiler for irregular tensor programs.☆138Updated 4 months ago
- ☆50Updated 5 years ago
- A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores☆51Updated last year
- ☆43Updated 4 years ago
- Implementation of TSM2L and TSM2R -- High-Performance Tall-and-Skinny Matrix-Matrix Multiplication Algorithms for CUDA☆32Updated 4 years ago
- play gemm with tvm☆90Updated last year
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆99Updated last month
- Artifacts of EVT ASPLOS'24☆23Updated last year
- TPP experimentation on MLIR for linear algebra☆127Updated this week
- Intel® Extension for MLIR. A staging ground for MLIR dialects and tools for Intel devices using the MLIR toolchain.☆134Updated this week
- An implementation of SGEMV with performance comparable to cuBLAS.☆9Updated 3 years ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆91Updated 6 years ago
- ☆31Updated 2 years ago
- ☆115Updated 4 months ago
- Third party assembler and GEMM library for NVIDIA Kepler GPU☆81Updated 5 years ago