dfyz / cutlass_grouped_gemm Public

forked from imoneoi/cutlass_grouped_gemm

Notifications You must be signed in to change notification settings
Fork 0
Star 0

PyTorch bindings for CUTLASS grouped GEMM.

Apache-2.0 license

0 stars 38 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 32 Commits
csrc		csrc
grouped_gemm		grouped_gemm
third_party		third_party
.gitignore		.gitignore
.gitmodules		.gitmodules
LICENSE		LICENSE
MANIFEST.in		MANIFEST.in
Makefile		Makefile
README.md		README.md
setup.py		setup.py

Repository files navigation

Grouped GEMM

A lighweight library exposing grouped GEMM kernels in PyTorch.

Installation

Run pip install grouped_gemm to install the package.

About

PyTorch bindings for CUTLASS grouped GEMM.

Apache-2.0 license

Report repository

Releases

Packages

No packages published

Languages

Cuda 69.1%
Python 29.4%
Other 1.5%