ardenma / implicit-gemm-tensor-core-convolutionLinks
Simple example of how to write an Implicit GEMM Convolution in CUDA using the tensor core WMMA API and bindings for PyTorch.
☆18Updated 2 years ago
Alternatives and similar repositories for implicit-gemm-tensor-core-convolution
Users that are interested in implicit-gemm-tensor-core-convolution are comparing it to the libraries listed below
Sorting:
- ☆111Updated last year
- ☆40Updated 5 years ago
- AMD RAD's multi-GPU Triton-based framework for seamless multi-GPU programming☆164Updated last week
- ☆165Updated last week
- Dissecting NVIDIA GPU Architecture☆117Updated 3 years ago
- ☆48Updated 5 years ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆127Updated last year
- Triton to TVM transpiler.☆22Updated last year
- ☆14Updated 6 years ago
- ☆50Updated 6 years ago
- An extension library of WMMA API (Tensor Core API)☆109Updated last year
- A tool for generating information about the matrix multiplication instructions in AMD Radeon™ and AMD Instinct™ accelerators☆125Updated 2 months ago
- ☆32Updated last year
- incubator repo for CUDA-TileIR backend☆97Updated 3 weeks ago
- Assembler for NVIDIA Volta and Turing GPUs☆238Updated 4 years ago
- Github mirror of trition-lang/triton repo.☆129Updated last week
- Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.☆91Updated 3 years ago
- OSDI 2023 Welder, deeplearning compiler☆32Updated 2 years ago
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆106Updated 7 months ago
- ☆41Updated 3 years ago
- NVIDIA cuTile learn☆158Updated last month
- LLVM/MLIR based compiler instrumentation of AMD GPU kernels☆20Updated 6 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆142Updated 2 years ago
- [DEPRECATED] Moved to ROCm/rocm-libraries repo☆138Updated last week
- amdgpu example code in hip/asm☆54Updated last week
- Artifacts of EVT ASPLOS'24☆28Updated last year
- Optimize GEMM with tensorcore step by step☆36Updated 2 years ago
- Automatic Mapping Generation, Verification, and Exploration for ISA-based Spatial Accelerators☆121Updated 3 years ago
- ☆173Updated 9 months ago
- MLIR-based partitioning system☆164Updated last week