krulis-martin / cuda-kmeans

Related projects ⓘ

Alternatives and complementary repositories for cuda-kmeans

• TiledTensor / TiledCUDA
  TiledCUDA is a highly efficient kernel template library designed to elevate CUDA C’s level of abstraction for processing tiles.
  ☆157Updated last week
• tlc-pack / libflash_attn
  Standalone Flash Attention v2 kernel without libtorch dependency
  ☆98Updated 2 months ago
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆19Updated 4 years ago
• owensgroup / SlabHash
  A warp-oriented dynamic hash table for GPUs
  ☆71Updated 10 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆84Updated 4 months ago
• uwsampl / SparseTIR
  SparseTIR: Sparse Tensor Compiler for Deep Learning
  ☆131Updated last year
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆141Updated 4 months ago
• owensgroup / merge-spmm
  Code for paper "Design Principles for Sparse Matrix Multiplication on the GPU" accepted to Euro-Par 2018
  ☆71Updated 4 years ago
• Shigangli / Magicube
  Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.
  ☆81Updated 2 years ago
• passlab / CUDAMicroBench
  ☆37Updated 3 years ago
• UDC-GAC / openCNN
  A Winograd Minimal Filter Implementation in CUDA
  ☆23Updated 3 years ago
• roastduck / FreeTensor
  A language and compiler for irregular tensor programs.
  ☆133Updated 6 months ago
• spcl / daceml
  A Data-Centric Compiler for Machine Learning
  ☆82Updated 10 months ago
• 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
• sunlex0717 / DissectingTensorCores
  ☆80Updated 7 months ago
• gty111 / GEMM_MMA
  Optimize GEMM with tensorcore step by step
  ☆15Updated 11 months ago
• UDC-GAC / venom
  A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
  ☆43Updated last year
• cupbop / CuPBoP
  A framework that support executing unmodified CUDA source code on non-NVIDIA devices.
  ☆105Updated 3 months ago
• NVIDIA / compute-sanitizer-samples
  Samples demonstrating how to use the Compute Sanitizer Tools and Public API
  ☆68Updated last year
• DD-DuDa / Cute-Learning
  Examples of CUDA implementations by Cutlass CuTe
  ☆101Updated 2 weeks ago
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆249Updated 4 years ago
• lixiuhong / batched_gemm
  ☆38Updated 4 years ago
• dgSPARSE / dgSPARSE-Lib
  PyTorch-Based Fast and Efficient Processing for Various Machine Learning Applications with Diverse Sparsity
  ☆99Updated this week
• illinois-impact / gpu-algorithms-labs
  IMPACT GPU Algorithms Teaching Labs
  ☆55Updated last year
• uiuc-arc / felix
  Optimize tensor program fast with Felix, a gradient descent autotuner.
  ☆19Updated 6 months ago
• apuaaChen / EVT_AE
  Artifacts of EVT ASPLOS'24
  ☆18Updated 8 months ago
• microsoft / ConvStencil
  ☆24Updated 7 months ago
• lenLRX / AmpereSparseMatmul
  study of Ampere' Sparse Matmul
  ☆14Updated 3 years ago
• mit-han-lab / parallel-computing-tutorial
  ☆144Updated last year
• ColfaxResearch / cfx-article-src
  ☆50Updated this week