update proute addr calculation

stanford-ppl · Jul 25, 2018 · c42f64b · c42f64b
1 parent 5bff6e6
commit c42f64b
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/codegen/dot/NetworkDotCodegen.scala b/src/codegen/dot/NetworkDotCodegen.scala
@@ -22,7 +22,8 @@ class NetworkDotCodegen[B<:PinType:ClassTag](val fileName:String)(implicit compi
         n match {
           case n:Routable =>
             val List(x,y) = indexOf(n)
-            val idx = (param.numTotalRows-1-y) * param.numTotalCols + x
+            val maxDim = math.max(param.numTotalRows, param.numTotalCols)
+            val idx = (param.numTotalRows-1-y) * maxDim + x
             label += s"\n($idx)"
           case n =>
         }