Skip to content

Commit

Permalink
[microNPU] Fix flaky compute cycle annotation test (#11510)
Browse files Browse the repository at this point in the history
Fixes non-deterministic test by disabling striping when running
the cascader.

Change-Id: Ib44f299f21fa0b41be4bfac3deb61a9c16818c58
  • Loading branch information
lhutton1 committed May 31, 2022
1 parent c1b22ee commit ac5d781
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions tests/python/contrib/test_ethosu/cascader/test_scheduler.py
Original file line number Diff line number Diff line change
Expand Up @@ -48,7 +48,6 @@ def test_cascade(SRAM, FLASH, TwoConv2DWithSliceTE, TwoConv2DTE, MobileNetv1Star
cs.cascade(sch, te_graph, const_dict, options, SRAM, FLASH, [SRAM], device_config)


@pytest.mark.skip(reason="See https://github.com/apache/tvm/issues/11483")
def test_compute_cycles_annotation(SRAM, FLASH, TwoConv2DTE):
device_config = cs.EthosuDeviceConfig("ethos-u55-256")
options = infra.make_options(
Expand All @@ -61,6 +60,7 @@ def test_compute_cycles_annotation(SRAM, FLASH, TwoConv2DTE):
always_copy_size=1024,
disable_pareto_plans=False,
disable_pareto_proposals=False,
enable_striping=False,
)
sch, te_graph, const_dict = TwoConv2DTE
cs.cascade(sch, te_graph, const_dict, options, SRAM, FLASH, [SRAM], device_config)
Expand All @@ -69,7 +69,7 @@ def test_compute_cycles_annotation(SRAM, FLASH, TwoConv2DTE):
# [copy, copy, conv2d, copy, conv2d]
stages = [6, 8, 9, 18, 19]
# Expected hints for each operation
compute_cycles_hints = [4096, 5120, 1632, 2560, 3072]
compute_cycles_hints = [4096, 5120, 1440, 2560, 3072]

for stage, compute_cycles_hint in zip(stages, compute_cycles_hints):
op = sch.stages[stage]
Expand Down

0 comments on commit ac5d781

Please sign in to comment.