Fix locking of switch demo PE inputs

Fix edge case where switch demo inputs are not locked when immediately starting read or write operations.
bittide · Feb 7, 2025 · 58ac6a6 · 58ac6a6
1 parent 8001616
commit 58ac6a6
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/bittide/src/Bittide/SwitchDemoProcessingElement.hs b/bittide/src/Bittide/SwitchDemoProcessingElement.hs
@@ -38,10 +38,10 @@ switchDemoPe ::
 switchDemoPe SNat localCounter linkIn maybeDna readStart readCycles writeStart writeCycles =
   (linkOut, buffer)
  where
-  readStartLocked = regEn maxBound (peState .==. pure Idle) readStart
-  readCyclesLocked = regEn maxBound (peState .==. pure Idle) ((* 3) . zeroExtend <$> readCycles)
-  writeStartLocked = regEn maxBound (peState .==. pure Idle) writeStart
-  writeCyclesLocked = regEn maxBound (peState .==. pure Idle) ((* 3) . zeroExtend <$> writeCycles)
+  readStartLocked = regEn maxBound (prevPeState .==. pure Idle) readStart
+  readCyclesLocked = regEn maxBound (prevPeState .==. pure Idle) ((* 3) . zeroExtend <$> readCycles)
+  writeStartLocked = regEn maxBound (prevPeState .==. pure Idle) writeStart
+  writeCyclesLocked = regEn maxBound (prevPeState .==. pure Idle) ((* 3) . zeroExtend <$> writeCycles)
 
   localData :: Signal dom (Vec 3 (BitVector 64))
   localData = bundle (unbundle dnaVec :< (pack <$> localCounter))