ROCm / composable_kernel
Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
☆355Updated this week
Alternatives and similar repositories for composable_kernel:
Users that are interested in composable_kernel are comparing it to the libraries listed below
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆354Updated 5 months ago
- collection of benchmarks to measure basic GPU capabilities☆302Updated 2 weeks ago
- CUDA Kernel Benchmarking Library☆578Updated 3 months ago
- OpenAI Triton backend for Intel® GPUs☆165Updated this week
- Shared Middle-Layer for Triton Compilation☆228Updated this week
- Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.☆325Updated 2 months ago
- AMD's graph optimization engine.☆209Updated this week
- Assembler for NVIDIA Volta and Turing GPUs☆214Updated 3 years ago
- ☆185Updated 7 months ago
- Yinghan's Code Sample☆311Updated 2 years ago
- A Easy-to-understand TensorOp Matmul Tutorial☆322Updated 5 months ago
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆446Updated last year
- A simple high performance CUDA GEMM implementation.☆348Updated last year
- Stretching GPU performance for GEMMs and tensor contractions.☆233Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆310Updated this week
- Experimental projects related to TensorRT☆89Updated this week
- Step-by-step optimization of CUDA SGEMM☆288Updated 2 years ago
- CUDA Matrix Multiplication Optimization☆164Updated 7 months ago
- PyTorch emulation library for Microscaling (MX)-compatible data formats☆204Updated 5 months ago
- ROCm Communication Collectives Library (RCCL)☆303Updated this week
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆302Updated this week
- Development repository for the Triton language and compiler☆108Updated this week
- ☆137Updated this week
- A fast communication-overlapping library for tensor parallelism on GPUs.☆319Updated this week
- Training material for Nsight developer tools☆149Updated 6 months ago
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆530Updated 2 weeks ago
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆508Updated last month
- A tool for generating information about the matrix multiplication instructions in AMD Radeon™ and AMD Instinct™ accelerators☆76Updated last year
- ☆101Updated 2 months ago
- Development repository for the Triton-Linalg conversion☆176Updated 3 weeks ago