Function topk_loader

imperative/python/megengine/traced_module/compat.py:231–246

(expr)

@register_functional_loader(("megengine.functional.math", "topk"))
def topk_loader(expr):
    import pkg_resources as pkg

    if not hasattr(expr, "version") or pkg.parse_version(
        expr.version
    ) <= pkg.parse_version("1.12.0"):

        def origin_topk_signature(
            inp, k, descending=False, kth_only=False, no_sort=False
        ):
            pass

        args, kwargs = _convert_kwargs_to_args(
            origin_topk_signature, expr.args, expr.kwargs
        )
        expr.set_args_kwargs(*args, **kwargs)


@register_functional_loader(("megengine.functional.tensor", "arange"))