Skip to content

Commit 31cc299

Browse files
committed
[TVMC] Workspace Pools Parameters
Add test for parameter overrides. Change-Id: I67d5470dcfbfbc9ab27f34e20a9269d2070193ca
1 parent 5b367a6 commit 31cc299

File tree

1 file changed

+27
-0
lines changed

1 file changed

+27
-0
lines changed

tests/python/driver/tvmc/test_workspace_pools.py

Lines changed: 27 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -328,3 +328,30 @@ def test_workspace_pools_recombobulate_multi_target_multi_pool():
328328
assert len(memory_pools.pools[0].target_burst_bytes) == 2
329329
assert memory_pools.pools[0].target_burst_bytes[c_target] == 8
330330
assert memory_pools.pools[0].target_burst_bytes[llvm_target] == 4
331+
332+
333+
def test_workspace_pools_recombobulate_parameter_overrides():
334+
parser = argparse.ArgumentParser()
335+
generate_workspace_pools_args(parser)
336+
parsed, _ = parser.parse_known_args(
337+
[
338+
"--workspace-pools=sram",
339+
"--workspace-pools-target-access=sram:c:rw",
340+
"--workspace-pools-size-hint-bytes=sram:800",
341+
"--workspace-pools-size-hint-bytes=sram:400",
342+
"--workspace-pools-clock-frequency-hz=sram:4000000",
343+
"--workspace-pools-clock-frequency-hz=sram:3600000",
344+
]
345+
)
346+
347+
c_target = Target("c")
348+
349+
targets = [c_target]
350+
memory_pools = workspace_pools_recombobulate(parsed, targets)
351+
352+
assert len(memory_pools.pools) == 1
353+
354+
assert len(memory_pools.pools[0].target_access) == 1
355+
assert memory_pools.pools[0].target_access[c_target] == PoolInfo.READ_WRITE_ACCESS
356+
assert memory_pools.pools[0].size_hint_bytes == 400
357+
assert memory_pools.pools[0].clock_frequency_hz == 3600000

0 commit comments

Comments
 (0)