void stratify(int *idx, int *stratIdx, int m, int nbFolds) {
  int j, start = 0;
  int nbInserted = 0;
  while (nbInserted < m) {
    j = start;
    while (j < m) {
      stratIdx[nbInserted++] = idx[j];
      j += nbFolds;
    }
    start++;
  }
}


x86 assembly (clang, -O0):
<stratify>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
mov    %ecx,-0x18(%rbp)
movl   $0x0,-0x20(%rbp)
movl   $0x0,-0x24(%rbp)
mov    -0x24(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    117a <stratify+0x7a>
mov    -0x20(%rbp),%eax
mov    %eax,-0x1c(%rbp)
mov    -0x1c(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    116c <stratify+0x6c>
mov    -0x8(%rbp),%rax
movslq -0x1c(%rbp),%rcx
mov    (%rax,%rcx,4),%edx
mov    -0x10(%rbp),%rax
mov    -0x24(%rbp),%ecx
mov    %ecx,%esi
add    $0x1,%esi
mov    %esi,-0x24(%rbp)
movslq %ecx,%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x18(%rbp),%eax
add    -0x1c(%rbp),%eax
mov    %eax,-0x1c(%rbp)
jmp    1132 <stratify+0x32>
mov    -0x20(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x20(%rbp)
jmp    1120 <stratify+0x20>
pop    %rbp
ret

x86 assembly (clang, -O2):
<stratify>:
push   %rbx
test   %edx,%edx
jle    1157 <stratify+0x57>
movslq %ecx,%r9
mov    %edx,%r10d
xor    %r8d,%r8d
xor    %r11d,%r11d
jmp    1129 <stratify+0x29>
cs nopw 0x0(%rax,%rax,1)
nopl   (%rax)
add    $0x1,%r8
cmp    %edx,%r11d
jge    1157 <stratify+0x57>
cmp    %edx,%r8d
jge    1120 <stratify+0x20>
movslq %r11d,%rax
lea    (%rsi,%rax,4),%rcx
mov    %r8,%rax
nopl   0x0(%rax,%rax,1)
mov    (%rdi,%rax,4),%ebx
mov    %ebx,(%rcx)
add    %r9,%rax
add    $0x4,%rcx
add    $0x1,%r11d
cmp    %r10,%rax
jl     1140 <stratify+0x40>
jmp    1120 <stratify+0x20>
pop    %rbx
ret

LLVM IR (-O0):
define void @stratify(i32* %0, i32* %1, i32 %2, i32 %3) {
%5 = alloca i32*
%6 = alloca i32*
%7 = alloca i32
%8 = alloca i32
%9 = alloca i32
%10 = alloca i32
%11 = alloca i32
store i32* %0, i32** %5
store i32* %1, i32** %6
store i32 %2, i32* %7
store i32 %3, i32* %8
store i32 0, i32* %10
store i32 0, i32* %11
br label %12
12: ; preds = %36, %4
%13 = load i32, i32* %11
%14 = load i32, i32* %7
%15 = icmp slt i32 %13, %14
br i1 %15, label %16, label %39
16: ; preds = %12
%17 = load i32, i32* %10
store i32 %17, i32* %9
br label %18
18: ; preds = %22, %16
%19 = load i32, i32* %9
%20 = load i32, i32* %7
%21 = icmp slt i32 %19, %20
br i1 %21, label %22, label %36
22: ; preds = %18
%23 = load i32*, i32** %5
%24 = load i32, i32* %9
%25 = sext i32 %24 to i64
%26 = getelementptr inbounds i32, i32* %23, i64 %25
%27 = load i32, i32* %26
%28 = load i32*, i32** %6
%29 = load i32, i32* %11
%30 = add nsw i32 %29, 1
store i32 %30, i32* %11
%31 = sext i32 %29 to i64
%32 = getelementptr inbounds i32, i32* %28, i64 %31
store i32 %27, i32* %32
%33 = load i32, i32* %8
%34 = load i32, i32* %9
%35 = add nsw i32 %34, %33
store i32 %35, i32* %9
br label %18
36: ; preds = %18
%37 = load i32, i32* %10
%38 = add nsw i32 %37, 1
store i32 %38, i32* %10
br label %12
39: ; preds = %12
ret void
}

LLVM IR (-O2):
define void @stratify(i32* nocapture %0, i32* nocapture writeonly %1, i32 %2, i32 %3) {
%5 = icmp sgt i32 %2, 0
br i1 %5, label %6, label %31
6: ; preds = %4
%7 = sext i32 %3 to i64
%8 = zext i32 %2 to i64
br label %9
9: ; preds = %6, %27
%10 = phi i64 [ 0, %6 ], [ %29, %27 ]
%11 = phi i32 [ 0, %6 ], [ %28, %27 ]
%12 = trunc i64 %10 to i32
%13 = icmp slt i32 %12, %2
br i1 %13, label %14, label %27
14: ; preds = %9
%15 = sext i32 %11 to i64
br label %16
16: ; preds = %14, %16
%17 = phi i64 [ %10, %14 ], [ %23, %16 ]
%18 = phi i64 [ %15, %14 ], [ %21, %16 ]
%19 = getelementptr inbounds i32, i32* %0, i64 %17
%20 = load i32, i32* %19
%21 = add nsw i64 %18, 1
%22 = getelementptr inbounds i32, i32* %1, i64 %18
store i32 %20, i32* %22
%23 = add i64 %17, %7
%24 = icmp slt i64 %23, %8
br i1 %24, label %16, label %25
25: ; preds = %16
%26 = trunc i64 %21 to i32
br label %27
27: ; preds = %25, %9
%28 = phi i32 [ %11, %9 ], [ %26, %25 ]
%29 = add nuw i64 %10, 1
%30 = icmp slt i32 %28, %2
br i1 %30, label %9, label %31
31: ; preds = %27, %4
ret void
}