|
|
|
@@ -7,14 +7,15 @@ from cocotb.triggers import RisingEdge
|
|
|
|
|
from model.ref_model import ( # noqa: E402
|
|
|
|
|
generate_write_flip_mask,
|
|
|
|
|
match_top1,
|
|
|
|
|
match_top1_with_read_noise,
|
|
|
|
|
random_hashes,
|
|
|
|
|
unpack_score_debug_flat,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
NUM_ROWS = 512
|
|
|
|
|
HASH_BITS = 512
|
|
|
|
|
LANES = 16
|
|
|
|
|
SCORE_BITS = 10
|
|
|
|
|
DEFAULT_NUM_ROWS = 4096
|
|
|
|
|
DEFAULT_HASH_BITS = 512
|
|
|
|
|
DEFAULT_LANES = 8
|
|
|
|
|
DEFAULT_SCORE_BITS = 10
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def _get_param(dut, name, default=None):
|
|
|
|
@@ -28,6 +29,38 @@ def _get_param(dut, name, default=None):
|
|
|
|
|
return default
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def dut_num_rows(dut):
|
|
|
|
|
val = _get_param(dut, "NUM_ROWS", None)
|
|
|
|
|
if val is not None:
|
|
|
|
|
return val
|
|
|
|
|
# Derive from wr_addr width (ROW_BITS): NUM_ROWS = 2^ROW_BITS
|
|
|
|
|
return 1 << len(dut.wr_addr)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def dut_hash_bits(dut):
|
|
|
|
|
val = _get_param(dut, "HASH_BITS", None)
|
|
|
|
|
if val is not None:
|
|
|
|
|
return val
|
|
|
|
|
# Derive from write_hash signal width
|
|
|
|
|
return len(dut.write_hash)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def dut_lanes(dut):
|
|
|
|
|
val = _get_param(dut, "LANES", None)
|
|
|
|
|
if val is not None:
|
|
|
|
|
return val
|
|
|
|
|
# Derive from rd_resp_row_ids width / ROW_BITS
|
|
|
|
|
return len(dut.rd_resp_row_ids) // len(dut.wr_addr)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
def dut_score_bits(dut):
|
|
|
|
|
val = _get_param(dut, "SCORE_BITS", None)
|
|
|
|
|
if val is not None:
|
|
|
|
|
return val
|
|
|
|
|
# Derive from top1_score signal width
|
|
|
|
|
return len(dut.top1_score)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ── Helpers ──────────────────────────────────────────────────────────────────
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@@ -57,14 +90,7 @@ async def wait_idle(dut):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def write_row(dut, addr, value):
|
|
|
|
|
"""Write a single row using wr_valid/wr_ready handshake.
|
|
|
|
|
|
|
|
|
|
1. Wait for system idle
|
|
|
|
|
2. Assert wr_valid + set addr/hash
|
|
|
|
|
3. Wait for handshake (wr_ready=1 on clock edge)
|
|
|
|
|
4. Deassert wr_valid
|
|
|
|
|
5. Wait for wr_ready to return 1 (commit complete)
|
|
|
|
|
"""
|
|
|
|
|
"""Write a single row using wr_valid/wr_ready handshake."""
|
|
|
|
|
await wait_idle(dut)
|
|
|
|
|
|
|
|
|
|
dut.wr_addr.value = addr
|
|
|
|
@@ -79,7 +105,7 @@ async def write_row(dut, addr, value):
|
|
|
|
|
|
|
|
|
|
dut.wr_valid.value = 0
|
|
|
|
|
|
|
|
|
|
# Wait for cam_noisy to finish GEN_MASK/COMMIT
|
|
|
|
|
# Wait for write pipeline to drain
|
|
|
|
|
await wait_idle(dut)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@@ -90,15 +116,7 @@ async def write_rows(dut, rows):
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
async def query_once(dut, query):
|
|
|
|
|
"""Issue a query and return (top1_index, top1_score, score_debug).
|
|
|
|
|
|
|
|
|
|
1. Wait for system idle
|
|
|
|
|
2. Assert query_valid + set query_hash
|
|
|
|
|
3. Wait for query_ready handshake
|
|
|
|
|
4. Deassert query_valid
|
|
|
|
|
5. Wait for result_valid
|
|
|
|
|
6. Read result, pulse result_ready to consume
|
|
|
|
|
"""
|
|
|
|
|
"""Issue a query and return (top1_index, top1_score, score_debug)."""
|
|
|
|
|
await wait_idle(dut)
|
|
|
|
|
|
|
|
|
|
dut.query_hash.value = int(query)
|
|
|
|
@@ -119,12 +137,14 @@ async def query_once(dut, query):
|
|
|
|
|
top1_index = int(dut.top1_index.value)
|
|
|
|
|
top1_score = int(dut.top1_score.value)
|
|
|
|
|
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
score_bits = dut_score_bits(dut)
|
|
|
|
|
score_debug = None
|
|
|
|
|
if hasattr(dut, "score_debug_flat"):
|
|
|
|
|
score_debug = unpack_score_debug_flat(
|
|
|
|
|
int(dut.score_debug_flat.value),
|
|
|
|
|
NUM_ROWS,
|
|
|
|
|
SCORE_BITS,
|
|
|
|
|
num_rows,
|
|
|
|
|
score_bits,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
dut.result_ready.value = 1
|
|
|
|
@@ -134,66 +154,83 @@ async def query_once(dut, query):
|
|
|
|
|
return top1_index, top1_score, score_debug
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ── Test A: Baseline (NOISE_EN=0) ────────────────────────────────────────────
|
|
|
|
|
# ── Compile smoke test ────────────────────────────────────────────────────────
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def compile_includes_grouped_noise_helper(dut):
|
|
|
|
|
"""Compilation test: new grouped noise helper must elaborate with cam_top."""
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
assert int(dut.wr_ready.value) in (0, 1)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ── Test A: Baseline (WRITE_NOISE_EN=0) ─────────────────────────────────────
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def baseline_no_noise(dut):
|
|
|
|
|
"""Verify write+query works exactly like the old CAM when NOISE_EN=0."""
|
|
|
|
|
noise_en = _get_param(dut, "NOISE_EN", 0)
|
|
|
|
|
if noise_en:
|
|
|
|
|
dut._log.info("Skipping baseline_no_noise: requires NOISE_EN=0.")
|
|
|
|
|
"""Verify write+query works exactly like the old CAM when noise disabled."""
|
|
|
|
|
noise_en = _get_param(dut, "WRITE_NOISE_EN", 0)
|
|
|
|
|
read_noise_en = _get_param(dut, "READ_NOISE_EN", 0)
|
|
|
|
|
if noise_en or read_noise_en:
|
|
|
|
|
dut._log.info("Skipping baseline_no_noise: requires noise disabled.")
|
|
|
|
|
return
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
rng = np.random.default_rng(1)
|
|
|
|
|
rows = random_hashes(rng, NUM_ROWS, width=HASH_BITS)
|
|
|
|
|
query_index = 123
|
|
|
|
|
rows = random_hashes(rng, num_rows, width=hash_bits)
|
|
|
|
|
query_index = min(123, num_rows - 1)
|
|
|
|
|
query = rows[query_index]
|
|
|
|
|
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
top1_index, top1_score, score_debug = await query_once(dut, query)
|
|
|
|
|
|
|
|
|
|
expected = match_top1(query, rows, width=HASH_BITS)
|
|
|
|
|
expected = match_top1(query, rows, width=hash_bits)
|
|
|
|
|
|
|
|
|
|
assert top1_index == expected.top1_index
|
|
|
|
|
assert top1_score == expected.top1_score
|
|
|
|
|
assert top1_index == query_index
|
|
|
|
|
assert top1_score == HASH_BITS
|
|
|
|
|
assert top1_score == hash_bits
|
|
|
|
|
|
|
|
|
|
if score_debug is not None:
|
|
|
|
|
assert np.array_equal(score_debug, expected.scores)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ── Test B: Zero noise rate (NOISE_EN=1, RATE_NUM=0) ────────────────────────
|
|
|
|
|
# ── Test B: Zero noise rate (WRITE_NOISE_EN=1, RATE_NUM=0) ──────────────────
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def zero_rate_noise(dut):
|
|
|
|
|
"""Noise module connected but THRESHOLD=0 → no flips, equivalent to NOISE_EN=0."""
|
|
|
|
|
noise_en = _get_param(dut, "NOISE_EN", 1)
|
|
|
|
|
rate_num = _get_param(dut, "NOISE_RATE_NUM", 1)
|
|
|
|
|
if not noise_en or rate_num != 0:
|
|
|
|
|
dut._log.info("Skipping zero_rate_noise: requires NOISE_EN=1, RATE_NUM=0.")
|
|
|
|
|
"""Noise module connected but THRESHOLD=0 → no flips."""
|
|
|
|
|
noise_en = _get_param(dut, "WRITE_NOISE_EN", 1)
|
|
|
|
|
rate_num = _get_param(dut, "WRITE_NOISE_RATE_NUM", 1)
|
|
|
|
|
read_noise_en = _get_param(dut, "READ_NOISE_EN", 0)
|
|
|
|
|
if not noise_en or rate_num != 0 or read_noise_en:
|
|
|
|
|
dut._log.info("Skipping zero_rate_noise: requires WRITE_NOISE_EN=1, RATE_NUM=0, READ_NOISE_EN=0.")
|
|
|
|
|
return
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
rng = np.random.default_rng(1)
|
|
|
|
|
rows = random_hashes(rng, NUM_ROWS, width=HASH_BITS)
|
|
|
|
|
query_index = 123
|
|
|
|
|
rows = random_hashes(rng, num_rows, width=hash_bits)
|
|
|
|
|
query_index = min(123, num_rows - 1)
|
|
|
|
|
query = rows[query_index]
|
|
|
|
|
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
top1_index, top1_score, score_debug = await query_once(dut, query)
|
|
|
|
|
|
|
|
|
|
expected = match_top1(query, rows, width=HASH_BITS)
|
|
|
|
|
expected = match_top1(query, rows, width=hash_bits)
|
|
|
|
|
|
|
|
|
|
assert top1_index == expected.top1_index
|
|
|
|
|
assert top1_score == expected.top1_score
|
|
|
|
|
assert top1_index == query_index
|
|
|
|
|
assert top1_score == HASH_BITS
|
|
|
|
|
assert top1_score == hash_bits
|
|
|
|
|
|
|
|
|
|
if score_debug is not None:
|
|
|
|
|
assert np.array_equal(score_debug, expected.scores)
|
|
|
|
@@ -204,17 +241,12 @@ async def zero_rate_noise(dut):
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def full_rate_noise(dut):
|
|
|
|
|
"""NOISE_RATE_NUM=1, NOISE_RATE_DEN=1 → every group flips its selected bit per write.
|
|
|
|
|
|
|
|
|
|
At full rate with default NOISE_BITS=8, exactly 8 deterministic bits flip per write
|
|
|
|
|
(one selected bit per group), not all 512 bits. We use ref_model.py PRNG to predict
|
|
|
|
|
the exact stored rows.
|
|
|
|
|
"""
|
|
|
|
|
noise_en = _get_param(dut, "NOISE_EN", 1)
|
|
|
|
|
rate_num = _get_param(dut, "NOISE_RATE_NUM", 1)
|
|
|
|
|
rate_den = _get_param(dut, "NOISE_RATE_DEN", 100)
|
|
|
|
|
"""WRITE_NOISE_RATE_NUM=1, WRITE_NOISE_RATE_DEN=1 → every group flips."""
|
|
|
|
|
noise_en = _get_param(dut, "WRITE_NOISE_EN", 1)
|
|
|
|
|
rate_num = _get_param(dut, "WRITE_NOISE_RATE_NUM", 1)
|
|
|
|
|
rate_den = _get_param(dut, "WRITE_NOISE_RATE_DEN", 100)
|
|
|
|
|
if not noise_en or rate_num != 1 or rate_den != 1:
|
|
|
|
|
dut._log.info("Skipping full_rate_noise: requires NOISE_EN=1, RATE_NUM=1, RATE_DEN=1.")
|
|
|
|
|
dut._log.info("Skipping full_rate_noise: requires WRITE_NOISE_EN=1, RATE_NUM=1, RATE_DEN=1.")
|
|
|
|
|
return
|
|
|
|
|
if not hasattr(dut, "score_debug_flat"):
|
|
|
|
|
dut._log.info("Skipping full_rate_noise: requires SIM_DEBUG (score_debug_flat).")
|
|
|
|
@@ -223,47 +255,41 @@ async def full_rate_noise(dut):
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
noise_bits = _get_param(dut, "NOISE_BITS", 8)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
noise_bits = _get_param(dut, "WRITE_NOISE_BITS", 8)
|
|
|
|
|
all_zero = 0
|
|
|
|
|
all_one = (1 << HASH_BITS) - 1
|
|
|
|
|
all_one = (1 << hash_bits) - 1
|
|
|
|
|
|
|
|
|
|
# Predict stored rows using the same RTL seed convention as exact_noise_model_match.
|
|
|
|
|
RTL_SEED = 0xB504_F32D_B504_F32D
|
|
|
|
|
prng_state = (RTL_SEED << 64) | RTL_SEED
|
|
|
|
|
|
|
|
|
|
# Flip mask for row 0 (all-zero written)
|
|
|
|
|
flip0, prng_state = generate_write_flip_mask(
|
|
|
|
|
prng_state, HASH_BITS, noise_bits, rate_num, rate_den,
|
|
|
|
|
prng_state, hash_bits, noise_bits, rate_num, rate_den,
|
|
|
|
|
)
|
|
|
|
|
expected_row0 = all_zero ^ flip0 # stored value after noise
|
|
|
|
|
expected_row0 = all_zero ^ flip0
|
|
|
|
|
|
|
|
|
|
# Flip mask for row 1 (all-one written)
|
|
|
|
|
flip1, prng_state = generate_write_flip_mask(
|
|
|
|
|
prng_state, HASH_BITS, noise_bits, rate_num, rate_den,
|
|
|
|
|
prng_state, hash_bits, noise_bits, rate_num, rate_den,
|
|
|
|
|
)
|
|
|
|
|
expected_row1 = all_one ^ flip1 # stored value after noise
|
|
|
|
|
expected_row1 = all_one ^ flip1
|
|
|
|
|
|
|
|
|
|
# Write all-zero to row 0, all-one to row 1, rest zero
|
|
|
|
|
rows = [0] * NUM_ROWS
|
|
|
|
|
rows = [0] * num_rows
|
|
|
|
|
rows[0] = all_zero
|
|
|
|
|
rows[1] = all_one
|
|
|
|
|
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
|
|
|
|
|
# Query expected_row0 → should exactly match row 0's stored value
|
|
|
|
|
top1_index, top1_score, score_debug = await query_once(dut, expected_row0)
|
|
|
|
|
assert score_debug is not None, "score_debug required for full_rate_noise"
|
|
|
|
|
assert int(score_debug[0]) == HASH_BITS, (
|
|
|
|
|
f"Row 0: expected exact match for predicted stored value, "
|
|
|
|
|
f"score={score_debug[0]} != {HASH_BITS}"
|
|
|
|
|
assert int(score_debug[0]) == hash_bits, (
|
|
|
|
|
f"Row 0: expected exact match, score={score_debug[0]} != {hash_bits}"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
# Query expected_row1 → should exactly match row 1's stored value
|
|
|
|
|
top1_index, top1_score, score_debug = await query_once(dut, expected_row1)
|
|
|
|
|
assert score_debug is not None, "score_debug required for full_rate_noise"
|
|
|
|
|
assert int(score_debug[1]) == HASH_BITS, (
|
|
|
|
|
f"Row 1: expected exact match for predicted stored value, "
|
|
|
|
|
f"score={score_debug[1]} != {HASH_BITS}"
|
|
|
|
|
assert score_debug is not None
|
|
|
|
|
assert int(score_debug[1]) == hash_bits, (
|
|
|
|
|
f"Row 1: expected exact match, score={score_debug[1]} != {hash_bits}"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@@ -273,111 +299,117 @@ async def full_rate_noise(dut):
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def default_noise_reproducible(dut):
|
|
|
|
|
"""Fixed seed → deterministic write noise. Two identical runs produce same results."""
|
|
|
|
|
noise_en = _get_param(dut, "NOISE_EN", 1)
|
|
|
|
|
noise_en = _get_param(dut, "WRITE_NOISE_EN", 1)
|
|
|
|
|
if not noise_en:
|
|
|
|
|
dut._log.info("Skipping default_noise_reproducible: requires NOISE_EN=1.")
|
|
|
|
|
dut._log.info("Skipping default_noise_reproducible: requires WRITE_NOISE_EN=1.")
|
|
|
|
|
return
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
rng = np.random.default_rng(42)
|
|
|
|
|
rows = random_hashes(rng, NUM_ROWS, width=HASH_BITS)
|
|
|
|
|
rows = random_hashes(rng, num_rows, width=hash_bits)
|
|
|
|
|
|
|
|
|
|
# ── First run ──
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
query = rows[50]
|
|
|
|
|
query = rows[min(50, num_rows - 1)]
|
|
|
|
|
top1_index_1, top1_score_1, _ = await query_once(dut, query)
|
|
|
|
|
|
|
|
|
|
# Reset for second run
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
# ── Second run with same data ──
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
top1_index_2, top1_score_2, _ = await query_once(dut, query)
|
|
|
|
|
|
|
|
|
|
# Deterministic: same seed → same PRNG sequence → same stored hashes → same result
|
|
|
|
|
assert top1_index_1 == top1_index_2
|
|
|
|
|
assert top1_score_1 == top1_score_2
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ── Preserved legacy tests (only meaningful for NOISE_EN=0) ──────────────────
|
|
|
|
|
# ── Preserved legacy tests (only meaningful for noise disabled) ──────────────
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def known_hamming_distance(dut):
|
|
|
|
|
"""Hamming distance verification — exact scores only valid without noise."""
|
|
|
|
|
if _get_param(dut, "NOISE_EN", 1):
|
|
|
|
|
dut._log.info("Skipping known_hamming_distance: NOISE_EN=1, stored hashes may differ.")
|
|
|
|
|
if _get_param(dut, "WRITE_NOISE_EN", 1) or _get_param(dut, "READ_NOISE_EN", 0):
|
|
|
|
|
dut._log.info("Skipping known_hamming_distance: requires noise disabled.")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
|
|
|
|
|
query = 0
|
|
|
|
|
rows = [0] * NUM_ROWS
|
|
|
|
|
rows[10] = (1 << 7) - 1
|
|
|
|
|
rows[11] = (1 << 31) - 1
|
|
|
|
|
rows[12] = (1 << 128) - 1
|
|
|
|
|
rows = [0] * num_rows
|
|
|
|
|
rows[min(10, num_rows - 1)] = (1 << 7) - 1
|
|
|
|
|
rows[min(11, num_rows - 1)] = (1 << 31) - 1
|
|
|
|
|
rows[min(12, num_rows - 1)] = (1 << 128) - 1
|
|
|
|
|
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
top1_index, top1_score, score_debug = await query_once(dut, query)
|
|
|
|
|
|
|
|
|
|
assert top1_index == 0
|
|
|
|
|
assert top1_score == HASH_BITS
|
|
|
|
|
assert top1_score == hash_bits
|
|
|
|
|
|
|
|
|
|
if score_debug is not None:
|
|
|
|
|
assert int(score_debug[10]) == HASH_BITS - 7
|
|
|
|
|
assert int(score_debug[11]) == HASH_BITS - 31
|
|
|
|
|
assert int(score_debug[12]) == HASH_BITS - 128
|
|
|
|
|
assert int(score_debug[min(10, num_rows - 1)]) == hash_bits - 7
|
|
|
|
|
assert int(score_debug[min(11, num_rows - 1)]) == hash_bits - 31
|
|
|
|
|
assert int(score_debug[min(12, num_rows - 1)]) == hash_bits - 128
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def tie_break_policy(dut):
|
|
|
|
|
"""Tie-break: lowest row index wins — only verified without noise."""
|
|
|
|
|
if _get_param(dut, "NOISE_EN", 1):
|
|
|
|
|
dut._log.info("Skipping tie_break_policy: NOISE_EN=1, stored hashes may differ.")
|
|
|
|
|
if _get_param(dut, "WRITE_NOISE_EN", 1) or _get_param(dut, "READ_NOISE_EN", 0):
|
|
|
|
|
dut._log.info("Skipping tie_break_policy: requires noise disabled.")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
rng = np.random.default_rng(2)
|
|
|
|
|
rows = random_hashes(rng, NUM_ROWS, width=HASH_BITS)
|
|
|
|
|
query = rows[200]
|
|
|
|
|
rows = random_hashes(rng, num_rows, width=hash_bits)
|
|
|
|
|
query = rows[min(200, num_rows - 1)]
|
|
|
|
|
rows[10] = query
|
|
|
|
|
rows[20] = query
|
|
|
|
|
rows[200] = query
|
|
|
|
|
rows[min(200, num_rows - 1)] = query
|
|
|
|
|
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
top1_index, top1_score, _ = await query_once(dut, query)
|
|
|
|
|
|
|
|
|
|
assert top1_index == 10
|
|
|
|
|
assert top1_score == HASH_BITS
|
|
|
|
|
assert top1_score == hash_bits
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def all_zero_all_one_boundary(dut):
|
|
|
|
|
"""All-zero / all-one boundary — only verified without noise."""
|
|
|
|
|
if _get_param(dut, "NOISE_EN", 1):
|
|
|
|
|
dut._log.info("Skipping all_zero_all_one_boundary: NOISE_EN=1, stored hashes may differ.")
|
|
|
|
|
if _get_param(dut, "WRITE_NOISE_EN", 1) or _get_param(dut, "READ_NOISE_EN", 0):
|
|
|
|
|
dut._log.info("Skipping all_zero_all_one_boundary: requires noise disabled.")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
rows = [0] * NUM_ROWS
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
|
|
|
|
|
rows = [0] * num_rows
|
|
|
|
|
rows[0] = 0
|
|
|
|
|
rows[1] = (1 << HASH_BITS) - 1
|
|
|
|
|
rows[1] = (1 << hash_bits) - 1
|
|
|
|
|
|
|
|
|
|
query = 0
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
top1_index, top1_score, score_debug = await query_once(dut, query)
|
|
|
|
|
|
|
|
|
|
assert top1_score == HASH_BITS
|
|
|
|
|
assert top1_score == hash_bits
|
|
|
|
|
assert top1_index == 0
|
|
|
|
|
|
|
|
|
|
if score_debug is not None:
|
|
|
|
|
assert int(score_debug[0]) == HASH_BITS
|
|
|
|
|
assert int(score_debug[0]) == hash_bits
|
|
|
|
|
assert int(score_debug[1]) == 0
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@@ -386,16 +418,15 @@ async def all_zero_all_one_boundary(dut):
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def exact_noise_model_match(dut):
|
|
|
|
|
"""Verify RTL stored hashes match ref_model.py for a known seed and rate.
|
|
|
|
|
|
|
|
|
|
Writes rows with noise enabled, then queries back via score_debug to
|
|
|
|
|
reconstruct stored hashes, and compares against Python model predictions.
|
|
|
|
|
"""
|
|
|
|
|
noise_en = _get_param(dut, "NOISE_EN", 1)
|
|
|
|
|
rate_num = _get_param(dut, "NOISE_RATE_NUM", 1)
|
|
|
|
|
rate_den = _get_param(dut, "NOISE_RATE_DEN", 100)
|
|
|
|
|
"""Verify RTL stored hashes match ref_model.py for a known seed and rate."""
|
|
|
|
|
noise_en = _get_param(dut, "WRITE_NOISE_EN", 1)
|
|
|
|
|
rate_num = _get_param(dut, "WRITE_NOISE_RATE_NUM", 1)
|
|
|
|
|
rate_den = _get_param(dut, "WRITE_NOISE_RATE_DEN", 100)
|
|
|
|
|
if not noise_en or rate_num == 0:
|
|
|
|
|
dut._log.info("Skipping exact_noise_model_match: requires NOISE_EN=1, RATE_NUM>0.")
|
|
|
|
|
dut._log.info("Skipping exact_noise_model_match: requires WRITE_NOISE_EN=1, RATE_NUM>0.")
|
|
|
|
|
return
|
|
|
|
|
if _get_param(dut, "READ_NOISE_EN", 0):
|
|
|
|
|
dut._log.info("Skipping exact_noise_model_match: requires READ_NOISE_EN=0 (read noise corrupts score comparison).")
|
|
|
|
|
return
|
|
|
|
|
if not hasattr(dut, "score_debug_flat"):
|
|
|
|
|
dut._log.info("Skipping exact_noise_model_match: requires SIM_DEBUG.")
|
|
|
|
@@ -404,39 +435,31 @@ async def exact_noise_model_match(dut):
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
noise_bits = _get_param(dut, "NOISE_BITS", 8)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
noise_bits = _get_param(dut, "WRITE_NOISE_BITS", 8)
|
|
|
|
|
|
|
|
|
|
# Use a small subset to keep test fast
|
|
|
|
|
n_test_rows = 4
|
|
|
|
|
rng = np.random.default_rng(99)
|
|
|
|
|
rows = random_hashes(rng, n_test_rows, width=HASH_BITS)
|
|
|
|
|
rows = random_hashes(rng, n_test_rows, width=hash_bits)
|
|
|
|
|
|
|
|
|
|
# Predict stored hashes with Python model using the same seed.
|
|
|
|
|
# RTL random128 seed: {NOISE_SEED, NOISE_SEED}, default 64'hB504_F32D_B504_F32D.
|
|
|
|
|
RTL_SEED = 0xB504_F32D_B504_F32D
|
|
|
|
|
prng_state = (RTL_SEED << 64) | RTL_SEED
|
|
|
|
|
expected_stored = []
|
|
|
|
|
for row in rows:
|
|
|
|
|
flip, prng_state = generate_write_flip_mask(
|
|
|
|
|
prng_state, HASH_BITS, noise_bits, rate_num, rate_den,
|
|
|
|
|
prng_state, hash_bits, noise_bits, rate_num, rate_den,
|
|
|
|
|
)
|
|
|
|
|
expected_stored.append(row ^ flip)
|
|
|
|
|
|
|
|
|
|
# Write only test rows (rest stay at 0 from reset)
|
|
|
|
|
for idx, val in enumerate(rows):
|
|
|
|
|
await write_row(dut, idx, val)
|
|
|
|
|
|
|
|
|
|
# Query all-zero to get Hamming distances (= HASH_BITS - popcount(stored ^ 0) = HASH_BITS - popcount(stored))
|
|
|
|
|
# So popcount(stored) = HASH_BITS - score
|
|
|
|
|
# This gives us the number of set bits but not the exact value.
|
|
|
|
|
# Instead, query each expected_stored value — it should score HASH_BITS if match is exact.
|
|
|
|
|
for idx, expected in enumerate(expected_stored):
|
|
|
|
|
top1_index, top1_score, score_debug = await query_once(dut, expected)
|
|
|
|
|
# The stored hash at idx should exactly match expected, so score == HASH_BITS
|
|
|
|
|
assert score_debug is not None, "score_debug required for mask match verification"
|
|
|
|
|
assert int(score_debug[idx]) == HASH_BITS, (
|
|
|
|
|
assert int(score_debug[idx]) == hash_bits, (
|
|
|
|
|
f"Row {idx}: expected stored hash to match model prediction, "
|
|
|
|
|
f"score={score_debug[idx]} != {HASH_BITS}"
|
|
|
|
|
f"score={score_debug[idx]} != {hash_bits}"
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@@ -445,52 +468,160 @@ async def exact_noise_model_match(dut):
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def half_duplex_write_priority(dut):
|
|
|
|
|
"""When wr_valid and query_valid are both high, write wins and query is held off.
|
|
|
|
|
Only runs with NOISE_EN=0 so stored hashes are predictable.
|
|
|
|
|
"""
|
|
|
|
|
if _get_param(dut, "NOISE_EN", 1):
|
|
|
|
|
dut._log.info("Skipping half_duplex_write_priority: requires NOISE_EN=0 for exact scores.")
|
|
|
|
|
"""When wr_valid and query_valid are both high, write wins and query is held off."""
|
|
|
|
|
if _get_param(dut, "WRITE_NOISE_EN", 1) or _get_param(dut, "READ_NOISE_EN", 0):
|
|
|
|
|
dut._log.info("Skipping half_duplex_write_priority: requires noise disabled.")
|
|
|
|
|
return
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
# Write a known value to row 0
|
|
|
|
|
test_val = (1 << HASH_BITS) - 1 # all-ones
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
test_val = (1 << hash_bits) - 1
|
|
|
|
|
await write_row(dut, 0, test_val)
|
|
|
|
|
|
|
|
|
|
# Now system is idle: wr_ready=1, query_ready=1
|
|
|
|
|
await wait_idle(dut)
|
|
|
|
|
assert int(dut.wr_ready.value) == 1
|
|
|
|
|
assert int(dut.query_ready.value) == 1
|
|
|
|
|
|
|
|
|
|
# Drive both wr_valid and query_valid simultaneously
|
|
|
|
|
dut.wr_valid.value = 1
|
|
|
|
|
dut.wr_addr.value = 1
|
|
|
|
|
dut.write_hash.value = 0 # write all-zeros to row 1
|
|
|
|
|
dut.write_hash.value = 0
|
|
|
|
|
dut.query_valid.value = 1
|
|
|
|
|
dut.query_hash.value = test_val # query for all-ones (in row 0)
|
|
|
|
|
dut.query_hash.value = test_val
|
|
|
|
|
|
|
|
|
|
await RisingEdge(dut.clk)
|
|
|
|
|
|
|
|
|
|
# Write should have been accepted (wr_ready was 1), query should NOT have been accepted
|
|
|
|
|
# because write-priority gates query_ready when wr_valid=1
|
|
|
|
|
wr_accepted = int(dut.wr_ready.value) == 0 # after handshake, wr_ready drops
|
|
|
|
|
# query_ready should have been 0 during the simultaneous assertion
|
|
|
|
|
# (it's !wr_valid gated), so query was blocked
|
|
|
|
|
|
|
|
|
|
# Deassert both
|
|
|
|
|
dut.wr_valid.value = 0
|
|
|
|
|
dut.query_valid.value = 0
|
|
|
|
|
|
|
|
|
|
# Wait for write to complete (noise generation + commit)
|
|
|
|
|
await wait_idle(dut)
|
|
|
|
|
|
|
|
|
|
# Now query should work — row 0 has all-ones (written first)
|
|
|
|
|
top1_index, top1_score, _ = await query_once(dut, test_val)
|
|
|
|
|
assert top1_index == 0
|
|
|
|
|
assert top1_score == HASH_BITS
|
|
|
|
|
assert top1_score == hash_bits
|
|
|
|
|
|
|
|
|
|
# Verify row 1 was written (all-zeros) — query all-zeros
|
|
|
|
|
top1_index, top1_score, _ = await query_once(dut, 0)
|
|
|
|
|
assert top1_index == 1
|
|
|
|
|
assert top1_score == HASH_BITS
|
|
|
|
|
assert top1_score == hash_bits
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ── Test G: Banked pipeline no-noise Top-1 ───────────────────────────────────
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def banked_pipeline_no_noise_top1(dut):
|
|
|
|
|
"""No-noise banked pipeline returns the same Top-1 as the pure model."""
|
|
|
|
|
if _get_param(dut, "WRITE_NOISE_EN", 0) or _get_param(dut, "READ_NOISE_EN", 0):
|
|
|
|
|
dut._log.info("Skipping banked_pipeline_no_noise_top1: requires noise disabled.")
|
|
|
|
|
return
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
rng = np.random.default_rng(7)
|
|
|
|
|
rows = random_hashes(rng, num_rows, width=hash_bits)
|
|
|
|
|
query_index = min(17, num_rows - 1)
|
|
|
|
|
query = rows[query_index]
|
|
|
|
|
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
top1_index, top1_score, score_debug = await query_once(dut, query)
|
|
|
|
|
expected = match_top1(query, rows, width=hash_bits)
|
|
|
|
|
|
|
|
|
|
assert top1_index == expected.top1_index
|
|
|
|
|
assert top1_score == expected.top1_score
|
|
|
|
|
assert top1_index == query_index
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ── Test H: Query scan blocks writes until result consumed ───────────────────
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def query_scan_blocks_writes_until_result_consumed(dut):
|
|
|
|
|
"""Half-duplex: active query scan deasserts wr_ready."""
|
|
|
|
|
if _get_param(dut, "WRITE_NOISE_EN", 0) or _get_param(dut, "READ_NOISE_EN", 0):
|
|
|
|
|
dut._log.info("Skipping query_scan_blocks_writes: requires noise disabled.")
|
|
|
|
|
return
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
rows = [0] * num_rows
|
|
|
|
|
rows[0] = (1 << hash_bits) - 1
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
|
|
|
|
|
await wait_idle(dut)
|
|
|
|
|
dut.query_hash.value = rows[0]
|
|
|
|
|
dut.query_valid.value = 1
|
|
|
|
|
await RisingEdge(dut.clk)
|
|
|
|
|
dut.query_valid.value = 0
|
|
|
|
|
|
|
|
|
|
dut.wr_valid.value = 1
|
|
|
|
|
dut.wr_addr.value = 1
|
|
|
|
|
dut.write_hash.value = 0
|
|
|
|
|
await RisingEdge(dut.clk)
|
|
|
|
|
assert int(dut.wr_ready.value) == 0
|
|
|
|
|
dut.wr_valid.value = 0
|
|
|
|
|
|
|
|
|
|
while int(dut.result_valid.value) == 0:
|
|
|
|
|
await RisingEdge(dut.clk)
|
|
|
|
|
dut.result_ready.value = 1
|
|
|
|
|
await RisingEdge(dut.clk)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
# ── Test I: Read noise model match ──────────────────────────────────────────
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
@cocotb.test()
|
|
|
|
|
async def read_noise_model_match(dut):
|
|
|
|
|
"""Read noise uses grouped masks and matches the Python model for one query."""
|
|
|
|
|
if not _get_param(dut, "READ_NOISE_EN", 0):
|
|
|
|
|
dut._log.info("Skipping read_noise_model_match: requires READ_NOISE_EN=1.")
|
|
|
|
|
return
|
|
|
|
|
|
|
|
|
|
cocotb.start_soon(Clock(dut.clk, 10, unit="ns").start())
|
|
|
|
|
await reset_dut(dut)
|
|
|
|
|
|
|
|
|
|
num_rows = dut_num_rows(dut)
|
|
|
|
|
hash_bits = dut_hash_bits(dut)
|
|
|
|
|
lanes = dut_lanes(dut)
|
|
|
|
|
rng = np.random.default_rng(123)
|
|
|
|
|
rows = random_hashes(rng, num_rows, width=hash_bits)
|
|
|
|
|
|
|
|
|
|
# If write noise is enabled, apply write flip masks to predict stored rows
|
|
|
|
|
stored_rows = list(rows)
|
|
|
|
|
if _get_param(dut, "WRITE_NOISE_EN", 0):
|
|
|
|
|
seed = 0xB504_F32D_B504_F32D
|
|
|
|
|
prng_state = (seed << 64) | seed
|
|
|
|
|
stored_rows = []
|
|
|
|
|
for row in rows:
|
|
|
|
|
flip, prng_state = generate_write_flip_mask(
|
|
|
|
|
prng_state,
|
|
|
|
|
hash_bits,
|
|
|
|
|
_get_param(dut, "WRITE_NOISE_BITS", 8),
|
|
|
|
|
_get_param(dut, "WRITE_NOISE_RATE_NUM", 1),
|
|
|
|
|
_get_param(dut, "WRITE_NOISE_RATE_DEN", 100),
|
|
|
|
|
)
|
|
|
|
|
stored_rows.append(row ^ flip)
|
|
|
|
|
|
|
|
|
|
query = rows[min(5, num_rows - 1)]
|
|
|
|
|
|
|
|
|
|
await write_rows(dut, rows)
|
|
|
|
|
top1_index, top1_score, score_debug = await query_once(dut, query)
|
|
|
|
|
|
|
|
|
|
expected = match_top1_with_read_noise(
|
|
|
|
|
query,
|
|
|
|
|
stored_rows,
|
|
|
|
|
width=hash_bits,
|
|
|
|
|
lanes=lanes,
|
|
|
|
|
noise_bits=_get_param(dut, "READ_NOISE_BITS", 8),
|
|
|
|
|
rate_num=_get_param(dut, "READ_NOISE_RATE_NUM", 1),
|
|
|
|
|
rate_den=_get_param(dut, "READ_NOISE_RATE_DEN", 100),
|
|
|
|
|
seed=0x6A09_E667_F3BC_C909,
|
|
|
|
|
)
|
|
|
|
|
|
|
|
|
|
assert top1_index == expected.top1_index
|
|
|
|
|
assert top1_score == expected.top1_score
|
|
|
|
|
if score_debug is not None:
|
|
|
|
|
assert np.array_equal(score_debug, expected.scores)
|
|
|
|
|