ROCm / composable_kernel
Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
☆338Updated this week
Alternatives and similar repositories for composable_kernel:
Users that are interested in composable_kernel are comparing it to the libraries listed below
- Assembler for NVIDIA Volta and Turing GPUs☆204Updated 3 years ago
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆334Updated 4 months ago
- collection of benchmarks to measure basic GPU capabilities☆287Updated 3 weeks ago
- Shared Middle-Layer for Triton Compilation☆220Updated this week
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆416Updated last year
- OpenAI Triton backend for Intel® GPUs☆156Updated this week
- CUDA Kernel Benchmarking Library☆550Updated 2 months ago
- AMD's graph optimization engine.☆196Updated this week
- Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.☆316Updated 3 weeks ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆294Updated this week
- Step-by-step optimization of CUDA SGEMM☆276Updated 2 years ago
- Stretching GPU performance for GEMMs and tensor contractions.☆231Updated this week
- ☆180Updated 6 months ago
- A tool for generating information about the matrix multiplication instructions in AMD Radeon™ and AMD Instinct™ accelerators☆73Updated last year
- Yinghan's Code Sample☆305Updated 2 years ago
- A simple high performance CUDA GEMM implementation.☆344Updated last year
- A Easy-to-understand TensorOp Matmul Tutorial☆307Updated 4 months ago
- Development repository for the Triton-Linalg conversion☆168Updated last month
- A fast communication-overlapping library for tensor parallelism on GPUs.☆280Updated 3 months ago
- CUDA Matrix Multiplication Optimization☆155Updated 6 months ago
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆99Updated this week
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆291Updated this week
- Experimental projects related to TensorRT☆86Updated this week
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆127Updated last year
- ☆133Updated this week
- AI Accelerator Benchmark focuses on evaluating AI Accelerators from a practical production perspective, including the ease of use and ver…☆219Updated 2 weeks ago
- Training material for Nsight developer tools☆143Updated 5 months ago
- BitBLAS is a library to support mixed-precision matrix multiplications, especially for quantized LLM deployment.☆503Updated this week
- Development repository for the Triton language and compiler☆103Updated this week
- PyTorch emulation library for Microscaling (MX)-compatible data formats☆193Updated 4 months ago