Alternatives and similar repositories for pytorch_cmake_example:

Users that are interested in pytorch_cmake_example are comparing it to the libraries listed below

• wzsh / wmma_tensorcore_sample
  Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)
  ☆127Updated 4 years ago
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆260Updated 4 years ago
• sunlex0717 / DissectingTensorCores
  ☆92Updated 11 months ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆130Updated 3 years ago
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆173Updated 8 months ago
• cmpute / pytorch-cmake-example
  Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)
  ☆11Updated 4 years ago
• sjfeng1999 / gpu-arch-microbenchmark
  Dissecting NVIDIA GPU Architecture
  ☆90Updated 2 years ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆91Updated 8 months ago
• daadaada / turingas
  Assembler for NVIDIA Volta and Turing GPUs
  ☆214Updated 3 years ago
• ParCIS / Magicube
  Magicube is a high-performance library for quantized sparse matrix operations (SpMM and SDDMM) of deep learning on Tensor Cores.
  ☆86Updated 2 years ago
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆151Updated 7 months ago
• hgyhungry / ge-spmm
  ☆107Updated 3 years ago
• wangzyon / NVIDIA_SGEMM_PRACTICE
  Step-by-step optimization of CUDA SGEMM
  ☆294Updated 2 years ago
• UDC-GAC / venom
  A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
  ☆50Updated last year
• YulhwaKim / cutlass_tilesparse
  CUDA templates for tile-sparse matrix multiplication based on CUTLASS.
  ☆50Updated 7 years ago
• triton-lang / kernels
  ☆73Updated 4 months ago
• parasailteam / coconet
  ☆76Updated 2 years ago
• Karbo123 / pytorch_grouped_gemm
  High Performance Grouped GEMM in PyTorch
  ☆28Updated 2 years ago
• ColfaxResearch / cutlass-kernels
  ☆193Updated 8 months ago
• KnowingNothing / MatmulTutorial
  A Easy-to-understand TensorOp Matmul Tutorial
  ☆331Updated 6 months ago
• NVIDIA / online-softmax
  Benchmark code for the "Online normalizer calculation for softmax" paper
  ☆85Updated 6 years ago
• sandeepkumar-skb / pytorch_custom_op
  End to End steps for adding custom ops in PyTorch.
  ☆21Updated 4 years ago
• mark-poscablo / gpu-prefix-sum
  CUDA implementation of exclusive prefix sum via Blelloch's algorithm
  ☆27Updated 7 years ago
• roastduck / FreeTensor
  A language and compiler for irregular tensor programs.
  ☆139Updated 3 months ago
• albanD / subclass_zoo
  ☆162Updated 9 months ago
• Cjkkkk / CUDA_gemm
  A simple high performance CUDA GEMM implementation.
  ☆355Updated last year
• cwpearson / nvidia-performance-tools
  Instructions, Docker images, and examples for Nsight Compute and Nsight Systems
  ☆130Updated 4 years ago
• bertmaher / simplegemm
  ☆82Updated last week
• yifuwang / symm-mem-recipes
  ☆57Updated 3 months ago
• pytorch / tensorpipe
  A tensor-aware point-to-point communication primitive for machine learning
  ☆256Updated 2 years ago