Commit 1eca1ad1 by Liangfu Chen Committed by Thierry Moreau

[VTA] Hotfix for padded load test in Chisel VTA (#4264)

* Update TensorUtil.scala

* Update test_vta_insn.py
parent 86b844b9
...@@ -319,8 +319,8 @@ class TensorDataCtrl(tensorType: String = "none", ...@@ -319,8 +319,8 @@ class TensorDataCtrl(tensorType: String = "none",
when(split) { when(split) {
caddr := caddr + xmax_bytes caddr := caddr + xmax_bytes
}.elsewhen(stride) { }.elsewhen(stride) {
caddr := baddr + (dec.xstride << log2Ceil(strideFactor)) caddr := baddr + (dec.xstride << log2Ceil(elemBytes))
baddr := baddr + (dec.xstride << log2Ceil(strideFactor)) baddr := baddr + (dec.xstride << log2Ceil(elemBytes))
} }
} }
......
...@@ -89,10 +89,10 @@ def test_padded_load(): ...@@ -89,10 +89,10 @@ def test_padded_load():
"""Test padded load.""" """Test padded load."""
def _run(env, remote): def _run(env, remote):
# declare # declare
n = 21 n = 3
m = 20 m = 5
pad_before = [0, 1, 0, 0] pad_before = [2, 1, 0, 0]
pad_after = [1, 3, 0, 0] pad_after = [1, 2, 0, 0]
x = tvm.placeholder( x = tvm.placeholder(
(n, m, env.BATCH, env.BLOCK_OUT), (n, m, env.BATCH, env.BLOCK_OUT),
name="x", name="x",
...@@ -126,7 +126,7 @@ def test_padded_load(): ...@@ -126,7 +126,7 @@ def test_padded_load():
f = remote.load_module("padded_load.o") f = remote.load_module("padded_load.o")
# verify # verify
ctx = remote.ext_dev(0) ctx = remote.ext_dev(0)
x_np = np.random.randint(1, 2, size=( x_np = np.random.randint(-10, 10, size=(
n, m, env.BATCH, env.BLOCK_OUT)).astype(x.dtype) n, m, env.BATCH, env.BLOCK_OUT)).astype(x.dtype)
y_np = np.zeros((n + pad_before[0] + pad_after[0], y_np = np.zeros((n + pad_before[0] + pad_after[0],
m + pad_before[1] + pad_after[1], m + pad_before[1] + pad_after[1],
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment