shisa-ai
/

megablocks-hip

Model card Files Files and versions

megablocks-hip / tests

63.7 kB

Ctrl+K

Ctrl+K

3 contributors

History: 9 commits

leonardlin's picture

Teach HIP grouped_gemm about autograd

4f9d694 7 months ago

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

0 Bytes
feat: initial port of megablocks to builder format 12 months ago
conftest.py

2.75 kB
feat: validate build with original test suite 12 months ago
layer_test.py

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

5.47 kB
Clean ROCm grouped_gemm fallback and add tests 8 months ago
parallel_layer_test.py

2.54 kB
fix: improve expert parallel implementation and refactors 11 months ago
test_gg.py

3.24 kB
Clean ROCm grouped_gemm fallback and add tests 8 months ago
test_grouped_gemm_autograd.py

1.25 kB
Teach HIP grouped_gemm about autograd 7 months ago
test_mb_moe.py

1.91 kB
feat: validate build with original test suite 12 months ago
test_mb_moe_shared_expert.py

5.03 kB
feat: support shared experts layer and tests 10 months ago
test_mb_moe_shared_expert_multi.py

7.36 kB
feat: support shared experts layer and tests 10 months ago