llvm/test/Transforms/InstCombine/memchr-3.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; RUN: opt < %s -passes=instcombine -S | FileCheck %s
   3
   4 ; Verify that the special case of memchr calls with the size of 1 are
   5 ; folded as expected.
   6
   7 declare ptr @memchr(ptr, i32, i64)
   8
   9 @ax = external global [0 x i8]
  10 @a12345 = constant [5 x i8] c"\01\02\03\04\05"
  11
  12
  13 ; Fold memchr(a12345, 1, 1) to a12345.
  14
  15 define ptr @fold_memchr_a12345_1_1() {
  16 ; CHECK-LABEL: @fold_memchr_a12345_1_1(
  17 ; CHECK-NEXT:    ret ptr @a12345
  18 ;
  19
  20   %res = call ptr @memchr(ptr @a12345, i32 1, i64 1)
  21   ret ptr %res
  22 }
  23
  24
  25 ; Fold memchr(a12345, 2, 1) to null.
  26
  27 define ptr @fold_memchr_a12345_2_1() {
  28 ; CHECK-LABEL: @fold_memchr_a12345_2_1(
  29 ; CHECK-NEXT:    ret ptr null
  30 ;
  31
  32   %res = call ptr @memchr(ptr @a12345, i32 2, i64 1)
  33   ret ptr %res
  34 }
  35
  36
  37 ; Fold memchr(ax, 257, 1) to (unsigned char)*ax == 1 ? ax : null
  38 ; to verify the constant 257 is converted to unsigned char (yielding 1).
  39
  40 define ptr @fold_memchr_ax_257_1(i32 %chr, i64 %n) {
  41 ; CHECK-LABEL: @fold_memchr_ax_257_1(
  42 ; CHECK-NEXT:    [[MEMCHR_CHAR0:%.*]] = load i8, ptr @ax, align 1
  43 ; CHECK-NEXT:    [[MEMCHR_CHAR0CMP:%.*]] = icmp eq i8 [[MEMCHR_CHAR0]], 1
  44 ; CHECK-NEXT:    [[MEMCHR_SEL:%.*]] = select i1 [[MEMCHR_CHAR0CMP]], ptr @ax, ptr null
  45 ; CHECK-NEXT:    ret ptr [[MEMCHR_SEL]]
  46 ;
  47
  48   %res = call ptr @memchr(ptr @ax, i32 257, i64 1)
  49   ret ptr %res
  50 }
  51
  52
  53 ; Fold memchr(ax, c, 1) to (unsigned char)*ax == (unsigned char)c ? ax : null.
  54
  55 define ptr @fold_memchr_ax_c_1(i32 %chr, i64 %n) {
  56 ; CHECK-LABEL: @fold_memchr_ax_c_1(
  57 ; CHECK-NEXT:    [[MEMCHR_CHAR0:%.*]] = load i8, ptr @ax, align 1
  58 ; CHECK-NEXT:    [[TMP1:%.*]] = trunc i32 [[CHR:%.*]] to i8
  59 ; CHECK-NEXT:    [[MEMCHR_CHAR0CMP:%.*]] = icmp eq i8 [[MEMCHR_CHAR0]], [[TMP1]]
  60 ; CHECK-NEXT:    [[MEMCHR_SEL:%.*]] = select i1 [[MEMCHR_CHAR0CMP]], ptr @ax, ptr null
  61 ; CHECK-NEXT:    ret ptr [[MEMCHR_SEL]]
  62 ;
  63
  64   %res = call ptr @memchr(ptr @ax, i32 %chr, i64 1)
  65   ret ptr %res
  66 }