From 80ee67c62ed9e5964532785d4aa751a9dc115604 Mon Sep 17 00:00:00 2001 From: Wuwei Lin Date: Wed, 22 Sep 2021 13:13:40 -0400 Subject: [PATCH] Update python/tvm/auto_scheduler/measure.py --- python/tvm/auto_scheduler/measure.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tvm/auto_scheduler/measure.py b/python/tvm/auto_scheduler/measure.py index 42c3b2a607685..8c6fd5f1a9492 100644 --- a/python/tvm/auto_scheduler/measure.py +++ b/python/tvm/auto_scheduler/measure.py @@ -920,7 +920,7 @@ def _timed_eval_func( random_fill(empty_array) loc_args.append(empty_array) else: - loc_args.append(ndarray.array(arg)) + loc_args.append(ndarray.array(args[idx], dev)) dev.sync() costs = time_f(*loc_args).results # pylint: disable=broad-except