#define NULL ((void *)0)
typedef unsigned long size_t; // Customize by platform.
typedef long intptr_t;
typedef unsigned long uintptr_t;
typedef long scalar_t__; // Either arithmetic or pointer type.
/* By default, we understand bool (as a convenience). */
typedef int bool;
#define false 0
#define true 1

/* Forward declarations */

/* Type definitions */

/* Variables and functions */
scalar_t__ malloc(unsigned int);
scalar_t__ rand_long_long(int /*<<< orphan*/, unsigned int);
unsigned int *build_training_set(unsigned int set_size,
                                 unsigned int subset_size) {
  unsigned int *train, i;
  train = (unsigned int *)malloc(set_size * sizeof(unsigned int));
  for (i = 0; i < set_size; i++)
    train[i] = i;
  for (i = 0; i < subset_size; i++) {
    unsigned int p;
    unsigned int t;
    p = (unsigned int)rand_long_long(0, set_size - 1);
    t = train[i];
    train[i] = train[p];
    train[p] = t;
  }
  return train;
}


x86 assembly (clang, -O0):
<build_training_set>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %edi,-0x4(%rbp)
mov    %esi,-0x8(%rbp)
mov    -0x4(%rbp),%eax
shl    $0x2,%rax
mov    %eax,%edi
call   1040 <malloc@plt>
mov    %rax,-0x10(%rbp)
movl   $0x0,-0x14(%rbp)
mov    -0x14(%rbp),%eax
cmp    -0x4(%rbp),%eax
jae    116e <build_training_set+0x4e>
mov    -0x14(%rbp),%edx
mov    -0x10(%rbp),%rax
mov    -0x14(%rbp),%ecx
mov    %edx,(%rax,%rcx,4)
mov    -0x14(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x14(%rbp)
jmp    1147 <build_training_set+0x27>
movl   $0x0,-0x14(%rbp)
mov    -0x14(%rbp),%eax
cmp    -0x8(%rbp),%eax
jae    11cd <build_training_set+0xad>
mov    -0x4(%rbp),%esi
sub    $0x1,%esi
xor    %edi,%edi
call   1030 <rand_long_long@plt>
mov    %eax,-0x18(%rbp)
mov    -0x10(%rbp),%rax
mov    -0x14(%rbp),%ecx
mov    (%rax,%rcx,4),%eax
mov    %eax,-0x1c(%rbp)
mov    -0x10(%rbp),%rax
mov    -0x18(%rbp),%ecx
mov    (%rax,%rcx,4),%edx
mov    -0x10(%rbp),%rax
mov    -0x14(%rbp),%ecx
mov    %edx,(%rax,%rcx,4)
mov    -0x1c(%rbp),%edx
mov    -0x10(%rbp),%rax
mov    -0x18(%rbp),%ecx
mov    %edx,(%rax,%rcx,4)
mov    -0x14(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x14(%rbp)
jmp    1175 <build_training_set+0x55>
mov    -0x10(%rbp),%rax
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<build_training_set>:
push   %rbp
push   %r15
push   %r14
push   %rbx
push   %rax
mov    %esi,%ebp
mov    %edi,%r14d
lea    0x0(,%r14,4),%edi
call   1040 <malloc@plt>
mov    %rax,%rbx
test   %r14d,%r14d
je     12bc <build_training_set+0x19c>
mov    %r14d,%eax
cmp    $0x8,%r14d
jae    1155 <build_training_set+0x35>
xor    %ecx,%ecx
jmp    12b0 <build_training_set+0x190>
mov    %eax,%ecx
and    $0xfffffff8,%ecx
lea    -0x8(%rcx),%rdi
mov    %rdi,%rsi
shr    $0x3,%rsi
add    $0x1,%rsi
mov    %esi,%edx
and    $0x3,%edx
cmp    $0x18,%rdi
jae    1183 <build_training_set+0x63>
movdqa 0xe84(%rip),%xmm0
xor    %edi,%edi
jmp    1258 <build_training_set+0x138>
and    $0xfffffffffffffffc,%rsi
movdqa 0xe71(%rip),%xmm0
xor    %edi,%edi
movdqa 0xe76(%rip),%xmm8
movdqa 0xe7d(%rip),%xmm9
movdqa 0xe84(%rip),%xmm10
movdqa 0xe8c(%rip),%xmm4
movdqa 0xe94(%rip),%xmm5
movdqa 0xe9c(%rip),%xmm6
movdqa 0xea4(%rip),%xmm7
movdqa 0xeac(%rip),%xmm1
cs nopw 0x0(%rax,%rax,1)
xchg   %ax,%ax
movdqa %xmm0,%xmm2
paddd  %xmm8,%xmm2
movdqu %xmm0,(%rbx,%rdi,4)
movdqu %xmm2,0x10(%rbx,%rdi,4)
movdqa %xmm0,%xmm2
paddd  %xmm9,%xmm2
movdqa %xmm0,%xmm3
paddd  %xmm10,%xmm3
movdqu %xmm2,0x20(%rbx,%rdi,4)
movdqu %xmm3,0x30(%rbx,%rdi,4)
movdqa %xmm0,%xmm2
paddd  %xmm4,%xmm2
movdqa %xmm0,%xmm3
paddd  %xmm5,%xmm3
movdqu %xmm2,0x40(%rbx,%rdi,4)
movdqu %xmm3,0x50(%rbx,%rdi,4)
movdqa %xmm0,%xmm2
paddd  %xmm6,%xmm2
movdqa %xmm0,%xmm3
paddd  %xmm7,%xmm3
movdqu %xmm2,0x60(%rbx,%rdi,4)
movdqu %xmm3,0x70(%rbx,%rdi,4)
add    $0x20,%rdi
paddd  %xmm1,%xmm0
add    $0xfffffffffffffffc,%rsi
jne    11e0 <build_training_set+0xc0>
test   %rdx,%rdx
je     12a0 <build_training_set+0x180>
lea    (%rbx,%rdi,4),%rsi
add    $0x10,%rsi
shl    $0x5,%rdx
xor    %edi,%edi
movdqa 0xd9d(%rip),%xmm1
movdqa 0xda5(%rip),%xmm2
nopl   0x0(%rax,%rax,1)
movdqa %xmm0,%xmm3
paddd  %xmm1,%xmm3
movdqu %xmm0,-0x10(%rsi,%rdi,1)
movdqu %xmm3,(%rsi,%rdi,1)
paddd  %xmm2,%xmm0
add    $0x20,%rdi
cmp    %rdi,%rdx
jne    1280 <build_training_set+0x160>
cmp    %rax,%rcx
je     12bc <build_training_set+0x19c>
cs nopw 0x0(%rax,%rax,1)
nop
mov    %ecx,(%rbx,%rcx,4)
add    $0x1,%rcx
cmp    %rcx,%rax
jne    12b0 <build_training_set+0x190>
test   %ebp,%ebp
je     12f1 <build_training_set+0x1d1>
add    $0xffffffff,%r14d
mov    %ebp,%r15d
xor    %ebp,%ebp
nopl   0x0(%rax)
xor    %edi,%edi
mov    %r14d,%esi
call   1030 <rand_long_long@plt>
mov    (%rbx,%rbp,4),%ecx
mov    %eax,%eax
mov    (%rbx,%rax,4),%edx
mov    %edx,(%rbx,%rbp,4)
mov    %ecx,(%rbx,%rax,4)
add    $0x1,%rbp
cmp    %rbp,%r15
jne    12d0 <build_training_set+0x1b0>
mov    %rbx,%rax
add    $0x8,%rsp
pop    %rbx
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define i32* @build_training_set(i32 %0, i32 %1) {
%3 = alloca i32
%4 = alloca i32
%5 = alloca i32*
%6 = alloca i32
%7 = alloca i32
%8 = alloca i32
store i32 %0, i32* %3
store i32 %1, i32* %4
%9 = load i32, i32* %3
%10 = zext i32 %9 to i64
%11 = mul i64 %10, 4
%12 = trunc i64 %11 to i32
%13 = call i64 @malloc(i32 %12)
%14 = inttoptr i64 %13 to i32*
store i32* %14, i32** %5
store i32 0, i32* %6
br label %15
15: ; preds = %25, %2
%16 = load i32, i32* %6
%17 = load i32, i32* %3
%18 = icmp ult i32 %16, %17
br i1 %18, label %19, label %28
19: ; preds = %15
%20 = load i32, i32* %6
%21 = load i32*, i32** %5
%22 = load i32, i32* %6
%23 = zext i32 %22 to i64
%24 = getelementptr inbounds i32, i32* %21, i64 %23
store i32 %20, i32* %24
br label %25
25: ; preds = %19
%26 = load i32, i32* %6
%27 = add i32 %26, 1
store i32 %27, i32* %6
br label %15
28: ; preds = %15
store i32 0, i32* %6
br label %29
29: ; preds = %57, %28
%30 = load i32, i32* %6
%31 = load i32, i32* %4
%32 = icmp ult i32 %30, %31
br i1 %32, label %33, label %60
33: ; preds = %29
%34 = load i32, i32* %3
%35 = sub i32 %34, 1
%36 = call i64 @rand_long_long(i32 0, i32 %35)
%37 = trunc i64 %36 to i32
store i32 %37, i32* %7
%38 = load i32*, i32** %5
%39 = load i32, i32* %6
%40 = zext i32 %39 to i64
%41 = getelementptr inbounds i32, i32* %38, i64 %40
%42 = load i32, i32* %41
store i32 %42, i32* %8
%43 = load i32*, i32** %5
%44 = load i32, i32* %7
%45 = zext i32 %44 to i64
%46 = getelementptr inbounds i32, i32* %43, i64 %45
%47 = load i32, i32* %46
%48 = load i32*, i32** %5
%49 = load i32, i32* %6
%50 = zext i32 %49 to i64
%51 = getelementptr inbounds i32, i32* %48, i64 %50
store i32 %47, i32* %51
%52 = load i32, i32* %8
%53 = load i32*, i32** %5
%54 = load i32, i32* %7
%55 = zext i32 %54 to i64
%56 = getelementptr inbounds i32, i32* %53, i64 %55
store i32 %52, i32* %56
br label %57
57: ; preds = %33
%58 = load i32, i32* %6
%59 = add i32 %58, 1
store i32 %59, i32* %6
br label %29
60: ; preds = %29
%61 = load i32*, i32** %5
ret i32* %61
}

LLVM IR (-O2):
define i32* @build_training_set(i32 %0, i32 %1) {
%3 = shl i32 %0, 2
%4 = tail call i64 @malloc(i32 %3)
%5 = inttoptr i64 %4 to i32*
%6 = icmp eq i32 %0, 0
br i1 %6, label %74, label %7
7: ; preds = %2
%8 = zext i32 %0 to i64
%9 = icmp ult i32 %0, 8
br i1 %9, label %72, label %10
10: ; preds = %7
%11 = and i64 %8, 4294967288
%12 = add nsw i64 %11, -8
%13 = lshr exact i64 %12, 3
%14 = add nuw nsw i64 %13, 1
%15 = and i64 %14, 3
%16 = icmp ult i64 %12, 24
br i1 %16, label %53, label %17
17: ; preds = %10
%18 = and i64 %14, 4611686018427387900
br label %19
19: ; preds = %19, %17
%20 = phi i64 [ 0, %17 ], [ %49, %19 ]
%21 = phi <4 x i32> [ <i32 0, i32 1, i32 2, i32 3>, %17 ], [ %50, %19 ]
%22 = phi i64 [ 0, %17 ], [ %51, %19 ]
%23 = add <4 x i32> %21, <i32 4, i32 4, i32 4, i32 4>
%24 = getelementptr inbounds i32, i32* %5, i64 %20
%25 = bitcast i32* %24 to <4 x i32>*
store <4 x i32> %21, <4 x i32>* %25
%26 = getelementptr inbounds i32, i32* %24, i64 4
%27 = bitcast i32* %26 to <4 x i32>*
store <4 x i32> %23, <4 x i32>* %27
%28 = or i64 %20, 8
%29 = add <4 x i32> %21, <i32 8, i32 8, i32 8, i32 8>
%30 = add <4 x i32> %21, <i32 12, i32 12, i32 12, i32 12>
%31 = getelementptr inbounds i32, i32* %5, i64 %28
%32 = bitcast i32* %31 to <4 x i32>*
store <4 x i32> %29, <4 x i32>* %32
%33 = getelementptr inbounds i32, i32* %31, i64 4
%34 = bitcast i32* %33 to <4 x i32>*
store <4 x i32> %30, <4 x i32>* %34
%35 = or i64 %20, 16
%36 = add <4 x i32> %21, <i32 16, i32 16, i32 16, i32 16>
%37 = add <4 x i32> %21, <i32 20, i32 20, i32 20, i32 20>
%38 = getelementptr inbounds i32, i32* %5, i64 %35
%39 = bitcast i32* %38 to <4 x i32>*
store <4 x i32> %36, <4 x i32>* %39
%40 = getelementptr inbounds i32, i32* %38, i64 4
%41 = bitcast i32* %40 to <4 x i32>*
store <4 x i32> %37, <4 x i32>* %41
%42 = or i64 %20, 24
%43 = add <4 x i32> %21, <i32 24, i32 24, i32 24, i32 24>
%44 = add <4 x i32> %21, <i32 28, i32 28, i32 28, i32 28>
%45 = getelementptr inbounds i32, i32* %5, i64 %42
%46 = bitcast i32* %45 to <4 x i32>*
store <4 x i32> %43, <4 x i32>* %46
%47 = getelementptr inbounds i32, i32* %45, i64 4
%48 = bitcast i32* %47 to <4 x i32>*
store <4 x i32> %44, <4 x i32>* %48
%49 = add nuw i64 %20, 32
%50 = add <4 x i32> %21, <i32 32, i32 32, i32 32, i32 32>
%51 = add i64 %22, 4
%52 = icmp eq i64 %51, %18
br i1 %52, label %53, label %19
53: ; preds = %19, %10
%54 = phi i64 [ 0, %10 ], [ %49, %19 ]
%55 = phi <4 x i32> [ <i32 0, i32 1, i32 2, i32 3>, %10 ], [ %50, %19 ]
%56 = icmp eq i64 %15, 0
br i1 %56, label %70, label %57
57: ; preds = %53, %57
%58 = phi i64 [ %66, %57 ], [ %54, %53 ]
%59 = phi <4 x i32> [ %67, %57 ], [ %55, %53 ]
%60 = phi i64 [ %68, %57 ], [ 0, %53 ]
%61 = add <4 x i32> %59, <i32 4, i32 4, i32 4, i32 4>
%62 = getelementptr inbounds i32, i32* %5, i64 %58
%63 = bitcast i32* %62 to <4 x i32>*
store <4 x i32> %59, <4 x i32>* %63
%64 = getelementptr inbounds i32, i32* %62, i64 4
%65 = bitcast i32* %64 to <4 x i32>*
store <4 x i32> %61, <4 x i32>* %65
%66 = add nuw i64 %58, 8
%67 = add <4 x i32> %59, <i32 8, i32 8, i32 8, i32 8>
%68 = add i64 %60, 1
%69 = icmp eq i64 %68, %15
br i1 %69, label %70, label %57
70: ; preds = %57, %53
%71 = icmp eq i64 %11, %8
br i1 %71, label %74, label %72
72: ; preds = %7, %70
%73 = phi i64 [ 0, %7 ], [ %11, %70 ]
br label %79
74: ; preds = %79, %70, %2
%75 = icmp eq i32 %1, 0
br i1 %75, label %95, label %76
76: ; preds = %74
%77 = add i32 %0, -1
%78 = zext i32 %1 to i64
br label %85
79: ; preds = %72, %79
%80 = phi i64 [ %83, %79 ], [ %73, %72 ]
%81 = getelementptr inbounds i32, i32* %5, i64 %80
%82 = trunc i64 %80 to i32
store i32 %82, i32* %81
%83 = add nuw nsw i64 %80, 1
%84 = icmp eq i64 %83, %8
br i1 %84, label %74, label %79
85: ; preds = %76, %85
%86 = phi i64 [ 0, %76 ], [ %93, %85 ]
%87 = tail call i64 @rand_long_long(i32 0, i32 %77)
%88 = getelementptr inbounds i32, i32* %5, i64 %86
%89 = load i32, i32* %88
%90 = and i64 %87, 4294967295
%91 = getelementptr inbounds i32, i32* %5, i64 %90
%92 = load i32, i32* %91
store i32 %92, i32* %88
store i32 %89, i32* %91
%93 = add nuw nsw i64 %86, 1
%94 = icmp eq i64 %93, %78
br i1 %94, label %95, label %85
95: ; preds = %85, %74
ret i32* %5
}