From 8b404f61dfc627af5151e9001cd14311671e87d0 Mon Sep 17 00:00:00 2001 From: Slava Shklyaev Date: Thu, 9 Apr 2020 14:44:53 +0100 Subject: Add a WHILE test with unused outputs Fix: 148208229 Test: NNT_static Change-Id: I853f1f294cedacb0cbb810042ef2a3466a072ffe --- nn/runtime/test/specs/V1_3/while_fib.mod.py | 19 +++++++++++++------ 1 file changed, 13 insertions(+), 6 deletions(-) (limited to 'nn/runtime/test/specs/V1_3') diff --git a/nn/runtime/test/specs/V1_3/while_fib.mod.py b/nn/runtime/test/specs/V1_3/while_fib.mod.py index 963c24bfc..d26488fcb 100644 --- a/nn/runtime/test/specs/V1_3/while_fib.mod.py +++ b/nn/runtime/test/specs/V1_3/while_fib.mod.py @@ -77,14 +77,18 @@ def MakeBodyModel(): return SubgraphReference("body", model), quant8, quant8_signed -def Test(n_data, fib_data): +def Test(n_data, fib_data, add_unused_output=False): n = Input("n", CounterType) fib_out = Output("fib_out", FibType) cond, cond_quant8, cond_quant8_signed = MakeConditionModel() body, body_quant8, body_quant8_signed = MakeBodyModel() fib_init = Parameter("fib_init", FibType, [1, 1]) i_init = [1] - model = Model().Operation("WHILE", cond, body, fib_init, i_init, n).To(fib_out) + outputs = [fib_out] + if add_unused_output: + i_out = Internal("i_out", CounterType) # Unused. + outputs.append(i_out) + model = Model().Operation("WHILE", cond, body, fib_init, i_init, n).To(outputs) quant8 = DataTypeConverter().Identify({ fib_init: FibTypeQuant8, @@ -99,15 +103,18 @@ def Test(n_data, fib_data): body: body_quant8_signed, }) - example = Example({n: [n_data], fib_out: fib_data}, name="n_{}".format(n_data)) + name = "n_{}".format(n_data) + if add_unused_output: + name += "_unused_output" + example = Example({n: [n_data], fib_out: fib_data}, name=name) example.AddVariations("relaxed", "float16", quant8, quant8_signed) example.AddVariations(AllOutputsAsInternalCoverter()) for use_shm_for_weights in [False, True]: Configuration.use_shm_for_weights = use_shm_for_weights # Fibonacci numbers: 1 1 2 3 5 8 - Test(n_data=1, fib_data=[1, 1]) - Test(n_data=2, fib_data=[1, 2]) - Test(n_data=3, fib_data=[2, 3]) + Test(n_data=1, fib_data=[1, 1], add_unused_output=True) + Test(n_data=2, fib_data=[1, 2], add_unused_output=True) + Test(n_data=3, fib_data=[2, 3], add_unused_output=True) Test(n_data=4, fib_data=[3, 5]) Test(n_data=5, fib_data=[5, 8]) -- cgit v1.2.3