Skip to content

Commit

Permalink
Only test unpacked, C interface API, AOT case
Browse files Browse the repository at this point in the history
Change-Id: I9082ae32079a1a3924c06c7f26c757aafa46dec2
  • Loading branch information
grant-arm committed Sep 1, 2021
1 parent 0f3ad93 commit 8c28118
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 8 deletions.
4 changes: 0 additions & 4 deletions tests/python/relay/aot/aot_test_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -528,12 +528,8 @@ def compile_and_run(

workspace_bytes += extract_main_workspace_size_bytes(base_path)

sanitized_names = []
for key in model.inputs:
sanitized_tensor_name = re.sub(r"\W", "_", key)
if sanitized_tensor_name in sanitized_names:
raise ValueError(f"Sanitized input tensor name clash: {sanitized_tensor_name}")
sanitized_names.append(sanitized_tensor_name)
create_header_file(
f'{mangle_name(model.name, "input_data")}_{sanitized_tensor_name}',
model.inputs[key],
Expand Down
14 changes: 10 additions & 4 deletions tests/python/relay/aot/test_crt_aot.py
Original file line number Diff line number Diff line change
Expand Up @@ -503,10 +503,13 @@ def test_transpose(interface_api, use_unpacked_api, test_runner):
)


@parametrize_aot_options
def test_name_sanitiser(interface_api, use_unpacked_api, test_runner):
def test_name_sanitiser():
"""Test that input tensors with special characters in the name don't break compilation"""

interface_api = "c"
use_unpacked_api = True
test_runner = AOT_DEFAULT_RUNNER

func = relay.var("input-x::2", "float32")
ident = relay.Function([func], func)
one = np.array(1.0, "float32")
Expand All @@ -522,10 +525,13 @@ def test_name_sanitiser(interface_api, use_unpacked_api, test_runner):
)


@parametrize_aot_options
def test_name_sanitiser_name_clash(interface_api, use_unpacked_api, test_runner):
def test_name_sanitiser_name_clash():
"""Test that 2 input tensors with names that clash once sanitized, generates an error"""

interface_api = "c"
use_unpacked_api = True
test_runner = AOT_DEFAULT_RUNNER

dtype = "float32"
x = relay.var("input::-1", shape=(10, 5), dtype=dtype)
# Next 2 input tensor names will clash once sanitized.
Expand Down

0 comments on commit 8c28118

Please sign in to comment.