core.transformer.moe.grouped_gemm_util

Module Contents

Functions

grouped_gemm_is_available	Check if grouped_gemm is available.
assert_grouped_gemm_is_available	Assert that grouped_gemm is available.

Data

ops

API

core.transformer.moe.grouped_gemm_util.grouped_gemm_is_available()

Check if grouped_gemm is available.

core.transformer.moe.grouped_gemm_util.assert_grouped_gemm_is_available()

Assert that grouped_gemm is available.

core.transformer.moe.grouped_gemm_util.ops

None