Generate NUM_CORES macro in the C++ header (#161)

OpenXiangShan · Sep 19, 2023 · de047e5 · de047e5
1 parent f22b2da
commit de047e5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/main/scala/Difftest.scala b/src/main/scala/Difftest.scala
@@ -457,7 +457,10 @@ object DifftestModule {
     difftestCpp += s"#define CPU_$cpu_s"
     difftestCpp += ""
 
-    val numCores = instances.filter(_._1.isUniqueIdentifier).length
+    val numCores = instances.count(_._1.isUniqueIdentifier)
+    difftestCpp += s"#define NUM_CORES $numCores"
+    difftestCpp += ""
+
     val uniqBundles = instances.groupBy(_._1.desiredModuleName)
     // Create cpp declaration for each bundle type
     uniqBundles.values.map(_.map(_._1)).foreach(bundles => {