Function enabled

python/tvm/runtime/module.py:477–500

Whether module runtime is enabled for target Parameters ---------- target : str or Dict[str, Any] or tvm.target.Target The target device type. Returns ------- enabled : bool Whether runtime is enabled. Examples -------- The following code checks

(target)

def enabled(target):
    """Whether module runtime is enabled for target

    Parameters
    ----------
    target : str or Dict[str, Any] or tvm.target.Target
        The target device type.

    Returns
    -------
    enabled : bool
        Whether runtime is enabled.

    Examples
    --------
    The following code checks if gpu is enabled.

    >>> tvm.runtime.enabled("gpu")
    """
    if isinstance(target, dict):
        target = target.get("kind", "")
    elif hasattr(target, "kind"):
        target = target.kind.name
    return _ffi_api.RuntimeEnabled(target)


def num_threads() -> int: