float_type assertion error fix

spcl · Oct 23, 2023 · 0a1cb86 · 0a1cb86
1 parent 5bff437
commit 0a1cb86
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(tok[0].value)
     index_type = astnodes.IndexType.from_lark
     signed_integer_type = astnodes.SignedIntegerType.from_lark
     unsigned_integer_type = astnodes.UnsignedIntegerType.from_lark