From 8d5c2ae0b965aa4f2940421265064e17867e237f Mon Sep 17 00:00:00 2001
From: Hendrik Borras <hendrikborras@web.de>
Date: Mon, 23 Aug 2021 14:15:48 +0100
Subject: [PATCH] Made sure test inputs start at 0 for nicer debugging.
---
tests/fpgadataflow/test_fpgadataflow_upsampler.py | 3 +++
1 file changed, 3 insertions(+)
diff --git a/tests/fpgadataflow/test_fpgadataflow_upsampler.py b/tests/fpgadataflow/test_fpgadataflow_upsampler.py
index cc398887f..64a0519c9 100644
--- a/tests/fpgadataflow/test_fpgadataflow_upsampler.py
+++ b/tests/fpgadataflow/test_fpgadataflow_upsampler.py
@@ -131,8 +131,11 @@ def test_fpgadataflow_upsampler(dt, IFMDim, OFMDim, NumChannels, exec_mode):
torch_model = PyTorchTestModel(upscale_factor=OFMDim / IFMDim)
input_shape = (1, NumChannels, IFMDim, IFMDim)
test_in = torch.arange(0, np.prod(np.asarray(input_shape)))
+ # Limit the input to values valid for the given datatype
test_in %= dt.max() - dt.min() + 1
test_in += dt.min()
+ # Additionally make sure we always start with 0, for convenience purposes.
+ test_in = torch.roll(test_in, dt.min())
test_in = test_in.view(*input_shape).type(torch.float32)
# Get golden PyTorch and ONNX inputs
--
GitLab