openxla / openxla-nvgpu
☆48Updated 6 months ago
Related projects: ⓘ
- Shared Middle-Layer for Triton Compilation☆160Updated this week
- ☆138Updated 2 months ago
- System for automated integration of deep learning backends.☆48Updated 2 years ago
- Benchmark code for the "Online normalizer calculation for softmax" paper☆52Updated 6 years ago
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆90Updated this week
- PET: Optimizing Tensor Programs with Partially Equivalent Transformations and Automated Corrections☆112Updated 2 years ago
- An extension library of WMMA API (Tensor Core API)☆81Updated 2 months ago
- Benchmark scripts for TVM☆73Updated 2 years ago
- Standalone Flash Attention v2 kernel without libtorch dependency☆93Updated last week
- A lightweight, Pythonic, frontend for MLIR☆79Updated 10 months ago
- TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.☆114Updated last week
- High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.☆81Updated 2 months ago
- SparseTIR: Sparse Tensor Compiler for Deep Learning☆129Updated last year
- An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.☆49Updated last month
- llama INT4 cuda inference with AWQ☆46Updated 2 months ago
- A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer☆82Updated 6 months ago
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆123Updated 11 months ago
- ☆73Updated 5 months ago
- ☆141Updated last year
- ☆14Updated 4 months ago
- Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …☆61Updated 2 years ago
- A language and compiler for irregular tensor programs.☆132Updated 4 months ago
- ☆66Updated last year
- OpenAI Triton backend for Intel® GPUs☆126Updated this week
- play gemm with tvm☆81Updated last year
- A home for the final text of all TVM RFCs.☆99Updated 3 months ago
- Experiments and prototypes associated with IREE or MLIR☆47Updated last month
- An experimental CPU backend for Triton (https//github.com/openai/triton)☆30Updated 4 months ago
- ☆39Updated 3 years ago
- PyTorch bindings for CUTLASS grouped GEMM.☆41Updated 3 weeks ago