diff --git a/python/tvm/utils/roofline.py b/python/tvm/utils/roofline.py index a273309eb9f47..45a88b2f4c704 100644 --- 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)