void count_fold_samples(int *ntrain, int *ntest, double *ntrainrecip,
                        double *ntestrecip, int *folds, int nfolds, int n) {
  int i, k;
  for (k = 0; k < nfolds; k++) {
    ntrain[k] = 0;
    for (i = 0; i < n; i++)
      ntrain[k] += folds[n * k + i];
    ntest[k] = n - ntrain[k];
    ntrainrecip[k] = 0;
    if (ntrain[k] > 0)
      ntrainrecip[k] = 1.0 / ntrain[k];
    ntestrecip[k] = 0;
    if (ntest[k] > 0)
      ntestrecip[k] = 1.0 / ntest[k];
  }
}


x86 assembly (clang, -O0):
<count_fold_samples>:
push   %rbp
mov    %rsp,%rbp
mov    0x10(%rbp),%eax
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %rdx,-0x18(%rbp)
mov    %rcx,-0x20(%rbp)
mov    %r8,-0x28(%rbp)
mov    %r9d,-0x2c(%rbp)
movl   $0x0,-0x34(%rbp)
mov    -0x34(%rbp),%eax
cmp    -0x2c(%rbp),%eax
jge    1240 <count_fold_samples+0x140>
mov    -0x8(%rbp),%rax
movslq -0x34(%rbp),%rcx
movl   $0x0,(%rax,%rcx,4)
movl   $0x0,-0x30(%rbp)
mov    -0x30(%rbp),%eax
cmp    0x10(%rbp),%eax
jge    1184 <count_fold_samples+0x84>
mov    -0x28(%rbp),%rax
mov    0x10(%rbp),%ecx
imul   -0x34(%rbp),%ecx
add    -0x30(%rbp),%ecx
movslq %ecx,%rcx
mov    (%rax,%rcx,4),%edx
mov    -0x8(%rbp),%rax
movslq -0x34(%rbp),%rcx
add    (%rax,%rcx,4),%edx
mov    %edx,(%rax,%rcx,4)
mov    -0x30(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x30(%rbp)
jmp    1148 <count_fold_samples+0x48>
mov    0x10(%rbp),%edx
mov    -0x8(%rbp),%rax
movslq -0x34(%rbp),%rcx
sub    (%rax,%rcx,4),%edx
mov    -0x10(%rbp),%rax
movslq -0x34(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x18(%rbp),%rax
movslq -0x34(%rbp),%rcx
xorps  %xmm0,%xmm0
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x8(%rbp),%rax
movslq -0x34(%rbp),%rcx
cmpl   $0x0,(%rax,%rcx,4)
jle    11e5 <count_fold_samples+0xe5>
mov    -0x8(%rbp),%rax
movslq -0x34(%rbp),%rcx
cvtsi2sdl (%rax,%rcx,4),%xmm1
movsd  0xe2c(%rip),%xmm0
divsd  %xmm1,%xmm0
mov    -0x18(%rbp),%rax
movslq -0x34(%rbp),%rcx
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x20(%rbp),%rax
movslq -0x34(%rbp),%rcx
xorps  %xmm0,%xmm0
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x10(%rbp),%rax
movslq -0x34(%rbp),%rcx
cmpl   $0x0,(%rax,%rcx,4)
jle    122d <count_fold_samples+0x12d>
mov    -0x10(%rbp),%rax
movslq -0x34(%rbp),%rcx
cvtsi2sdl (%rax,%rcx,4),%xmm1
movsd  0xde4(%rip),%xmm0
divsd  %xmm1,%xmm0
mov    -0x20(%rbp),%rax
movslq -0x34(%rbp),%rcx
movsd  %xmm0,(%rax,%rcx,8)
jmp    1232 <count_fold_samples+0x132>
mov    -0x34(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x34(%rbp)
jmp    1126 <count_fold_samples+0x26>
pop    %rbp
ret

x86 assembly (clang, -O2):
<count_fold_samples>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
test   %r9d,%r9d
jle    123d <count_fold_samples+0x13d>
mov    0x38(%rsp),%r10d
movslq %r10d,%r11
mov    %r9d,%r9d
mov    %r11d,%r15d
lea    -0x1(%r15),%rax
mov    %rax,-0x8(%rsp)
mov    %r15d,%r12d
and    $0x3,%r12d
and    $0xfffffffc,%r15d
lea    0xc(%r8),%r13
shl    $0x2,%r11
xor    %eax,%eax
movsd  0xeb9(%rip),%xmm0
jmp    1177 <count_fold_samples+0x77>
nopl   0x0(%rax)
xorps  %xmm2,%xmm2
cvtsi2sd %ebp,%xmm2
movapd %xmm0,%xmm1
divsd  %xmm2,%xmm1
movsd  %xmm1,(%rcx,%rax,8)
add    $0x1,%rax
add    %r11,%r13
add    %r11,%r8
cmp    %r9,%rax
je     123d <count_fold_samples+0x13d>
movl   $0x0,(%rdi,%rax,4)
mov    $0x0,%ebx
test   %r10d,%r10d
jle    1201 <count_fold_samples+0x101>
cmpq   $0x3,-0x8(%rsp)
jae    11a0 <count_fold_samples+0xa0>
xor    %ebx,%ebx
xor    %ebp,%ebp
jmp    11d9 <count_fold_samples+0xd9>
cs nopw 0x0(%rax,%rax,1)
xor    %ebx,%ebx
xor    %ebp,%ebp
cs nopw 0x0(%rax,%rax,1)
xchg   %ax,%ax
add    -0xc(%r13,%rbp,4),%ebx
mov    %ebx,(%rdi,%rax,4)
add    -0x8(%r13,%rbp,4),%ebx
mov    %ebx,(%rdi,%rax,4)
add    -0x4(%r13,%rbp,4),%ebx
mov    %ebx,(%rdi,%rax,4)
add    0x0(%r13,%rbp,4),%ebx
mov    %ebx,(%rdi,%rax,4)
add    $0x4,%rbp
cmp    %rbp,%r15
jne    11b0 <count_fold_samples+0xb0>
test   %r12,%r12
je     1201 <count_fold_samples+0x101>
lea    (%r8,%rbp,4),%rbp
xor    %r14d,%r14d
cs nopw 0x0(%rax,%rax,1)
nop
add    0x0(%rbp,%r14,4),%ebx
mov    %ebx,(%rdi,%rax,4)
add    $0x1,%r14
cmp    %r14,%r12
jne    11f0 <count_fold_samples+0xf0>
mov    %r10d,%ebp
sub    %ebx,%ebp
mov    %ebp,(%rsi,%rax,4)
movq   $0x0,(%rdx,%rax,8)
mov    (%rdi,%rax,4),%ebx
test   %ebx,%ebx
jle    122c <count_fold_samples+0x12c>
xorps  %xmm1,%xmm1
cvtsi2sd %ebx,%xmm1
movapd %xmm0,%xmm2
divsd  %xmm1,%xmm2
movsd  %xmm2,(%rdx,%rax,8)
test   %ebp,%ebp
jg     1150 <count_fold_samples+0x50>
xorpd  %xmm1,%xmm1
jmp    115f <count_fold_samples+0x5f>
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define void @count_fold_samples(i32* %0, i32* %1, double* %2, double* %3, i32* %4, i32 %5, i32 %6) {
%8 = alloca i32*
%9 = alloca i32*
%10 = alloca double*
%11 = alloca double*
%12 = alloca i32*
%13 = alloca i32
%14 = alloca i32
%15 = alloca i32
%16 = alloca i32
store i32* %0, i32** %8
store i32* %1, i32** %9
store double* %2, double** %10
store double* %3, double** %11
store i32* %4, i32** %12
store i32 %5, i32* %13
store i32 %6, i32* %14
store i32 0, i32* %16
br label %17
17: ; preds = %107, %7
%18 = load i32, i32* %16
%19 = load i32, i32* %13
%20 = icmp slt i32 %18, %19
br i1 %20, label %21, label %110
21: ; preds = %17
%22 = load i32*, i32** %8
%23 = load i32, i32* %16
%24 = sext i32 %23 to i64
%25 = getelementptr inbounds i32, i32* %22, i64 %24
store i32 0, i32* %25
store i32 0, i32* %15
br label %26
26: ; preds = %46, %21
%27 = load i32, i32* %15
%28 = load i32, i32* %14
%29 = icmp slt i32 %27, %28
br i1 %29, label %30, label %49
30: ; preds = %26
%31 = load i32*, i32** %12
%32 = load i32, i32* %14
%33 = load i32, i32* %16
%34 = mul nsw i32 %32, %33
%35 = load i32, i32* %15
%36 = add nsw i32 %34, %35
%37 = sext i32 %36 to i64
%38 = getelementptr inbounds i32, i32* %31, i64 %37
%39 = load i32, i32* %38
%40 = load i32*, i32** %8
%41 = load i32, i32* %16
%42 = sext i32 %41 to i64
%43 = getelementptr inbounds i32, i32* %40, i64 %42
%44 = load i32, i32* %43
%45 = add nsw i32 %44, %39
store i32 %45, i32* %43
br label %46
46: ; preds = %30
%47 = load i32, i32* %15
%48 = add nsw i32 %47, 1
store i32 %48, i32* %15
br label %26
49: ; preds = %26
%50 = load i32, i32* %14
%51 = load i32*, i32** %8
%52 = load i32, i32* %16
%53 = sext i32 %52 to i64
%54 = getelementptr inbounds i32, i32* %51, i64 %53
%55 = load i32, i32* %54
%56 = sub nsw i32 %50, %55
%57 = load i32*, i32** %9
%58 = load i32, i32* %16
%59 = sext i32 %58 to i64
%60 = getelementptr inbounds i32, i32* %57, i64 %59
store i32 %56, i32* %60
%61 = load double*, double** %10
%62 = load i32, i32* %16
%63 = sext i32 %62 to i64
%64 = getelementptr inbounds double, double* %61, i64 %63
store double 0.000000e+00, double* %64
%65 = load i32*, i32** %8
%66 = load i32, i32* %16
%67 = sext i32 %66 to i64
%68 = getelementptr inbounds i32, i32* %65, i64 %67
%69 = load i32, i32* %68
%70 = icmp sgt i32 %69, 0
br i1 %70, label %71, label %83
71: ; preds = %49
%72 = load i32*, i32** %8
%73 = load i32, i32* %16
%74 = sext i32 %73 to i64
%75 = getelementptr inbounds i32, i32* %72, i64 %74
%76 = load i32, i32* %75
%77 = sitofp i32 %76 to double
%78 = fdiv double 1.000000e+00, %77
%79 = load double*, double** %10
%80 = load i32, i32* %16
%81 = sext i32 %80 to i64
%82 = getelementptr inbounds double, double* %79, i64 %81
store double %78, double* %82
br label %83
83: ; preds = %71, %49
%84 = load double*, double** %11
%85 = load i32, i32* %16
%86 = sext i32 %85 to i64
%87 = getelementptr inbounds double, double* %84, i64 %86
store double 0.000000e+00, double* %87
%88 = load i32*, i32** %9
%89 = load i32, i32* %16
%90 = sext i32 %89 to i64
%91 = getelementptr inbounds i32, i32* %88, i64 %90
%92 = load i32, i32* %91
%93 = icmp sgt i32 %92, 0
br i1 %93, label %94, label %106
94: ; preds = %83
%95 = load i32*, i32** %9
%96 = load i32, i32* %16
%97 = sext i32 %96 to i64
%98 = getelementptr inbounds i32, i32* %95, i64 %97
%99 = load i32, i32* %98
%100 = sitofp i32 %99 to double
%101 = fdiv double 1.000000e+00, %100
%102 = load double*, double** %11
%103 = load i32, i32* %16
%104 = sext i32 %103 to i64
%105 = getelementptr inbounds double, double* %102, i64 %104
store double %101, double* %105
br label %106
106: ; preds = %94, %83
br label %107
107: ; preds = %106
%108 = load i32, i32* %16
%109 = add nsw i32 %108, 1
store i32 %109, i32* %16
br label %17
110: ; preds = %17
ret void
}

LLVM IR (-O2):
define void @count_fold_samples(i32* nocapture %0, i32* nocapture %1, double* nocapture writeonly %2, double* nocapture writeonly %3, i32* nocapture %4, i32 %5, i32 %6) {
%8 = icmp sgt i32 %5, 0
br i1 %8, label %9, label %85
9: ; preds = %7
%10 = icmp sgt i32 %6, 0
%11 = sext i32 %6 to i64
%12 = zext i32 %5 to i64
%13 = zext i32 %6 to i64
%14 = add nsw i64 %13, -1
%15 = and i64 %13, 3
%16 = icmp ult i64 %14, 3
%17 = and i64 %13, 4294967292
%18 = icmp eq i64 %15, 0
br label %19
19: ; preds = %9, %81
%20 = phi i64 [ 0, %9 ], [ %83, %81 ]
%21 = getelementptr inbounds i32, i32* %0, i64 %20
store i32 0, i32* %21
br i1 %10, label %22, label %65
22: ; preds = %19
%23 = mul nsw i64 %20, %11
br i1 %16, label %50, label %24
24: ; preds = %22, %24
%25 = phi i32 [ %46, %24 ], [ 0, %22 ]
%26 = phi i64 [ %47, %24 ], [ 0, %22 ]
%27 = phi i64 [ %48, %24 ], [ 0, %22 ]
%28 = add nsw i64 %26, %23
%29 = getelementptr inbounds i32, i32* %4, i64 %28
%30 = load i32, i32* %29
%31 = add nsw i32 %25, %30
store i32 %31, i32* %21
%32 = or i64 %26, 1
%33 = add nsw i64 %32, %23
%34 = getelementptr inbounds i32, i32* %4, i64 %33
%35 = load i32, i32* %34
%36 = add nsw i32 %31, %35
store i32 %36, i32* %21
%37 = or i64 %26, 2
%38 = add nsw i64 %37, %23
%39 = getelementptr inbounds i32, i32* %4, i64 %38
%40 = load i32, i32* %39
%41 = add nsw i32 %36, %40
store i32 %41, i32* %21
%42 = or i64 %26, 3
%43 = add nsw i64 %42, %23
%44 = getelementptr inbounds i32, i32* %4, i64 %43
%45 = load i32, i32* %44
%46 = add nsw i32 %41, %45
store i32 %46, i32* %21
%47 = add nuw nsw i64 %26, 4
%48 = add i64 %27, 4
%49 = icmp eq i64 %48, %17
br i1 %49, label %50, label %24
50: ; preds = %24, %22
%51 = phi i32 [ undef, %22 ], [ %46, %24 ]
%52 = phi i32 [ 0, %22 ], [ %46, %24 ]
%53 = phi i64 [ 0, %22 ], [ %47, %24 ]
br i1 %18, label %65, label %54
54: ; preds = %50, %54
%55 = phi i32 [ %61, %54 ], [ %52, %50 ]
%56 = phi i64 [ %62, %54 ], [ %53, %50 ]
%57 = phi i64 [ %63, %54 ], [ 0, %50 ]
%58 = add nsw i64 %56, %23
%59 = getelementptr inbounds i32, i32* %4, i64 %58
%60 = load i32, i32* %59
%61 = add nsw i32 %55, %60
store i32 %61, i32* %21
%62 = add nuw nsw i64 %56, 1
%63 = add i64 %57, 1
%64 = icmp eq i64 %63, %15
br i1 %64, label %65, label %54
65: ; preds = %50, %54, %19
%66 = phi i32 [ 0, %19 ], [ %51, %50 ], [ %61, %54 ]
%67 = sub nsw i32 %6, %66
%68 = getelementptr inbounds i32, i32* %1, i64 %20
store i32 %67, i32* %68
%69 = getelementptr inbounds double, double* %2, i64 %20
store double 0.000000e+00, double* %69
%70 = load i32, i32* %21
%71 = icmp sgt i32 %70, 0
br i1 %71, label %72, label %75
72: ; preds = %65
%73 = sitofp i32 %70 to double
%74 = fdiv double 1.000000e+00, %73
store double %74, double* %69
br label %75
75: ; preds = %72, %65
%76 = getelementptr inbounds double, double* %3, i64 %20
%77 = icmp sgt i32 %67, 0
br i1 %77, label %78, label %81
78: ; preds = %75
%79 = sitofp i32 %67 to double
%80 = fdiv double 1.000000e+00, %79
br label %81
81: ; preds = %75, %78
%82 = phi double [ %80, %78 ], [ 0.000000e+00, %75 ]
store double %82, double* %76
%83 = add nuw nsw i64 %20, 1
%84 = icmp eq i64 %83, %12
br i1 %84, label %85, label %19
85: ; preds = %81, %7
ret void
}