openchatdev / cublas_sm90_grouped_gemm Public

forked from tgale96/grouped_gemm

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

[For SM90 and cuBLAS] PyTorch bindings for CUTLASS grouped GEMM.

Apache-2.0 license

1 star 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

[For SM90 and cuBLAS] PyTorch bindings for CUTLASS grouped GEMM.

Apache-2.0 license

Custom properties

Report repository

Releases

No releases published

Packages

No packages published

Languages

Cuda 64.2%
Python 33.8%
C++ 1.1%
Makefile 0.9%