Fix 'float_type' assertion error (#28)

spcl · Nov 9, 2023 · bddedc2 · bddedc2
1 parent 5bff437
commit bddedc2
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mlir/parser_transformer.py b/mlir/parser_transformer.py
@@ -63,7 +63,7 @@ def suffix_id(self, *suffix):
     BF16 = lambda self, tok: astnodes.FloatTypeEnum("bf16")
     F32 = lambda self, tok: astnodes.FloatTypeEnum("f32")
     F64 = lambda self, tok: astnodes.FloatTypeEnum("f64")
-    float_type = astnodes.FloatType.from_lark
+    float_type = lambda self, tok: astnodes.FloatType(astnodes.FloatTypeEnum(tok[0].value))
     index_type = astnodes.IndexType.from_lark
     signed_integer_type = astnodes.SignedIntegerType.from_lark
     unsigned_integer_type = astnodes.UnsignedIntegerType.from_lark