diff --git a/tests/python/unittest/test_runtime_module_export.py b/tests/python/unittest/test_runtime_module_export.py index 6a8b45172694..9ea1ff437f61 100644 --- a/tests/python/unittest/test_runtime_module_export.py +++ b/tests/python/unittest/test_runtime_module_export.py @@ -74,10 +74,10 @@ def verify_gpu_mod_export(obj_format): synthetic_llvm_mod, synthetic_llvm_params = relay.testing.synthetic.get_workload() with tvm.transform.PassContext(opt_level=3): _, synthetic_gpu_lib, _ = relay.build_module.build( - synthetic_mod, "cuda", params=synthetic_params + synthetic_mod, "cuda", params=synthetic_params, mod_name="cudalib" ) _, synthetic_llvm_cpu_lib, _ = relay.build_module.build( - synthetic_llvm_mod, "llvm", params=synthetic_llvm_params + synthetic_llvm_mod, "llvm", params=synthetic_llvm_params, mod_name="llvmlib" ) from tvm.contrib import utils