bagua.torch_api.model_parallel.moe.utils

Module Contents

bagua.torch_api.model_parallel.moe.utils.is_moe_param(param)

Parameters:

param (torch.Tensor) –

Return type:

bool