shisa-ai
/

megablocks-hip

Model card Files Files and versions

megablocks-hip / torch-ext

249 kB

3 contributors

History: 19 commits

leonardlin's picture

Clean ROCm grouped_gemm fallback and add tests

aeb3812 about 1 month ago

megablocks
Fix ROCm grouped_gemm accumulation corruption about 1 month ago
registration.h

259 Bytes

Add ROCm build artifacts and HIP backend about 1 month ago
torch_binding.cpp

4.6 kB

Clean ROCm grouped_gemm fallback and add tests about 1 month ago
torch_binding.h

220 Bytes

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