Fix wrong gep tokenization

fe-lang · Sep 11, 2023 · 067f42b · 067f42b
1 parent 117024e
commit 067f42b
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/crates/parser/src/lexer.rs b/crates/parser/src/lexer.rs
@@ -223,6 +223,7 @@ impl<'a> Lexer<'a> {
     fn try_eat_opcode(&mut self) -> Option<Code> {
         try_eat_variant! {
             self,
+            (b"gep", Code::Gep),
             (b"not", Code::Not),
             (b"neg", Code::Neg),
             (b"add", Code::Add),
@@ -254,7 +255,6 @@ impl<'a> Lexer<'a> {
             (b"fallthrough", Code::FallThrough),
             (b"br_table", Code::BrTable),
             (b"br", Code::Br),
-            (b"gep", Code::Gep),
             (b"alloca", Code::Alloca),
             (b"return", Code::Return),
             (b"phi", Code::Phi),

diff --git a/crates/parser/src/parser.rs b/crates/parser/src/parser.rs
@@ -930,6 +930,16 @@ mod tests {
         ));
     }
 
+    #[test]
+    fn test_gep() {
+        assert!(test_func_parser(
+            "func public %test(v0.**i32) -> **i32:
+    block0:
+        v1.*i32 = gep v0 10.i32;
+        return v1;"
+        ));
+    }
+
     #[test]
     fn parser_with_phi() {
         assert!(test_func_parser(