ROCm / composable_kernel
Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
☆309Updated this week
Related projects ⓘ
Alternatives and complementary repositories for composable_kernel
- AMD's graph optimization engine.☆185Updated this week
- Assembler for NVIDIA Volta and Turing GPUs☆200Updated 2 years ago
- collection of benchmarks to measure basic GPU capabilities☆264Updated 4 months ago
- Shared Middle-Layer for Triton Compilation☆185Updated this week
- Stretching GPU performance for GEMMs and tensor contractions.☆220Updated this week
- CUDA Kernel Benchmarking Library☆512Updated 2 weeks ago
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆295Updated 2 months ago
- OpenAI Triton backend for Intel® GPUs☆142Updated this week
- Experimental projects related to TensorRT☆77Updated this week
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆401Updated last year
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆268Updated this week
- Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.☆274Updated 2 years ago
- ☆57Updated this week
- ☆128Updated this week
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆95Updated this week
- A simple high performance CUDA GEMM implementation.☆334Updated 10 months ago
- ROCm Communication Collectives Library (RCCL)☆265Updated this week
- PyTorch emulation library for Microscaling (MX)-compatible data formats☆162Updated last month
- A Easy-to-understand TensorOp Matmul Tutorial☆285Updated last month
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆123Updated last year
- Development repository for the Triton-Linalg conversion☆147Updated 3 weeks ago
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆446Updated 2 weeks ago
- CUDA Matrix Multiplication Optimization☆138Updated 3 months ago
- An extension library of WMMA API (Tensor Core API)☆82Updated 3 months ago
- Yinghan's Code Sample☆283Updated 2 years ago
- A collection of examples for the ROCm software stack☆166Updated this week
- Representation and Reference Lowering of ONNX Models in MLIR Compiler Infrastructure☆760Updated this week
- Training material for Nsight developer tools☆128Updated 3 months ago
- ☆162Updated 3 months ago
- ☆398Updated this week