NVIDIA / TensorRT-Incubator
Experimental projects related to TensorRT
☆81Updated this week
Related projects ⓘ
Alternatives and complementary repositories for TensorRT-Incubator
- Shared Middle-Layer for Triton Compilation☆191Updated this week
- ☆167Updated 4 months ago
- ☆48Updated this week
- Assembler for NVIDIA Volta and Turing GPUs☆201Updated 2 years ago
- An extension library of WMMA API (Tensor Core API)☆84Updated 4 months ago
- A fast communication-overlapping library for tensor parallelism on GPUs.☆224Updated 3 weeks ago
- PyTorch emulation library for Microscaling (MX)-compatible data formats☆163Updated last month
- collection of benchmarks to measure basic GPU capabilities☆265Updated 5 months ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆59Updated 6 years ago
- ☆80Updated 7 months ago
- A home for the final text of all TVM RFCs.☆101Updated last month
- A Easy-to-understand TensorOp Matmul Tutorial☆290Updated 2 months ago
- MSCCL++: A GPU-driven communication stack for scalable AI applications☆250Updated this week
- ☆196Updated last year
- TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.☆154Updated 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
- ☆79Updated 8 months ago
- Composable Kernel: Performance Portable Programming Model for Machine Learning Tensor Operators☆313Updated this week
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆270Updated this week
- ☆79Updated 2 months ago
- OpenAI Triton backend for Intel® GPUs☆143Updated this week
- ☆48Updated 8 months ago
- Dissecting NVIDIA GPU Architecture☆82Updated 2 years ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆98Updated 2 months ago
- play gemm with tvm☆84Updated last year
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆90Updated 4 months ago
- CUDA Matrix Multiplication Optimization☆141Updated 4 months ago
- ☆138Updated 2 weeks ago
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆124Updated last year
- Examples of CUDA implementations by Cutlass CuTe☆98Updated last week