diff --git a/python/tvm/relay/backend/vm.py b/python/tvm/relay/backend/vm.py index e470540dceda..75a11b31e628 100644 --- a/python/tvm/relay/backend/vm.py +++ b/python/tvm/relay/backend/vm.py @@ -1,4 +1,4 @@ -# License .to the Apache Software Foundation (ASF) under one +# Licensed to the Apache Software Foundation (ASF) under one # or more contributor license agreements. See the NOTICE file # distributed with this work for additional information # regarding copyright ownership. The ASF licenses this file @@ -73,6 +73,7 @@ def compile(mod, target=None, target_host=None, params=None): class VMCompiler(object): """Compiler that compiles Relay module to VM executable.""" + def __init__(self): self.mod = _vm._VMCompiler() self._lower = self.mod["lower"] @@ -239,6 +240,7 @@ class VMExecutor(Executor): target : :py:class:`Target` The target option to build the function. """ + def __init__(self, mod, ctx, target): if mod is None: raise RuntimeError("Must provide module to get VM executor.") diff --git a/python/tvm/runtime/profiler_vm.py b/python/tvm/runtime/profiler_vm.py index 11cedd5fa200..9d604833b83c 100644 --- a/python/tvm/runtime/profiler_vm.py +++ b/python/tvm/runtime/profiler_vm.py @@ -1,4 +1,4 @@ -# License .to the Apache Software Foundation (ASF) under one +# Licensed to the Apache Software Foundation (ASF) under one # or more contributor license agreements. See the NOTICE file # distributed with this work for additional information # regarding copyright ownership. The ASF licenses this file @@ -23,12 +23,15 @@ from tvm.runtime import _ffi_api from . import vm + def enabled(): """Whether vm profiler is enabled.""" return hasattr(_ffi_api, "_VirtualMachineDebug") + class VirtualMachineProfiler(vm.VirtualMachine): """Relay profile VM runtime.""" + def __init__(self, mod): super(VirtualMachineProfiler, self).__init__(mod) m = mod.module if isinstance(mod, vm.Executable) else mod diff --git a/python/tvm/runtime/vm.py b/python/tvm/runtime/vm.py index 8a85051d6bbd..d7d945153d57 100644 --- a/python/tvm/runtime/vm.py +++ b/python/tvm/runtime/vm.py @@ -1,4 +1,4 @@ -# License .to the Apache Software Foundation (ASF) under one +# Licensed to the Apache Software Foundation (ASF) under one # or more contributor license agreements. See the NOTICE file # distributed with this work for additional information # regarding copyright ownership. The ASF licenses this file @@ -28,6 +28,7 @@ from .object import Object from . import _ffi_api, container + def _convert(arg, cargs): if isinstance(arg, Object): cargs.append(arg) @@ -59,6 +60,7 @@ def convert(args): class Executable(object): """Relay VM executable""" + def __init__(self, mod): self.mod = mod self._function_params = {} @@ -272,6 +274,7 @@ def get_function_params(self, func_name): class VirtualMachine(object): """Relay VM runtime.""" + def __init__(self, mod): if not isinstance(mod, (Executable, tvm.runtime.Module)): raise TypeError("mod is expected to be the type of Executable or " +