test/Transforms/InstCombine/memset-1.ll

   1 ; NOTE: Assertions have been autogenerated by utils/update_test_checks.py
   2 ; Test that the memset library call simplifier works correctly.
   3 ;
   4 ; RUN: opt < %s -instcombine -S | FileCheck %s
   5
   6 target datalayout = "e-p:32:32:32-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:32:64-f32:32:32-f64:32:64-v64:64:64-v128:128:128-a0:0:64-f80:128:128"
   7
   8 declare i8* @memset(i8*, i32, i32)
   9 declare void @llvm.memset.p0i8.i32(i8* nocapture writeonly, i8, i32, i32, i1)
  10 declare noalias i8* @malloc(i32) #1
  11
  12 ; Check memset(mem1, val, size) -> llvm.memset(mem1, val, size, 1).
  13
  14 define i8* @test_simplify1(i8* %mem, i32 %val, i32 %size) {
  15 ; CHECK-LABEL: @test_simplify1(
  16 ; CHECK-NEXT:    [[TMP1:%.*]] = trunc i32 [[VAL:%.*]] to i8
  17 ; CHECK-NEXT:    call void @llvm.memset.p0i8.i32(i8* align 1 [[MEM:%.*]], i8 [[TMP1]], i32 [[SIZE:%.*]], i1 false)
  18 ; CHECK-NEXT:    ret i8* [[MEM]]
  19 ;
  20   %ret = call i8* @memset(i8* %mem, i32 %val, i32 %size)
  21   ret i8* %ret
  22 }
  23
  24 define i8* @pr25892_lite(i32 %size) #0 {
  25 ; CHECK-LABEL: @pr25892_lite(
  26 ; CHECK-NEXT:    [[CALLOC:%.*]] = call i8* @calloc(i32 1, i32 [[SIZE:%.*]])
  27 ; CHECK-NEXT:    ret i8* [[CALLOC]]
  28 ;
  29   %call1 = call i8* @malloc(i32 %size) #1
  30   %call2 = call i8* @memset(i8* %call1, i32 0, i32 %size) #1
  31   ret i8* %call2
  32 }
  33
  34 ; FIXME: A memset intrinsic should be handled similarly to a memset() libcall.
  35
  36 define i8* @malloc_and_memset_intrinsic(i32 %n) #0 {
  37 ; CHECK-LABEL: @malloc_and_memset_intrinsic(
  38 ; CHECK-NEXT:    [[CALL:%.*]] = call i8* @malloc(i32 [[N:%.*]])
  39 ; CHECK-NEXT:    call void @llvm.memset.p0i8.i32(i8* align 1 [[CALL]], i8 0, i32 [[N]], i1 false)
  40 ; CHECK-NEXT:    ret i8* [[CALL]]
  41 ;
  42   %call = call i8* @malloc(i32 %n)
  43   call void @llvm.memset.p0i8.i32(i8* %call, i8 0, i32 %n, i32 1, i1 false)
  44   ret i8* %call
  45 }
  46
  47 ; This should not create a calloc and should not crash the compiler.
  48
  49 define i8* @notmalloc_memset(i32 %size, i8*(i32)* %notmalloc) {
  50 ; CHECK-LABEL: @notmalloc_memset(
  51 ; CHECK-NEXT:    [[CALL1:%.*]] = call i8* [[NOTMALLOC:%.*]](i32 [[SIZE:%.*]]) #0
  52 ; CHECK-NEXT:    call void @llvm.memset.p0i8.i32(i8* align 1 [[CALL1]], i8 0, i32 [[SIZE]], i1 false)
  53 ; CHECK-NEXT:    ret i8* [[CALL1]]
  54 ;
  55   %call1 = call i8* %notmalloc(i32 %size) #1
  56   %call2 = call i8* @memset(i8* %call1, i32 0, i32 %size) #1
  57   ret i8* %call2
  58 }
  59
  60 ; FIXME: memset(malloc(x), 0, x) -> calloc(1, x)
  61 ; This doesn't fire currently because the malloc has more than one use.
  62
  63 define float* @pr25892(i32 %size) #0 {
  64 ; CHECK-LABEL: @pr25892(
  65 ; CHECK-NEXT:  entry:
  66 ; CHECK-NEXT:    [[CALL:%.*]] = tail call i8* @malloc(i32 [[SIZE:%.*]]) #0
  67 ; CHECK-NEXT:    [[CMP:%.*]] = icmp eq i8* [[CALL]], null
  68 ; CHECK-NEXT:    br i1 [[CMP]], label [[CLEANUP:%.*]], label [[IF_END:%.*]]
  69 ; CHECK:       if.end:
  70 ; CHECK-NEXT:    [[BC:%.*]] = bitcast i8* [[CALL]] to float*
  71 ; CHECK-NEXT:    call void @llvm.memset.p0i8.i32(i8* nonnull align 1 [[CALL]], i8 0, i32 [[SIZE]], i1 false)
  72 ; CHECK-NEXT:    br label [[CLEANUP]]
  73 ; CHECK:       cleanup:
  74 ; CHECK-NEXT:    [[RETVAL_0:%.*]] = phi float* [ [[BC]], [[IF_END]] ], [ null, [[ENTRY:%.*]] ]
  75 ; CHECK-NEXT:    ret float* [[RETVAL_0]]
  76 ;
  77 entry:
  78   %call = tail call i8* @malloc(i32 %size) #1
  79   %cmp = icmp eq i8* %call, null
  80   br i1 %cmp, label %cleanup, label %if.end
  81 if.end:
  82   %bc = bitcast i8* %call to float*
  83   %call2 = tail call i8* @memset(i8* nonnull %call, i32 0, i32 %size) #1
  84   br label %cleanup
  85 cleanup:
  86   %retval.0 = phi float* [ %bc, %if.end ], [ null, %entry ]
  87   ret float* %retval.0
  88 }
  89
  90 ; If there's a calloc transform, the store must also be eliminated.
  91
  92 define i8* @buffer_is_modified_then_memset(i32 %size) {
  93 ; CHECK-LABEL: @buffer_is_modified_then_memset(
  94 ; CHECK-NEXT:    [[PTR:%.*]] = tail call i8* @malloc(i32 [[SIZE:%.*]]) #0
  95 ; CHECK-NEXT:    store i8 1, i8* [[PTR]], align 1
  96 ; CHECK-NEXT:    call void @llvm.memset.p0i8.i32(i8* align 1 [[PTR]], i8 0, i32 [[SIZE]], i1 false)
  97 ; CHECK-NEXT:    ret i8* [[PTR]]
  98 ;
  99   %ptr = tail call i8* @malloc(i32 %size) #1
 100   store i8 1, i8* %ptr           ;; fdata[0] = 1;
 101   %memset = tail call i8* @memset(i8* nonnull %ptr, i32 0, i32 %size) #1
 102   ret i8* %memset
 103 }
 104
 105 attributes #0 = { nounwind ssp uwtable }
 106 attributes #1 = { nounwind }
 107 attributes #2 = { nounwind readnone }
 108