mbs-octoml commented on code in PR #11770:
URL: https://github.com/apache/tvm/pull/11770#discussion_r907628048
##########
python/tvm/relay/build_module.py:
##########
@@ -607,16 +607,16 @@ class AotExecutor(_interpreter.Executor):
device : :py:class:`Device`
The runtime device to run the code on.
- target : :py:class:`Target`
- The target option to build the function.
+ raw_targets : Array[tvm.target.Target]
+ The available targets.
"""
- def __init__(self, mod, device, target):
+ def __init__(self, mod, device, raw_targets):
assert mod is not None
self.mod = mod
self.device = device
- self.target = target
- assert target.attrs.get("executor", "graph") == "aot"
+ self.raw_targets = raw_targets
+ assert raw_targets[0].attrs.get("executor", "graph") == "aot"
Review Comment:
See following comment.
--
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.
To unsubscribe, e-mail: [email protected]
For queries about this service, please contact Infrastructure at:
[email protected]
