diff --git a/mlir/parser_transformer.py b/mlir/parser_transformer.py index 89fe8ad..f2f62f0 100644 --- 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