def METHOD_NAME(features, out_features, out_bp, indice_pairs,
                            indice_pair_num):
    if features.dtype == torch.float32 or features.dtype == torch.half:
        return ext_module.METHOD_NAME(features, out_features,
                                                  out_bp, indice_pairs,
                                                  indice_pair_num)
    else:
        raise NotImplementedError