Fix variant extension handling

wallento · Jul 23, 2020 · 3d49550 · 3d49550
1 parent fec7a1e
commit 3d49550
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/riscvmodel/variant.py b/riscvmodel/variant.py
@@ -58,10 +58,10 @@ def __init__(self, name: str, *, custext=None):
             for ext in match.group(4).split("_"):
                 if ext[0] == "Z":
                     self.extensions |= set(
-                        [ext] + Variant.stdextZ["Z" + ext[1:].lower()].implies)
+                        [ext[0]+ext[1:].lower()] + Variant.stdextZ["Z" + ext[1:].lower()].implies)
                 elif ext[0] == "X":
                     self.extensions |= set(
-                        [ext] + self.custext["X" + ext[1:].lower()].implies)
+                        [ext[0]+ext[1:].lower()] + self.custext["X" + ext[1:].lower()].implies)
 
     def __str__(self):
         return self.name