ROCm / composable_kernel
Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
☆383Updated this week
Alternatives and similar repositories for composable_kernel:
Users that are interested in composable_kernel are comparing it to the libraries listed below
- collection of benchmarks to measure basic GPU capabilities☆359Updated 2 months ago
- CUDA Kernel Benchmarking Library☆620Updated this week
- Assembler for NVIDIA Volta and Turing GPUs☆217Updated 3 years ago
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆393Updated 7 months ago
- Shared Middle-Layer for Triton Compilation☆245Updated this week
- Stretching GPU performance for GEMMs and tensor contractions.☆235Updated last week
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆483Updated 2 years ago
- ☆198Updated 9 months ago
- Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.☆337Updated 3 months ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆341Updated this week
- OpenAI Triton backend for Intel® GPUs☆179Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆318Updated this week
- A Easy-to-understand TensorOp Matmul Tutorial☆342Updated 7 months ago
- ROCm Communication Collectives Library (RCCL)☆317Updated this week
- AMD's graph optimization engine.☆215Updated this week
- Experimental projects related to TensorRT☆97Updated this week
- A simple high performance CUDA GEMM implementation.☆361Updated last year
- Step-by-step optimization of CUDA SGEMM☆308Updated 3 years ago
- Yinghan's Code Sample☆323Updated 2 years ago
- PyTorch emulation library for Microscaling (MX)-compatible data formats☆217Updated this week
- CUDA Matrix Multiplication Optimization☆179Updated 9 months ago
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆373Updated 2 weeks ago
- A tool for bandwidth measurements on NVIDIA GPUs.☆407Updated last week
- ☆410Updated this week
- ☆141Updated this week
- Development repository for the Triton language and compiler☆118Updated this week
- Development repository for the Triton-Linalg conversion☆185Updated 2 months ago
- ☆101Updated last month
- Microsoft Collective Communication Library☆343Updated last year
- AI Tensor Engine for ROCm☆168Updated this week