flashinfer-ai / debug-print

Related projects ⓘ

Alternatives and complementary repositories for debug-print

• feifeibear / ChituAttention
  Quantized Attention on GPU
  ☆29Updated last week
• LeiWang1999 / Stream-k.tvm
  ☆18Updated last month
• TiledTensor / ThrillerFlow
  ThrillerFlow is a Dataflow Analysis and Codegen Framework written in Rust.
  ☆10Updated last month
• LeiWang1999 / AutoGPTQ.tvm
  GPTQ inference TVM kernel
  ☆35Updated 6 months ago
• mlc-ai / xgrammar
  ☆22Updated this week
• tgale96 / grouped_gemm
  PyTorch bindings for CUTLASS grouped GEMM.
  ☆53Updated last week
• hao-ai-lab / MuxServe
  ☆42Updated 5 months ago
• triton-lang / kernels
  ☆43Updated last week
• cat538 / SKVQ
  SKVQ: Sliding-window Key and Value Cache Quantization for Large Language Models
  ☆12Updated last month
• thu-ml / Jetfire-INT8Training
  ☆21Updated 3 months ago
• sgl-project / tensorrt-demo
  TensorRT LLM Benchmark Configuration
  ☆11Updated 3 months ago
• weishengying / cutlass_flash_atten_fp8
  使用 cutlass 仓库在 ada 架构上实现 fp8 的 flash attention
  ☆52Updated 3 months ago
• mit-han-lab / Block-Sparse-Attention
  A sparse attention kernel supporting mix sparse patterns
  ☆53Updated 3 weeks ago
• zheng-ningxin / SparTA
  ☆8Updated last year
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆85Updated 8 months ago
• hgyhungry / alcop-artifact
  ☆21Updated last year
• bytedance / QSync
  Official resporitory for "IPDPS' 24 QSync: Quantization-Minimized Synchronous Distributed Training Across Hybrid Devices".
  ☆19Updated 8 months ago
• xdit-project / DistVAE
  A parallelism VAE avoids OOM for high resolution image generation
  ☆40Updated last month
• pku-liang / MAGIS
  MAGIS: Memory Optimization via Coordinated Graph Transformation and Scheduling for DNN (ASPLOS'24)
  ☆43Updated 5 months ago
• LeiWang1999 / TVM.CMakeExtend
  Tutorials of Extending and importing TVM with CMAKE Include dependency.
  ☆10Updated last month
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆49Updated 3 months ago
• chengzeyi / ParaAttention
  [WIP] Context parallel attention that works with torch.compile
  ☆20Updated last week
• humuyan / Korch
  ASPLOS'24: Optimal Kernel Orchestration for Tensor Programs with Korch
  ☆29Updated 3 months ago
• heheda12345 / MagPy
  ☆31Updated 5 months ago
• zhuzilin / pytorch-malloc
  An external memory allocator example for PyTorch.
  ☆13Updated 3 years ago
• ankan-ban / llama_cu_awq
  llama INT4 cuda inference with AWQ
  ☆47Updated 4 months ago
• INT-FlashAttention2024 / INT-FlashAttention
  ☆46Updated last month
• TiledTensor / TiledCUDA
  TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.
  ☆151Updated this week
• InternLM / Awesome-LLM-Training-System
  ☆18Updated 3 months ago
• AlibabaPAI / FLASHNN
  ☆79Updated 2 months ago