From dd1549ecdd26db0d78d728f3d83b5e18eb96e79c Mon Sep 17 00:00:00 2001 From: Leandro Nunes Date: Wed, 24 Feb 2021 18:06:41 +0000 Subject: [PATCH] [TVMC] rename composite target "acl" to "compute-library" (#7508) * Renames the "acl" composite target to point to the specific library it represents --- python/tvm/driver/tvmc/composite_target.py | 2 +- tests/python/driver/tvmc/test_common.py | 4 ++-- tests/python/driver/tvmc/test_composite_target.py | 2 +- 3 files changed, 4 insertions(+), 4 deletions(-) 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