e6d53185b96cc39f2aaec5e86ae11ca0ac675b8a,python/tvm/auto_scheduler/workload_registry.py,,workload_key_to_tensors,#Any#,164

Before Change

    value = WORKLOAD_FUNC_REGISTRY[name]

    // "value" can be either a function or a list of tensors
    if callable(value):  // if it is a func
        args = deserialize_args(workload[1:])
        return value(*args)
    // otherwise, it is a list of tensors
    return value


def serialize_workload_registry_entry(workload_key):

After Change

    workload = json.loads(workload_key)
    name = workload[0]
    value = WORKLOAD_FUNC_REGISTRY[name]
    assert callable(value)

    args = deserialize_args(workload[1:])
    return value(*args)

Instances