Alternatives and similar repositories for pytorch_cmake_example

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

• google-research / sputnik
  A library of GPU kernels for sparse matrix operations.
  ☆275Updated 4 years ago
• wzsh / wmma_tensorcore_sample
  Matrix Multiply-Accumulate with CUDA and WMMA( Tensor Core)
  ☆145Updated 5 years ago
• pytorch / helion
  A Python-embedded DSL that makes it easy to write fast, scalable ML kernels with minimal boilerplate.
  ☆585Updated this week
• wangzyon / NVIDIA_SGEMM_PRACTICE
  Step-by-step optimization of CUDA SGEMM
  ☆390Updated 3 years ago
• leimao / CUTLASS-Examples
  CUTLASS and CuTe Examples
  ☆98Updated 3 weeks ago
• leimao / CUDA-GEMM-Optimization
  CUDA Matrix Multiplication Optimization
  ☆235Updated last year
• ColfaxResearch / cutlass-kernels
  ☆243Updated last year
• wmmae / wmma_extension
  An extension library of WMMA API (Tensor Core API)
  ☆108Updated last year
• GPUPeople / spECK
  Efficient SpGEMM on GPU using CUDA and CSR
  ☆57Updated 2 years ago
• NVIDIA / nsight-training
  Training material for Nsight developer tools
  ☆170Updated last year
• sunlex0717 / DissectingTensorCores
  ☆109Updated last year
• anilshanbhag / gpu-topk
  Efficient Top-K implementation on the GPU
  ☆187Updated 6 years ago
• Karbo123 / pytorch_grouped_gemm
  High Performance Grouped GEMM in PyTorch
  ☆31Updated 3 years ago
• ColfaxResearch / cfx-article-src
  ☆154Updated 6 months ago
• TiledTensor / TiledCUDA
  We invite you to visit and follow our new repository at https://github.com/microsoft/TileFusion. TiledCUDA is a highly efficient kernel …
  ☆186Updated 9 months ago
• daadaada / turingas
  Assembler for NVIDIA Volta and Turing GPUs
  ☆231Updated 3 years ago
• albanD / subclass_zoo
  ☆180Updated last year
• meta-pytorch / tritonbench
  Tritonbench is a collection of PyTorch custom operators with example inputs to measure their performance.
  ☆277Updated this week
• microsoft / TileFusion
  TileFusion is an experimental C++ macro kernel template library that elevates the abstraction level in CUDA C for tile processing.
  ☆100Updated 4 months ago
• cmpute / pytorch-cmake-example
  Example to build PyTorch CUDA extension using CMake (with pybind11 and scikit-build)
  ☆12Updated 5 years ago
• roastduck / FreeTensor
  A language and compiler for irregular tensor programs.
  ☆150Updated 11 months ago
• pranjalssh / fast.cu
  Fastest kernels written from scratch
  ☆386Updated last month
• NVIDIA / nvshmem
  NVIDIA NVSHMEM is a parallel programming interface for NVIDIA GPUs based on OpenSHMEM. NVSHMEM can significantly reduce multi-process com…
  ☆369Updated 3 weeks ago
• UDC-GAC / venom
  A Vectorized N:M Format for Unleashing the Power of Sparse Tensor Cores
  ☆54Updated last year
• NVIDIA / compute-sanitizer-samples
  Samples demonstrating how to use the Compute Sanitizer Tools and Public API
  ☆89Updated 2 years ago
• yifuwang / symm-mem-recipes
  ☆146Updated 10 months ago
• ColfaxResearch / layout-categories
  This repository contains companion software for the Colfax Research paper "Categorical Foundations for CuTe Layouts".
  ☆74Updated last month
• bertmaher / simplegemm
  ☆123Updated 2 weeks ago
• Dao-AILab / quack
  A Quirky Assortment of CuTe Kernels
  ☆651Updated last week
• parasj / checkmate
  Training neural networks in TensorFlow 2.0 with 5x less memory
  ☆136Updated 3 years ago