Index: tests_lit/llvm2ice_tests/square.ll |
diff --git a/tests_lit/llvm2ice_tests/square.ll b/tests_lit/llvm2ice_tests/square.ll |
index 50fa9e2dd5d0aa439977ee7ce8fdca2a2c07f633..137310a9cf0d28306a7eeb67972d9cf3fcc0b796 100644 |
--- a/tests_lit/llvm2ice_tests/square.ll |
+++ b/tests_lit/llvm2ice_tests/square.ll |
@@ -33,18 +33,22 @@ entry: |
; CHECK-LABEL: Square_i32 |
; CHECK: imul [[REG:e..]],[[REG]] |
-define internal i16 @Square_i16(i16 %a) { |
+define internal i32 @Square_i16(i32 %a) { |
entry: |
- %result = mul i16 %a, %a |
- ret i16 %result |
+ %a.16 = trunc i32 %a to i16 |
+ %result = mul i16 %a.16, %a.16 |
+ %result.i32 = sext i16 %result to i32 |
+ ret i32 %result.i32 |
} |
; CHECK-LABEL: Square_i16 |
; CHECK: imul [[REG:..]],[[REG]] |
-define internal i8 @Square_i8(i8 %a) { |
+define internal i32 @Square_i8(i32 %a) { |
entry: |
- %result = mul i8 %a, %a |
- ret i8 %result |
+ %a.8 = trunc i32 %a to i8 |
+ %result = mul i8 %a.8, %a.8 |
+ %result.i32 = sext i8 %result to i32 |
+ ret i32 %result.i32 |
} |
; CHECK-LABEL: Square_i8 |
; CHECK: imul al |