[TESTS] Triage the testcases to fit the the new namespaces (#5071)
authorTianqi Chen <tqchen@users.noreply.github.com>
Sun, 15 Mar 2020 04:40:50 +0000 (21:40 -0700)
committerGitHub <noreply@github.com>
Sun, 15 Mar 2020 04:40:50 +0000 (21:40 -0700)
* [TESTS] Triage the testcases to fit the naming convention of the new namespaces

* Remove multiple usage of system lib to avoid test problems

69 files changed:
tests/python/unittest/test_autotvm_graph_tuner_core.py [moved from tests/python/unittest/test_graph_tuner_core.py with 100% similarity]
tests/python/unittest/test_autotvm_graph_tuner_utils.py [moved from tests/python/unittest/test_graph_tuner_utils.py with 100% similarity]
tests/python/unittest/test_ir_container.py [moved from tests/python/unittest/test_lang_container.py with 100% similarity]
tests/python/unittest/test_runtime_container.py [moved from tests/python/unittest/test_container.py with 100% similarity]
tests/python/unittest/test_target_codegen_arm.py [moved from tests/python/unittest/test_codegen_arm.py with 100% similarity]
tests/python/unittest/test_target_codegen_blob.py [moved from tests/python/unittest/test_codegen_blob.py with 93% similarity]
tests/python/unittest/test_target_codegen_bool.py [moved from tests/python/unittest/test_codegen_bool.py with 100% similarity]
tests/python/unittest/test_target_codegen_c_host.py [moved from tests/python/unittest/test_codegen_c_host.py with 100% similarity]
tests/python/unittest/test_target_codegen_cross_llvm.py [moved from tests/python/unittest/test_codegen_cross_llvm.py with 100% similarity]
tests/python/unittest/test_target_codegen_cuda.py [moved from tests/python/unittest/test_codegen_cuda.py with 100% similarity]
tests/python/unittest/test_target_codegen_device.py [moved from tests/python/unittest/test_codegen_device.py with 100% similarity]
tests/python/unittest/test_target_codegen_extern.py [moved from tests/python/unittest/test_codegen_extern.py with 100% similarity]
tests/python/unittest/test_target_codegen_llvm.py [moved from tests/python/unittest/test_codegen_llvm.py with 100% similarity]
tests/python/unittest/test_target_codegen_opencl.py [moved from tests/python/unittest/test_codegen_opencl.py with 100% similarity]
tests/python/unittest/test_target_codegen_rocm.py [moved from tests/python/unittest/test_codegen_rocm.py with 100% similarity]
tests/python/unittest/test_target_codegen_static_init.py [moved from tests/python/unittest/test_codegen_static_init.py with 100% similarity]
tests/python/unittest/test_target_codegen_vm_basic.py [moved from tests/python/unittest/test_codegen_vm_basic.py with 100% similarity]
tests/python/unittest/test_target_codegen_vulkan.py [moved from tests/python/unittest/test_codegen_vulkan.py with 100% similarity]
tests/python/unittest/test_target_codegen_x86.py [moved from tests/python/unittest/test_codegen_x86.py with 100% similarity]
tests/python/unittest/test_target_custom_datatypes.py [moved from tests/python/unittest/test_custom_datatypes_mybfloat16.py with 100% similarity]
tests/python/unittest/test_target_target.py [moved from tests/python/unittest/test_lang_target.py with 100% similarity]
tests/python/unittest/test_te_build_lower.py [moved from tests/python/unittest/test_build_lower.py with 100% similarity]
tests/python/unittest/test_te_group.py [moved from tests/python/unittest/test_lang_group.py with 100% similarity]
tests/python/unittest/test_te_schedule.py [moved from tests/python/unittest/test_lang_schedule.py with 100% similarity]
tests/python/unittest/test_te_schedule_bound_inference.py [moved from tests/python/unittest/test_schedule_bound_inference.py with 100% similarity]
tests/python/unittest/test_te_schedule_graph.py [moved from tests/python/unittest/test_schedule_graph.py with 100% similarity]
tests/python/unittest/test_te_schedule_lstm.py [moved from tests/python/unittest/test_schedule_lstm.py with 100% similarity]
tests/python/unittest/test_te_schedule_ops.py [moved from tests/python/unittest/test_schedule_schedule_ops.py with 100% similarity]
tests/python/unittest/test_te_schedule_tensor_core.py [moved from tests/python/unittest/test_schedule_tensor_core.py with 100% similarity]
tests/python/unittest/test_te_schedule_tensorize.py [moved from tests/python/unittest/test_schedule_tensorize.py with 100% similarity]
tests/python/unittest/test_te_tag.py [moved from tests/python/unittest/test_lang_tag.py with 100% similarity]
tests/python/unittest/test_te_tensor.py [moved from tests/python/unittest/test_lang_tensor.py with 100% similarity]
tests/python/unittest/test_te_tensor_overload.py [moved from tests/python/unittest/test_lang_tensor_overload_op.py with 100% similarity]
tests/python/unittest/test_te_verify_compute.py [moved from tests/python/unittest/test_lang_verify_compute.py with 100% similarity]
tests/python/unittest/test_tir_buffer.py [moved from tests/python/unittest/test_lang_buffer.py with 100% similarity]
tests/python/unittest/test_tir_constructor.py [moved from tests/python/unittest/test_lang_constructor.py with 100% similarity]
tests/python/unittest/test_tir_data_layout.py [moved from tests/python/unittest/test_lang_data_layout.py with 100% similarity]
tests/python/unittest/test_tir_intrin.py [moved from tests/python/unittest/test_tvm_intrin.py with 100% similarity]
tests/python/unittest/test_tir_ir_builder.py [moved from tests/python/unittest/test_ir_builder.py with 100% similarity]
tests/python/unittest/test_tir_ops.py [moved from tests/python/unittest/test_lang_operator.py with 100% similarity]
tests/python/unittest/test_tir_pass_attrs_hash_equal.py [moved from tests/python/unittest/test_pass_attrs_hash_equal.py with 100% similarity]
tests/python/unittest/test_tir_pass_basic.py [moved from tests/python/unittest/test_pass_basic.py with 100% similarity]
tests/python/unittest/test_tir_pass_bound_checkers.py [moved from tests/python/unittest/test_pass_bound_checkers.py with 100% similarity]
tests/python/unittest/test_tir_pass_combine_context_call.py [moved from tests/python/unittest/test_pass_combine_context_call.py with 100% similarity]
tests/python/unittest/test_tir_pass_decorate_device_scope.py [moved from tests/python/unittest/test_pass_decorate_device_scope.py with 100% similarity]
tests/python/unittest/test_tir_pass_equal.py [moved from tests/python/unittest/test_pass_equal.py with 100% similarity]
tests/python/unittest/test_tir_pass_hoist_if.py [moved from tests/python/unittest/test_pass_hoist_if.py with 100% similarity]
tests/python/unittest/test_tir_pass_inject_copy_intrin.py [moved from tests/python/unittest/test_pass_inject_copy_intrin.py with 100% similarity]
tests/python/unittest/test_tir_pass_inject_double_buffer.py [moved from tests/python/unittest/test_pass_inject_double_buffer.py with 100% similarity]
tests/python/unittest/test_tir_pass_inject_vthread.py [moved from tests/python/unittest/test_pass_inject_vthread.py with 100% similarity]
tests/python/unittest/test_tir_pass_inline.py [moved from tests/python/unittest/test_pass_inline.py with 100% similarity]
tests/python/unittest/test_tir_pass_ir_transform.py [moved from tests/python/unittest/test_pass_ir_transform.py with 100% similarity]
tests/python/unittest/test_tir_pass_lift_attr_scope.py [moved from tests/python/unittest/test_pass_lift_attr_scope.py with 100% similarity]
tests/python/unittest/test_tir_pass_loop_partition.py [moved from tests/python/unittest/test_pass_loop_partition.py with 100% similarity]
tests/python/unittest/test_tir_pass_lower_intrin.py [moved from tests/python/unittest/test_pass_lower_intrin.py with 100% similarity]
tests/python/unittest/test_tir_pass_lower_warp_memory.py [moved from tests/python/unittest/test_pass_lower_warp_memory.py with 100% similarity]
tests/python/unittest/test_tir_pass_makeapi.py [moved from tests/python/unittest/test_pass_makeapi.py with 100% similarity]
tests/python/unittest/test_tir_pass_remove_no_op.py [moved from tests/python/unittest/test_pass_remove_no_op.py with 100% similarity]
tests/python/unittest/test_tir_pass_rewrite_for_tensor_core.py [moved from tests/python/unittest/test_pass_rewrite_for_tensor_core.py with 100% similarity]
tests/python/unittest/test_tir_pass_rewrite_unsafe_select.py [moved from tests/python/unittest/test_pass_rewrite_unsafe_select.py with 100% similarity]
tests/python/unittest/test_tir_pass_split_host_device.py [moved from tests/python/unittest/test_pass_split_host_device.py with 100% similarity]
tests/python/unittest/test_tir_pass_storage_flatten.py [moved from tests/python/unittest/test_pass_storage_flatten.py with 100% similarity]
tests/python/unittest/test_tir_pass_storage_rewrite.py [moved from tests/python/unittest/test_pass_storage_rewrite.py with 100% similarity]
tests/python/unittest/test_tir_pass_storage_sync.py [moved from tests/python/unittest/test_pass_storage_sync.py with 100% similarity]
tests/python/unittest/test_tir_pass_unroll.py [moved from tests/python/unittest/test_pass_unroll.py with 100% similarity]
tests/python/unittest/test_tir_pass_vectorize.py [moved from tests/python/unittest/test_pass_vectorize.py with 100% similarity]
tests/python/unittest/test_tir_pass_verify_gpu_code.py [moved from tests/python/unittest/test_pass_verify_gpu_code.py with 100% similarity]
tests/python/unittest/test_tir_pass_verify_memory.py [moved from tests/python/unittest/test_pass_verify_memory.py with 100% similarity]
tests/python/unittest/test_tir_pass_virtual_thread.py [moved from tests/python/unittest/test_pass_virtual_thread.py with 100% similarity]

@@ -68,7 +68,7 @@ def test_resnet18():
     tvm.testing.assert_allclose(out, verify(data), atol=1e-5)
 
 
-def test_system_lib():
+def test_cuda_lib():
     ctx = tvm.gpu(0)
     for device in ["llvm", "cuda"]:
         if not tvm.runtime.enabled(device):
@@ -85,15 +85,10 @@ def test_system_lib():
 
     from tvm.contrib import util
     temp = util.tempdir()
-    fn_add = tvm.build(s, [A, B], target="cuda", target_host="llvm -system-lib", name="add")
-    path_obj = temp.relpath("add.o")
+    fn_add = tvm.build(s, [A, B], target="cuda", target_host="llvm", name="add")
     path_lib = temp.relpath("deploy_lib.so")
-    fn_add.save(path_obj)
     fn_add.export_library(path_lib)
-    # Load dll, will trigger system library registration
-    dll = ctypes.CDLL(path_lib)
-    # Load the system wide library
-    m = tvm.runtime.system_lib()
+    m = tvm.runtime.load_module(path_lib)
     a = tvm.nd.array(np.random.uniform(size=nn).astype(A.dtype), ctx)
     b = tvm.nd.array(np.zeros(nn, dtype=A.dtype), ctx)
     m['add'](a, b)