r-barnes / pytorch_cmake_example

Related projects ⓘ

Alternatives and complementary repositories for pytorch_cmake_example

• wzsh / wmma_tensorcore_sample
  Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)
  ☆116Updated 4 years ago
• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆249Updated 3 years ago
• dumerrill / merge-spmv
  ☆90Updated 7 years ago
• GPUPeople / spECK
  Efficient SpGEMM on GPU using CUDA and CSR
  ☆50Updated last year
• YulhwaKim / cutlass_tilesparse
  CUDA templates for tile-sparse matrix multiplication based on CUTLASS.
  ☆49Updated 6 years ago
• Huanghongru / SGEMM-Implementation-and-Optimization
  Some source code about matrix multiplication implementation on CUDA
  ☆35Updated 6 years ago
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆129Updated 2 years ago
• cmpute / pytorch-cmake-example
  Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)
  ☆11Updated 4 years ago
• sunlex0717 / DissectingTensorCores
  ☆80Updated 7 months ago
• wangzyon / NVIDIA_SGEMM_PRACTICE
  Step-by-step optimization of CUDA SGEMM
  ☆242Updated 2 years ago
• UDC-GAC / venom
  A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
  ☆43Updated 11 months ago
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆84Updated 4 months ago
• gunrock / loops
  🎃 GPU load-balancing library for regular and irregular computations.
  ☆57Updated 5 months ago
• poojahira / spmv-cuda
  Implementation and analysis of five different GPU based SPMV algorithms in CUDA
  ☆35Updated 5 years 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
• ColfaxResearch / cutlass-kernels
  ☆169Updated 4 months ago
• NVIDIA / nvbench
  CUDA Kernel Benchmarking Library
  ☆519Updated this week
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆129Updated 3 months ago
• oresths / tSparse
  A GPU algorithm for sparse matrix-matrix multiplication
  ☆66Updated 4 years ago
• c3sr / tcu_scope
  ☆44Updated 5 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
• Karbo123 / pytorch_grouped_gemm
  High Performance Grouped GEMM in PyTorch
  ☆22Updated 2 years ago
• albanD / subclass_zoo
  ☆149Updated 5 months ago
• hgyhungry / ge-spmm
  ☆101Updated 3 years ago
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆141Updated 4 months ago
• RustingSword / tensorboard_logger
  C++ API to log data in tensorboard format.
  ☆76Updated 2 years ago
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆129Updated last year
• iree-org / iree-nvgpu
  ☆48Updated 8 months ago
• Cjkkkk / CUDA_gemm
  A simple high performance CUDA GEMM implementation.
  ☆335Updated 10 months ago