diff --git a/python/tvm/driver/tvmc/composite_target.py b/python/tvm/driver/tvmc/composite_target.py index 7c08994e0e75..0a2592685646 100644 --- a/python/tvm/driver/tvmc/composite_target.py +++ b/python/tvm/driver/tvmc/composite_target.py @@ -32,7 +32,7 @@ # to be used in the PassContext (if any), and a function # responsible for partitioning to that target. REGISTERED_CODEGEN = { - "acl": { + "compute-library": { "config_key": None, "pass_pipeline": partition_for_arm_compute_lib, }, diff --git a/tests/python/driver/tvmc/test_common.py b/tests/python/driver/tvmc/test_common.py index 253f32d3f0aa..b272ceccea39 100644 --- a/tests/python/driver/tvmc/test_common.py +++ b/tests/python/driver/tvmc/test_common.py @@ -258,10 +258,10 @@ def test_parse_single_target_with_opts(): def test_parse_multiple_target(): - targets = tvmc.common.parse_target("acl, llvm -device=arm_cpu --system-lib") + targets = tvmc.common.parse_target("compute-library, llvm -device=arm_cpu --system-lib") assert len(targets) == 2 - assert "acl" == targets[0]["name"] + assert "compute-library" == targets[0]["name"] assert "llvm" == targets[1]["name"] diff --git a/tests/python/driver/tvmc/test_composite_target.py b/tests/python/driver/tvmc/test_composite_target.py index eda7cd9224fd..cef8b117d989 100644 --- a/tests/python/driver/tvmc/test_composite_target.py +++ b/tests/python/driver/tvmc/test_composite_target.py @@ -38,7 +38,7 @@ def test_get_codegen_names(): def test_valid_codegen(): - codegen = tvmc.composite_target.get_codegen_by_target("acl") + codegen = tvmc.composite_target.get_codegen_by_target("compute-library") assert codegen is not None assert codegen["pass_pipeline"] is not None