shisa-ai
/

megablocks-hip

Model card Files Files and versions

megablocks-hip / torch-ext

252 kB

Ctrl+K

Ctrl+K

3 contributors

History: 20 commits

leonardlin's picture

Teach HIP grouped_gemm about autograd

5ce4c31 9 months ago

megablocks
Teach HIP grouped_gemm about autograd 9 months ago
registration.h

259 Bytes
Add ROCm build artifacts and HIP backend 11 months ago
torch_binding.cpp

4.6 kB
Clean ROCm grouped_gemm fallback and add tests 11 months ago
torch_binding.h

220 Bytes
feat: initial port of megablocks to builder format about 1 year ago