- Only AOT compilation paths ensure "executor" is provided as a Target attribute.

Author: mbs-octoml

python/tvm/relay/build_module.py

@@ -740,13 +740,13 @@ def create_executor(kind="debug", mod=None, device=None, target="llvm", params=N
     if device is not None:
         assert device.device_type == raw_targets[0].kind.device_type
     else:
-        # Use the first target as the device.
+        # Derive the default device from the first target.
         device = _nd.device(raw_targets[0].kind.device_type, 0)
 
     if params is not None:
         mod = IRModule.from_expr(bind_params_by_name(mod["main"], params))
 
-    assert raw_targets[0].attrs.get("executor") == kind
+    assert "executor" not in raw_targets[0].attrs or raw_targets[0].attrs["executor"] == kind
 
     if kind == "debug":
         assert len(raw_targets) == 1, "The interpreter currently only supports a single target"
@@ -756,5 +756,9 @@ def create_executor(kind="debug", mod=None, device=None, target="llvm", params=N
     if kind == "vm":
         return VMExecutor(mod, device, raw_targets)
     if kind == "aot":
+        # The AOT requires the executor as a target attribute.
+        # (The compilation paths for the other executors currently do not always provide this
+        # attribute, hence the above generic assert is more forgiving).
+        assert "executor" in raw_targets[0].attrs
         return AotExecutor(mod, device, raw_targets)
     raise RuntimeError("unknown execution strategy: {0}".format(kind))