From 773eeda067f88122b151ed2f25c07a749a748e89 Mon Sep 17 00:00:00 2001
From: Jeehoon Kang <jeehoon.kang@kaist.ac.kr>
Date: Mon, 4 May 2020 12:33:26 +0900
Subject: [PATCH] Fix mem2reg test

---
 examples/mem2reg/mem2reg.input.ir  | 14 +++++++++
 examples/mem2reg/mem2reg.output.ir | 48 +++++++++++++++++++-----------
 2 files changed, 45 insertions(+), 17 deletions(-)

diff --git a/examples/mem2reg/mem2reg.input.ir b/examples/mem2reg/mem2reg.input.ir
index f843cc7..0c06db7 100644
--- a/examples/mem2reg/mem2reg.input.ir
+++ b/examples/mem2reg/mem2reg.input.ir
@@ -8,6 +8,16 @@ block b0:
   ret unit:unit
 }
 
+fun unit @sinkptr {
+init:
+  bid: b0
+  allocations:
+
+block b0:
+  %b0:p0:*i8
+  ret unit:unit
+}
+
 fun i32 @single_block {
 init:
   bid: b0
@@ -100,8 +110,12 @@ init:
     %l0:i32:x
     %l1:i16:y
     %l2:i64:z
+    %l3:i8:w
 
 block b666:
+  %b666:i0:unit = store %l3:*i8 42:i8
+  %b666:i1:unit = call @sinkptr(%l3:*i8)
+  %b666:i2:i8 = load %l3:*i8
   j b0()
 
 block b0:
diff --git a/examples/mem2reg/mem2reg.output.ir b/examples/mem2reg/mem2reg.output.ir
index 97b33e1..ff4ce9e 100644
--- a/examples/mem2reg/mem2reg.output.ir
+++ b/examples/mem2reg/mem2reg.output.ir
@@ -8,6 +8,16 @@ block b0:
   ret unit:unit
 }
 
+fun unit @sinkptr {
+init:
+  bid: b0
+  allocations:
+
+block b0:
+  %b0:p0:*i8
+  ret unit:unit
+}
+
 fun i32 @single_block {
 init:
   bid: b0
@@ -15,13 +25,13 @@ init:
     %l0:i32:x
 
 block b0:
-  %b0:i0:i32 = load %l0:*i32
+  %b0:i0:unit = nop
 
   %b0:i1:unit = nop
-  %b0:i2:i32 = load %l0:*i32
+  %b0:i2:unit = nop
 
   %b0:i3:unit = nop
-  %b0:i4:i32 = load %l0:*i32
+  %b0:i4:unit = nop
 
   %b0:i5:unit = call @sink(undef:i32)
   %b0:i6:unit = call @sink(42:i32)
@@ -37,18 +47,18 @@ init:
     %l0:i32:x
 
 block b0:
-  %b0:i0:i32 = load %l0:*i32
+  %b0:i0:unit = nop
 
   %b0:i1:unit = nop
-  %b0:i2:i32 = load %l0:*i32
+  %b0:i2:unit = nop
 
   %b0:i3:unit = nop
-  %b0:i4:i32 = load %l0:*i32
+  %b0:i4:unit = nop
 
   j b1()
 
 block b1:
-  %b1:i0:i32 = load %l0:*i32
+  %b1:i0:unit = nop
 
   %b1:i1:unit = call @sink(undef:i32)
   %b1:i2:unit = call @sink(42:i32)
@@ -70,7 +80,7 @@ block b666:
 block b0:
   %b0:p0:i32:x
 
-  %b0:i0:i32 = load %l0:*i32
+  %b0:i0:unit = nop
   %b0:i1:unit = call @sink(%b0:p0:i32)
   %b0:i2:unit = nop
   switch undef:i32 default b1(0:i32) [
@@ -80,14 +90,14 @@ block b0:
 
 block b1:
   %b1:p0:i32:x
-  %b1:i0:i32 = load %l0:*i32
+  %b1:i0:unit = nop
   %b1:i1:unit = call @sink(%b1:p0:i32)
   %b1:i2:unit = nop
   j b2(100:i32)
 
 block b2:
   %b2:p0:i32:x
-  %b2:i0:i32 = load %l0:*i32
+  %b2:i0:unit = nop
   %b2:i1:unit = call @sink(%b2:p0:i32)
   %b2:i2:unit = nop
   j b3()
@@ -104,8 +114,12 @@ init:
     %l0:i32:x
     %l1:i16:y
     %l2:i64:z
+    %l3:i8:w
 
 block b666:
+  %b666:i0:unit = store %l3:*i8 42:i8
+  %b666:i1:unit = call @sinkptr(%l3:*i8)
+  %b666:i2:i8 = load %l3:*i8
   j b0(undef:i32, undef:i16, undef:i64)
 
 block b0:
@@ -113,15 +127,15 @@ block b0:
   %b0:p1:i16:y
   %b0:p2:i64:z
 
-  %b0:i0:i32 = load %l0:*i32
+  %b0:i0:unit = nop
   %b0:i1:unit = call @sink(%b0:p0:i32)
   %b0:i2:unit = nop
 
-  %b0:i3:i16 = load %l1:*i16
+  %b0:i3:unit = nop
   %b0:i4:unit = call @sink(%b0:p1:i16)
   %b0:i5:unit = nop
 
-  %b0:i6:i64 = load %l2:*i64
+  %b0:i6:unit = nop
   %b0:i7:unit = call @sink(%b0:p2:i64)
   %b0:i8:unit = nop
 
@@ -135,11 +149,11 @@ block b1:
   %b1:p1:i16:y
   %b1:p2:i64:z
 
-  %b1:i0:i32 = load %l0:*i32
+  %b1:i0:unit = nop
   %b1:i1:unit = call @sink(%b1:p0:i32)
   %b1:i2:unit = nop
 
-  %b1:i3:i64 = load %l2:*i64
+  %b1:i3:unit = nop
   %b1:i4:unit = call @sink(%b1:p2:i64)
   %b1:i5:unit = nop
 
@@ -149,11 +163,11 @@ block b2:
   %b2:p0:i32:x
   %b2:p1:i16:y
 
-  %b2:i0:i32 = load %l0:*i32
+  %b2:i0:unit = nop
   %b2:i1:unit = call @sink(%b2:p0:i32)
   %b2:i2:unit = nop
 
-  %b2:i3:i16 = load %l1:*i16
+  %b2:i3:unit = nop
   %b2:i4:unit = call @sink(%b2:p1:i16)
   %b2:i5:unit = nop