sandeepkumar-skb / pytorch_custom_op

Related projects ⓘ

Alternatives and complementary repositories for pytorch_custom_op

• ColfaxResearch / cfx-article-src
  ☆48Updated this week
• TiledTensor / TiledCUDA
  TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.
  ☆154Updated this week
• cchan / tccl
  extensible collectives library in triton
  ☆71Updated last month
• facebookresearch / fairring
  Fairring (FAIR + Herring) is a plug-in for PyTorch that provides a process group for distributed training that outperforms NCCL at large …
  ☆63Updated 2 years ago
• pytorch-labs / triton-cpu
  An experimental CPU backend for Triton (https//github.com/openai/triton)
  ☆35Updated 6 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆84Updated 4 months ago
• wangsiping97 / FastGEMV
  High-speed GEMV kernels, at most 2.7x speedup compared to pytorch baseline.
  ☆90Updated 4 months ago
• ColfaxResearch / cutlass-kernels
  ☆167Updated 4 months ago
• leimao / CUTLASS-Examples
  CUTLASS and CuTe Examples
  ☆13Updated this week
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆270Updated this week
• parasailteam / coconet
  ☆73Updated last year
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆59Updated 6 years ago
• tlc-pack / cutlass_fpA_intB_gemm
  A standalone GEMM kernel for fp16 activation and quantized weight, extracted from FasterTransformer
  ☆85Updated 8 months ago
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated 2 months ago
• sunlex0717 / DissectingTensorCores
  ☆80Updated 7 months ago
• sjtu-epcc / Tacker
  Tacker: Tensor-CUDA Core Kernel Fusion for Improving the GPU Utilization while Ensuring QoS
  ☆17Updated 2 years ago
• mmperf / mmperf
  MatMul Performance Benchmarks for a Single CPU Core comparing both hand engineered and codegen kernels.
  ☆124Updated last year
• triton-lang / triton-cpu
  An experimental CPU backend for Triton
  ☆56Updated last week
• ROCm / rocMLIR
  ☆128Updated this week
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆141Updated 4 months ago
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆131Updated last year
• octoml / octoml-profile
  Home for OctoML PyTorch Profiler
  ☆107Updated last year
• Bruce-Lee-LY / cuda_back2back_hgemm
  Use tensor core to calculate back-to-back HGEMM (half-precision general matrix multiplication) with MMA PTX instruction.
  ☆11Updated last year
• wzsh / wmma_tensorcore_sample
  Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)
  ☆115Updated 4 years ago
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• microsoft / triton-shared
  Shared Middle-Layer for Triton Compilation
  ☆191Updated this week
• nox-410 / tvm.tl
  An extention of TVMScript to write simple and high performance GPU kernels with tensorcore.
  ☆49Updated 3 months ago
• lixiuhong / batched_gemm
  ☆38Updated 4 years ago
• UofT-EcoSystem / hotline
  ☆31Updated last year
• facebookresearch / FAMBench
  Benchmarks to capture important workloads.
  ☆28Updated 5 months ago