tmoreau89 commented on a change in pull request #6126:
URL: https://github.com/apache/incubator-tvm/pull/6126#discussion_r477022969
##########
File path: python/tvm/autotvm/task/topi_integration.py
##########
@@ -227,17 +227,21 @@ def wrapper(outs, *args, **kwargs):
return _decorate
-def get_workload(outs):
+def get_workload(outs, task_name=None):
"""Retrieve the workload from outputs"""
def traverse(tensors):
"""traverse all ops to find attached workload"""
for t in tensors:
op = t.op
- if 'workload' in op.attrs:
- return args_to_workload(op.attrs['workload'])
wkl = traverse(op.input_tensors)
if wkl:
return wkl
+
Review comment:
Thanks for clarifying. How do we guard against extracting add as a
standalone op for other backends int his case?
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
[email protected]
