google / tsl
☆82Updated this week
Related projects: ⓘ
- A lightweight memory allocator for hardware-accelerated machine learning☆114Updated last month
- ☆65Updated last month
- ☆392Updated this week
- oneAPI Collective Communications Library (oneCCL)☆189Updated 3 weeks ago
- ☆124Updated this week
- A sandbox for quick iteration and experimentation on projects related to IREE, MLIR, and LLVM☆52Updated last week
- Starlark implementation of bazel rules for CUDA.☆85Updated this week
- MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.☆123Updated 11 months ago
- Stores documents and resources used by the OpenXLA developer community☆105Updated last month
- Experiments and prototypes associated with IREE or MLIR☆47Updated last month
- Unified Collective Communication Library☆190Updated this week
- ROCm Thrust - run Thrust dependent software on AMD GPUs☆100Updated this week
- TPP experimentation on MLIR for linear algebra☆110Updated last week
- Intel® Extension for MLIR. A staging ground for MLIR dialects and tools for Intel devices using the MLIR toolchain.☆116Updated this week
- ☆297Updated 2 months ago
- Open source cross-platform compiler for compute-intensive loops used in AI algorithms, from Microsoft Research☆95Updated 11 months ago
- Tests and benchmarks for cudnn (and in the future, other nvidia libraries)☆52Updated 3 years ago
- A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")☆250Updated this week
- ☆52Updated last week
- oneAPI Specification source files☆183Updated this week
- OpenAI Triton backend for Intel® GPUs☆126Updated this week
- An extension library of WMMA API (Tensor Core API)☆81Updated 2 months ago
- Bazel wrapper around the pybind11 repository☆96Updated 3 weeks ago
- Stretching GPU performance for GEMMs and tensor contractions.☆213Updated this week
- ☆80Updated 3 months ago
- The NVIDIA® Tools Extension SDK (NVTX) is a C-based Application Programming Interface (API) for annotating events, code ranges, and resou…☆279Updated last week
- Unified compiler/runtime for interfacing with PyTorch Dynamo.☆90Updated this week
- ROCm BLAS marshalling library☆110Updated this week
- An implementation of BLAS using the SYCL open standard.☆250Updated 2 weeks ago
- GPUOcelot: A dynamic compilation framework for PTX☆136Updated 3 months ago