tvm_tirx.target.register_tag#

tvm_tirx.target.register_tag(name, config, override=False)#

Add a user-defined tag into the target tag registry.

Parameters:

name (str) – Name of the target, e.g. “nvidia/gtx1080ti”
config (Dict[str, Any]) – The config dict used to create the target
override (bool) – A boolean flag indicating if overriding existing tags are allowed. If False and the tag has been registered already, an exception will be thrown.

Returns:

target (Optional[Target]) – The target corresponding to the tag None if TVM is built in runtime-only mode.

Return type:

Target | None

Examples

register_tag("nvidia/gtx1080ti", config={
    "kind": "cuda",
    "arch": "sm_61",
})

tvm_tirx.target.register_tag