Ascend / torchairLinks
☆22Updated this week
Alternatives and similar repositories for torchair
Users that are interested in torchair are comparing it to the libraries listed below
Sorting:
- ☆65Updated 8 months ago
- DLSlime: Flexible & Efficient Heterogeneous Transfer Toolkit☆91Updated this week
- ☆96Updated 9 months ago
- Triton adapter for Ascend. Mirror of https://gitee.com/ascend/triton-ascend☆98Updated this week
- ☆69Updated last week
- Standalone Flash Attention v2 kernel without libtorch dependency☆113Updated last year
- GPTQ inference TVM kernel☆41Updated last year
- A Triton JIT runtime and ffi provider in C++☆30Updated 2 weeks ago
- Compare different hardware platforms via the Roofline Model for LLM inference tasks.☆120Updated last year
- ☆104Updated last year
- TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.☆104Updated 6 months ago
- DeeperGEMM: crazy optimized version☆74Updated 8 months ago
- Framework to reduce autotune overhead to zero for well known deployments.☆92Updated 3 months ago
- An experimental communicating attention kernel based on DeepEP.☆35Updated 5 months ago
- NVSHMEM‑Tutorial: Build a DeepEP‑like GPU Buffer☆152Updated 3 months ago
- ⚡️Write HGEMM from scratch using Tensor Cores with WMMA, MMA and CuTe API, Achieve Peak⚡️ Performance.☆141Updated 8 months ago
- A practical way of learning Swizzle☆36Updated 11 months ago
- Tile-based language built for AI computation across all scales☆115Updated 2 weeks ago
- ☆19Updated last year
- ☆116Updated 7 months ago
- Building the Virtuous Cycle for AI-driven LLM Systems☆112Updated this week
- a size profiler for cuda binary☆69Updated 3 months ago
- A lightweight design for computation-communication overlap.☆207Updated 2 weeks ago
- Decoding Attention is specially optimized for MHA, MQA, GQA and MLA using CUDA core for the decoding stage of LLM inference.☆46Updated 7 months ago
- Automated Parallelization System and Infrastructure for Multiple Ecosystems☆82Updated last year
- We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …☆191Updated 11 months ago
- ☆102Updated last week
- A NCCL extension library, designed to efficiently offload GPU memory allocated by the NCCL communication library.☆77Updated 3 weeks ago
- ☆125Updated 4 months ago
- ☆153Updated 10 months ago