You are viewing a plain text version of this content. The canonical link for it is here.

Posted to commits@tvm.apache.org by GitBox <gi...@apache.org> on 2021/02/03 13:23:07 UTC

[GitHub] [tvm] jcf94 commented on a change in pull request #7313: [AutoSchedule] Sparse dense tuning support with custom sketch rule

jcf94 commented on a change in pull request #7313:
URL: https://github.com/apache/tvm/pull/7313#discussion_r569407977



##########
File path: python/tvm/auto_scheduler/measure.py
##########
@@ -1132,3 +1249,44 @@ def rpc_runner_run(
         print("")
 
     return results
+
+
+# The map stores special registered buffer for measurement
+#  This can be used for sparse workloads when we cannot use random tensors for measurment.
+global special_buffer_table
+special_buffer_table = {}

Review comment:
       Please have another look~ 😄 
   I've moved the input data to serialize in the SearchTask. (Even though we may still have to get a global table in the search_task.py)




----------------------------------------------------------------
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:
users@infra.apache.org