Hugging Face
Models
Datasets
Spaces
Community
Docs
Enterprise
Pricing
Log In
Sign Up
shisa-ai
/
megablocks-hip
like
1
Follow
Shisa.AI
87
kernel
License:
apache-2.0
Model card
Files
Files and versions
xet
Community
1
main
megablocks-hip
/
torch-ext
/
megablocks
247 kB
3 contributors
History:
18 commits
leonardlin
Teach HIP grouped_gemm about autograd
5ce4c31
about 1 month ago
_layers
fix: fully vendor stk and fix imports
5 months ago
backend
fix: add quickstart and avoid autotune when no cuda
4 months ago
grouped_gemm
Teach HIP grouped_gemm about autograd
about 1 month ago
ops
fix: fully vendor stk and fix imports
5 months ago
stk
fix: fully vendor stk and fix imports
5 months ago
__init__.py
Safe
5.13 kB
feat: add functional version of moe class and layer
5 months ago
_version.py
Safe
126 Bytes
feat: initial port of megablocks to builder format
6 months ago
bak.__init__.py
Safe
363 Bytes
feat: initial port of megablocks to builder format
6 months ago
benchmark_util.py
Safe
880 Bytes
feat: initial port of megablocks to builder format
6 months ago
grouped_gemm_util.py
Safe
807 Bytes
feat: vendor grouped gemm
6 months ago
layers.py
41.8 kB
fix: include torch compile flag
4 months ago
py.typed
Safe
0 Bytes
feat: initial port of megablocks to builder format
6 months ago
