[TEST] Fix duplicate definition error for gpu export mod testcase (ap…

…ache#9538)
masahi · Nov 19, 2021 · 454e956 · 454e956
1 parent 53cb5ea
commit 454e956
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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