tvm-commits mailing list archives

Site index · List index
Message view « Date » · « Thread »
Top « Date » · « Thread »
From GitBox <...@apache.org>
Subject [GitHub] [incubator-tvm] FrozenGene commented on a change in pull request #5200: Fix intel conv2d auto tune
Date Thu, 02 Apr 2020 02:14:31 GMT
FrozenGene commented on a change in pull request #5200: Fix intel conv2d auto tune
URL: https://github.com/apache/incubator-tvm/pull/5200#discussion_r402013632
 
 

 ##########
 File path: python/tvm/autotvm/measure/measure_methods.py
 ##########
 @@ -479,9 +479,10 @@ def run_through_rpc(measure_input, build_result,
         if ref_input:
             args = [nd.array(x, ctx=ctx) for x in ref_input]
         else:
-            # create empty arrays on the remote device and copy them once.
+            # create random arrays on the remote device and copy them once.
             # This can avoid some memory issues that make the measurement results unreliable.
-            args = [nd.empty(x[0], dtype=x[1], ctx=ctx) for x in build_result.arg_info]
+            args = [nd.array(np.random.uniform(0.0, 255.0, size=x[0]).astype(dtype=x[1]),
ctx=ctx)
 
 Review comment:
   @merrymercy Sure. I will port it to our upstream soon.

----------------------------------------------------------------
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


With regards,
Apache Git Services

Mime
View raw message