ROCm / composable_kernel
Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators
☆313Updated this week
Related projects ⓘ
Alternatives and complementary repositories for composable_kernel
- Assembler for NVIDIA Volta and Turing GPUs☆201Updated 2 years ago
- collection of benchmarks to measure basic GPU capabilities☆265Updated 5 months ago
- Shared Middle-Layer for Triton Compilation☆191Updated this week
- OpenAI Triton backend for Intel® GPUs☆143Updated this week
- Several optimization methods of half-precision general matrix multiplication (HGEMM) using tensor core with WMMA API and MMA PTX instruct…☆302Updated 2 months ago
- An unofficial cuda assembler, for all generations of SASS, hopefully :)☆405Updated last year
- CUDA Kernel Benchmarking Library☆519Updated this week
- AMD's graph optimization engine.☆186Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆270Updated this week
- Stretching GPU performance for GEMMs and tensor contractions.☆223Updated this week
- Experimental projects related to TensorRT☆81Updated this week
- ☆167Updated 4 months ago
- Optimizing SGEMM kernel functions on NVIDIA GPUs to a close-to-cuBLAS performance.☆280Updated 2 years ago
- A simple high performance CUDA GEMM implementation.☆335Updated 10 months ago
- Yinghan's Code Sample☆289Updated 2 years ago
- ☆128Updated this week
- A tool for generating information about the matrix multiplication instructions in AMD Radeon™ and AMD Instinct™ accelerators☆67Updated 10 months ago
- A Easy-to-understand TensorOp Matmul Tutorial☆290Updated 2 months ago
- ☆80Updated 7 months ago
- ROCm Communication Collectives Library (RCCL)☆268Updated this week
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆95Updated last week
- Step-by-step optimization of CUDA SGEMM☆240Updated 2 years ago
- cudnn_frontend provides a c++ wrapper for the cudnn backend API and samples on how to use it☆455Updated 3 weeks ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆250Updated this week
- PyTorch emulation library for Microscaling (MX)-compatible data formats☆163Updated last month
- ☆59Updated this week
- CUDA Matrix Multiplication Optimization☆141Updated 4 months ago
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆124Updated last year
- Training material for Nsight developer tools☆129Updated 3 months ago
- A home for the final text of all TVM RFCs.☆101Updated last month