iree-org / iree-nvgpu
☆48Updated 8 months ago
Related projects ⓘ
Alternatives and complementary repositories for iree-nvgpu
- Shared Middle-Layer for Triton Compilation☆191Updated this week
- ☆167Updated 4 months ago
- Benchmark scripts for TVM☆73Updated 2 years ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆131Updated last year
- An extension library of WMMA API (Tensor Core API)☆84Updated 4 months ago
- ☆128Updated this week
- TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.☆154Updated this week
- ☆80Updated 7 months ago
- Samples demonstrating how to use the Compute Sanitizer Tools and Public API☆67Updated last year
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆49Updated 3 months ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆98Updated 2 months ago
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆95Updated last week
- MLIR-based partitioning system☆40Updated this week
- Benchmark code for the "Online normalizer calculation for softmax" paper☆59Updated 6 years ago
- A home for the final text of all TVM RFCs.☆101Updated last month
- ☆48Updated this week
- A lightweight, Pythonic, frontend for MLIR☆80Updated last year
- ☆67Updated last year
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆114Updated 2 years ago
- Experimental projects related to TensorRT☆81Updated this week
- OpenAI Triton backend for Intel® GPUs☆143Updated this week
- play gemm with tvm☆84Updated last year
- TPP experimentation on MLIR for linear algebra☆110Updated last month
- A language and compiler for irregular tensor programs.☆133Updated 6 months ago
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆124Updated last year
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆270Updated this week
- Assembler for NVIDIA Volta and Turing GPUs☆201Updated 2 years ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆85Updated 8 months ago
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆90Updated 4 months ago