forgot to include remote

apache · May 10, 2022 · 4be1d83 · 4be1d83
1 parent 4e9e05f
commit 4be1d83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/tvm/utils/roofline.py b/python/tvm/utils/roofline.py
@@ -481,7 +481,7 @@ def roofline_analysis(
         lib = remote.load_module("roofline_lib.tar")
     vmexec = profiler_vm.VirtualMachineProfiler(lib, dev)
 
-    args = _create_args(mod, dev)
+    args = _create_args(mod, dev, remote=remote)
     report = vmexec.profile(*args)
 
     return roofline_from_existing(report, save_tir.functions, target, dev, remote=remote)