Repository navigation

cublas_sm90_grouped_gemm

Public

forked from tgale96/grouped_gemm

About

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

Custom properties

1 star

0 watching

Report repository

Name		Name	Last commit message	Last commit date
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.

Custom properties

1 star

0 watching

Report repository

Releases

Packages

Contributors

Languages