LICENSE
README.md
pyproject.toml
setup.cfg
setup.py
sonic_moe.egg-info/PKG-INFO
sonic_moe.egg-info/SOURCES.txt
sonic_moe.egg-info/dependency_links.txt
sonic_moe.egg-info/requires.txt
sonic_moe.egg-info/top_level.txt
sonicmoe/__init__.py
sonicmoe/enums.py
sonicmoe/jit.py
sonicmoe/moe.py
sonicmoe/utils.py
sonicmoe/count_cumsum/__init__.py
sonicmoe/count_cumsum/kernel.cu
sonicmoe/functional/__init__.py
sonicmoe/functional/__init__copy.py
sonicmoe/functional/backward.py
sonicmoe/functional/forward.py
sonicmoe/functional/grouped_gemm.py
sonicmoe/functional/moe_config.py
sonicmoe/functional/reduction_over_k_gather.py
sonicmoe/functional/tile_scheduler.py
sonicmoe/functional/topk_softmax.py
sonicmoe/functional/utils.py
sonicmoe/functional/triton_kernels/__init__.py
sonicmoe/functional/triton_kernels/bitmatrix.py
sonicmoe/include/utils.h
sonicmoe/quack_utils/__init__.py
sonicmoe/quack_utils/gemm_dgated.py
sonicmoe/quack_utils/gemm_gated.py
sonicmoe/quack_utils/gemm_interface.py
tests/test_commons.py