Support more than 32 bits in readmask

sjshao09 · Aug 31, 2015 · a6f06b8 · a6f06b8
1 parent 7c91576
commit a6f06b8
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/com/custom_computing_ic/dfe_snippets/sparse/ParallelCsrReadControl.java b/src/com/custom_computing_ic/dfe_snippets/sparse/ParallelCsrReadControl.java
@@ -38,7 +38,7 @@ public ParallelCsrReadControl(DFEManager owner, int inputWidth, boolean dbg) {
       mode = state.enumerated(Mode.class, Mode.ReadingLength);
       iLength = io.pullInput("length", dfeUInt(32));
 
-      oReadMask    = io.pushOutput("readmask", dfeUInt(32), 1);
+      oReadMask    = io.pushOutput("readmask", dfeUInt(inputWidth), 1);
       oReadEnable  = io.pushOutput("readenable", dfeUInt(32), 1);
       oRowFinished = io.pushOutput("rowFinished", dfeUInt(32), 1);
       oRowLength   = io.pushOutput("rowLength", dfeUInt(32), 1);
@@ -55,7 +55,7 @@ public ParallelCsrReadControl(DFEManager owner, int inputWidth, boolean dbg) {
       rowFinishedData = state.value(dfeUInt(32), true);
       rowLengthData = state.value(dfeUInt(32), 0);
       readMaskOutValid = state.value(dfeBool(), false);
-      readMaskData = state.value(dfeUInt(32));
+      readMaskData = state.value(dfeUInt(inputWidth));
       iLengthReady = state.value(dfeBool(), false);
     }
 
@@ -125,7 +125,7 @@ protected void nextState() {
 
             for (int i = 0; i <= inputWidth; i++)
               IF (crtPos === i)
-                readMaskData.next <== pattern.shiftLeft(i).cast(dfeUInt(32));
+                readMaskData.next <== pattern.shiftLeft(i).cast(dfeUInt(inputWidth));
 
             // crtPos = (crtPos + canread ) % inputWidth;
             IF (crtPos + canread >= inputWidth) {