r-barnes / pytorch_cmake_example

Related projects: ⓘ

• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆240Updated 3 years ago
• wzsh / wmma_tensorcore_sample
  Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)
  ☆109Updated 4 years ago
• cmpute / pytorch-cmake-example
  Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)
  ☆11Updated 4 years ago
• dumerrill / merge-spmv
  ☆88Updated 7 years ago
• wangzyon / NVIDIA_SGEMM_PRACTICE
  Step-by-step optimization of CUDA SGEMM
  ☆207Updated 2 years ago
• chrischoy / MakePytorchPlusPlus
  How and why you want to make your pytorch CUDA/CPP extension with a Makefile
  ☆171Updated 5 years ago
• dfm / extending-jax
  Extending JAX with custom C++ and CUDA code
  ☆369Updated last month
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆127Updated 2 years ago
• siboehm / SGEMM_CUDA
  Fast CUDA matrix multiplication from scratch
  ☆423Updated 8 months ago
• KernelTuner / kernel_tuner
  Kernel Tuner
  ☆273Updated this week
• DeMoriarty / custom_matmul_kernels
  Customized matrix multiplication kernels
  ☆53Updated 2 years ago
• YulhwaKim / cutlass_tilesparse
  CUDA templates for tile-sparse matrix multiplication based on CUTLASS.
  ☆48Updated 6 years ago
• RustingSword / tensorboard_logger
  C++ API to log data in tensorboard format.
  ☆75Updated last year
• sunlex0717 / DissectingTensorCores
  ☆73Updated 5 months ago
• GPUPeople / spECK
  Efficient SpGEMM on GPU using CUDA and CSR
  ☆50Updated last year
• ptillet / torch-blocksparse
  Block-sparse primitives for PyTorch
  ☆147Updated 3 years ago
• HazyResearch / butterfly
  Butterfly matrix multiplication in PyTorch
  ☆160Updated 11 months ago
• owensgroup / merge-spmm
  Code for paper "Design Principles for Sparse Matrix Multiplication on the GPU" accepted to Euro-Par 2018
  ☆70Updated 3 years ago
• nvixnu / pmpp__programming_massively_parallel_processors
  Examples and exercises from the book Programming Massively Parallel Processors - A Hands-on Approach. David B. Kirk and Wen-mei W. Hwu (T…
  ☆33Updated 3 years ago
• PatWie / cuda-design-patterns
  Some CUDA design patterns and a bit of template magic for CUDA
  ☆144Updated last year
• albanD / subclass_zoo
  ☆137Updated 3 months ago
• NVIDIA / Fuser
  A Fusion Code Generator for NVIDIA GPUs (commonly known as "nvFuser")
  ☆250Updated this week
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆125Updated last month
• hgyhungry / ge-spmm
  ☆96Updated 3 years ago
• csc-training / CUDA
  Introduction to CUDA programming
  ☆111Updated 7 years ago
• gevtushenko / matrix_format_performance
  ☆26Updated 4 years ago
• xuqiantong / CUDA-Winograd
  Fast CUDA Kernels for ResNet Inference.
  ☆164Updated 5 years ago
• uwsampl / dtr-prototype
  Dynamic Tensor Rematerialization prototype (modified PyTorch) and simulator. Paper: https://arxiv.org/abs/2006.09616
  ☆129Updated last year
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆118Updated 2 months ago
• daadaada / turingas
  Assembler for NVIDIA Volta and Turing GPUs
  ☆195Updated 2 years ago