shisa-ai
/

megablocks-hip

Model card Files Files and versions

megablocks-hip / tests

63.7 kB

3 contributors

History: 9 commits

leonardlin's picture

Teach HIP grouped_gemm about autograd

4f9d694 3 months ago

fixtures
feat: validate build with original test suite 8 months ago
layers
feat: add functional version of moe class and layer 7 months ago
ops
feat: validate build with original test suite 8 months ago
__init__.py
0 Bytes

feat: initial port of megablocks to builder format 8 months ago
conftest.py
2.75 kB

feat: validate build with original test suite 8 months ago
layer_test.py
1.61 kB

feat: add functional version of moe class and layer 7 months ago
ops_test.py
5.47 kB

Clean ROCm grouped_gemm fallback and add tests 5 months ago
parallel_layer_test.py
2.54 kB

fix: improve expert parallel implementation and refactors 7 months ago
test_gg.py
3.24 kB

Clean ROCm grouped_gemm fallback and add tests 5 months ago
test_grouped_gemm_autograd.py
1.25 kB

Teach HIP grouped_gemm about autograd 3 months ago
test_mb_moe.py
1.91 kB

feat: validate build with original test suite 8 months ago
test_mb_moe_shared_expert.py
5.03 kB

feat: support shared experts layer and tests 7 months ago
test_mb_moe_shared_expert_multi.py
7.36 kB

feat: support shared experts layer and tests 7 months ago