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 16 days ago

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

0 Bytes

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

2.75 kB

feat: validate build with original test suite 5 months ago
layer_test.py

1.61 kB

feat: add functional version of moe class and layer 5 months ago
ops_test.py

5.47 kB

Clean ROCm grouped_gemm fallback and add tests about 2 months ago
parallel_layer_test.py

2.54 kB

fix: improve expert parallel implementation and refactors 5 months ago
test_gg.py

3.24 kB

Clean ROCm grouped_gemm fallback and add tests about 2 months ago
test_grouped_gemm_autograd.py

1.25 kB

Teach HIP grouped_gemm about autograd 16 days ago
test_mb_moe.py

1.91 kB

feat: validate build with original test suite 5 months ago
test_mb_moe_shared_expert.py

5.03 kB

feat: support shared experts layer and tests 4 months ago
test_mb_moe_shared_expert_multi.py

7.36 kB

feat: support shared experts layer and tests 4 months ago