cores/cpu: Fix automatic collection for Rocket/BlackParrot/EOS-S3.

litex/soc/cores/cpu/__init__.py

@@ -94,11 +94,10 @@ def collect_cpus():
                 continue
 
             # OK, it seems to be a CPU; now get the class and add it to dict.
-            cpu        = file
+            cpu = file
-            cpu_module = f"{cpu}"
             sys.path.append(path)
-            for cpu_name, cpu_cls in inspect.getmembers(importlib.import_module(cpu_module), inspect.isclass):
+            for cpu_name, cpu_cls in inspect.getmembers(importlib.import_module(cpu), inspect.isclass):
-                if cpu.replace("_", "") == cpu_name.lower():
+                if cpu_name.lower() in [cpu, cpu.replace("_", "")]:
                     cpus[cpu] = cpu_cls
 
     # Return collected CPUs.

litex/soc/cores/cpu/blackparrot/__init__.py

@@ -1 +1 @@
-from litex.soc.cores.cpu.blackparrot.core import BlackParrotRV64
+from litex.soc.cores.cpu.blackparrot.core import BlackParrot

litex/soc/cores/cpu/blackparrot/core.py

@@ -52,9 +52,9 @@ GCC_FLAGS = {
     "sim":      "-march=rv64imafd -mabi=lp64d ",
 }
 
-# BlackParrotRV64 ----------------------------------------------------------------------------------
+# BlackParrot --------------------------------------------------------------------------------------
 
-class BlackParrotRV64(CPU):
+class BlackParrot(CPU):
     family               = "riscv"
     name                 = "blackparrot"
     human_name           = "BlackParrotRV64[imafd]"

litex/soc/cores/cpu/rocket/__init__.py

@@ -1 +1 @@
-from litex.soc.cores.cpu.rocket.core import RocketRV64
+from litex.soc.cores.cpu.rocket.core import Rocket

litex/soc/cores/cpu/rocket/core.py

@@ -86,9 +86,9 @@ CPU_SIZE_PARAMS = {
     "full4q":   (   256,      64,         4),
 }
 
-# Rocket RV64 --------------------------------------------------------------------------------------
+# Rocket  ------------------------------------------------------------------------------------------
 
-class RocketRV64(CPU):
+class Rocket(CPU):
     family               = "riscv"
     name                 = "rocket"
     human_name           = "RocketRV64[imac]"