shisa-ai
/

megablocks-hip

Model card Files Files and versions

megablocks-hip / torch-ext

252 kB

3 contributors

History: 20 commits

leonardlin's picture

Teach HIP grouped_gemm about autograd

5ce4c31 16 days ago

megablocks
Teach HIP grouped_gemm about autograd 16 days ago
registration.h

259 Bytes

Add ROCm build artifacts and HIP backend 2 months ago
torch_binding.cpp

4.6 kB

Clean ROCm grouped_gemm fallback and add tests about 2 months ago
torch_binding.h

220 Bytes

feat: initial port of megablocks to builder format 5 months ago