Skip to content
Snippets Groups Projects
Commit 53f37b8a authored by auphelia's avatar auphelia
Browse files

[Test] Added all signed test case

parent 2fff01d6
No related branches found
No related tags found
No related merge requests found
......@@ -149,5 +149,36 @@ def test_fpgadataflow_fclayer_all_bipolar():
assert (produced.reshape(expected.shape) == expected).all()
def test_fpgadataflow_fclayer_all_signed():
mh = 8
mw = 8
wdt = idt = odt = DataType.INT2
tdt = DataType.INT32
# generate weights
W = gen_FINN_dt_tensor(wdt, [mh, mw])
# single global threshold at zero
T = np.zeros((1, 1))
# generate input data
x = gen_FINN_dt_tensor(idt, mw)
# set up layers with different pe and simd
pe_values = [1, int(mh/2), mh]
simd_values = [1, int(mw/2), mw]
for pe in pe_values:
for simd in simd_values:
model = make_single_fclayer_modelwrapper(W, pe, simd, wdt, idt, odt, T, tdt)
# prepare input data
input_dict = prepare_inputs(model, x, idt)
# execute model
produced = oxe.execute_onnx(model, input_dict)["outp"]
# expected output
oshape = model.get_tensor_shape("outp")
y = np.dot(W, x).reshape(oshape.shape)
expected = multithreshold(y.reshape(1, mh), T)
assert (produced.reshape(expected.shape) == expected).all()
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment