kernels-community
/

megablocks

Model card Files Files and versions

megablocks / tests

Ctrl+K

Ctrl+K

2 contributors

History: 7 commits

drbh

feat: support shared experts layer and tests

89e2950 14 days ago

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

0 Bytes

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

2.75 kB

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

1.61 kB

feat: add functional version of moe class and layer about 1 month ago
ops_test.py

5.46 kB

fix: bump builds about 2 months ago
parallel_layer_test.py

2.54 kB

fix: improve expert parallel implementation and refactors about 1 month ago
test_gg.py

1.63 kB

feat: vendor grouped gemm about 2 months ago
test_mb_moe.py

1.91 kB

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

5.03 kB

feat: support shared experts layer and tests 14 days ago
test_mb_moe_shared_expert_multi.py

7.36 kB

feat: support shared experts layer and tests 14 days ago