#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 */
double INFINITY;
void choose_all_clusters_from_distances(int dim, int n, int k,
                                        double *distance_array,
                                        int *cluster_assignment_index) {
  for (int ii = 0; ii < n; ii++) {
    int best_index = -1;
    double closest_distance = (INFINITY);
    for (int jj = 0; jj < k; jj++) {
      double cur_distance = distance_array[ii * k + jj];
      if (cur_distance < closest_distance) {
        best_index = jj;
        closest_distance = cur_distance;
      }
    }
    cluster_assignment_index[ii] = best_index;
  }
}


x86 assembly (clang, -O0):
<choose_all_clusters_from_distances>:
push   %rbp
mov    %rsp,%rbp
mov    %edi,-0x4(%rbp)
mov    %esi,-0x8(%rbp)
mov    %edx,-0xc(%rbp)
mov    %rcx,-0x18(%rbp)
mov    %r8,-0x20(%rbp)
movl   $0x0,-0x24(%rbp)
mov    -0x24(%rbp),%eax
cmp    -0x8(%rbp),%eax
jge    11c0 <choose_all_clusters_from_distances+0xc0>
movl   $0xffffffff,-0x28(%rbp)
mov    0x2eb2(%rip),%rax
movsd  (%rax),%xmm0
movsd  %xmm0,-0x30(%rbp)
movl   $0x0,-0x34(%rbp)
mov    -0x34(%rbp),%eax
cmp    -0xc(%rbp),%eax
jge    11a4 <choose_all_clusters_from_distances+0xa4>
mov    -0x18(%rbp),%rax
mov    -0x24(%rbp),%ecx
imul   -0xc(%rbp),%ecx
add    -0x34(%rbp),%ecx
movslq %ecx,%rcx
movsd  (%rax,%rcx,8),%xmm0
movsd  %xmm0,-0x40(%rbp)
movsd  -0x40(%rbp),%xmm1
movsd  -0x30(%rbp),%xmm0
ucomisd %xmm1,%xmm0
jbe    1191 <choose_all_clusters_from_distances+0x91>
mov    -0x34(%rbp),%eax
mov    %eax,-0x28(%rbp)
movsd  -0x40(%rbp),%xmm0
movsd  %xmm0,-0x30(%rbp)
jmp    1196 <choose_all_clusters_from_distances+0x96>
mov    -0x34(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x34(%rbp)
jmp    1146 <choose_all_clusters_from_distances+0x46>
mov    -0x28(%rbp),%edx
mov    -0x20(%rbp),%rax
movslq -0x24(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x24(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x24(%rbp)
jmp    111c <choose_all_clusters_from_distances+0x1c>
pop    %rbp
ret

x86 assembly (clang, -O2):
<choose_all_clusters_from_distances>:
push   %rbp
push   %r15
push   %r14
push   %r12
push   %rbx
test   %esi,%esi
jle    1213 <choose_all_clusters_from_distances+0x113>
mov    0x2ed1(%rip),%rax
movsd  (%rax),%xmm0
movslq %edx,%r14
mov    %esi,%r10d
mov    %r14d,%r12d
lea    -0x1(%r12),%r9
mov    %r12d,%r11d
and    $0x3,%r11d
and    $0xfffffffc,%r12d
lea    0x18(%rcx),%rsi
shl    $0x3,%r14
xor    %r15d,%r15d
jmp    1167 <choose_all_clusters_from_distances+0x67>
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax,%rax,1)
mov    %ebp,(%r8,%r15,4)
add    $0x1,%r15
add    %r14,%rsi
add    %r14,%rcx
cmp    %r10,%r15
je     1213 <choose_all_clusters_from_distances+0x113>
mov    $0xffffffff,%ebp
test   %edx,%edx
jle    1150 <choose_all_clusters_from_distances+0x50>
xor    %eax,%eax
movapd %xmm0,%xmm1
cmp    $0x3,%r9
jb     11d5 <choose_all_clusters_from_distances+0xd5>
nopl   0x0(%rax)
movsd  -0x18(%rsi,%rax,8),%xmm2
ucomisd %xmm2,%xmm1
cmova  %eax,%ebp
movsd  -0x10(%rsi,%rax,8),%xmm3
minsd  %xmm1,%xmm2
ucomisd %xmm3,%xmm2
lea    0x1(%rax),%ebx
cmovbe %ebp,%ebx
minsd  %xmm2,%xmm3
movsd  -0x8(%rsi,%rax,8),%xmm2
ucomisd %xmm2,%xmm3
lea    0x2(%rax),%edi
cmovbe %ebx,%edi
minsd  %xmm3,%xmm2
movsd  (%rsi,%rax,8),%xmm1
ucomisd %xmm1,%xmm2
lea    0x3(%rax),%ebp
cmovbe %edi,%ebp
minsd  %xmm2,%xmm1
add    $0x4,%rax
cmp    %r12,%rax
jne    1180 <choose_all_clusters_from_distances+0x80>
test   %r11,%r11
je     1150 <choose_all_clusters_from_distances+0x50>
mov    %r11,%rbx
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax,%rax,1)
movsd  (%rcx,%rax,8),%xmm2
ucomisd %xmm2,%xmm1
cmova  %eax,%ebp
minsd  %xmm1,%xmm2
add    $0x1,%rax
movapd %xmm2,%xmm1
add    $0xffffffffffffffff,%rbx
jne    11f0 <choose_all_clusters_from_distances+0xf0>
jmp    1150 <choose_all_clusters_from_distances+0x50>
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
@INFINITY = global double 0.000000e+00
define void @choose_all_clusters_from_distances(i32 %0, i32 %1, i32 %2, double* %3, i32* %4) {
%6 = alloca i32
%7 = alloca i32
%8 = alloca i32
%9 = alloca double*
%10 = alloca i32*
%11 = alloca i32
%12 = alloca i32
%13 = alloca double
%14 = alloca i32
%15 = alloca double
store i32 %0, i32* %6
store i32 %1, i32* %7
store i32 %2, i32* %8
store double* %3, double** %9
store i32* %4, i32** %10
store i32 0, i32* %11
br label %16
16: ; preds = %52, %5
%17 = load i32, i32* %11
%18 = load i32, i32* %7
%19 = icmp slt i32 %17, %18
br i1 %19, label %20, label %55
20: ; preds = %16
store i32 -1, i32* %12
%21 = load double, double* @INFINITY
store double %21, double* %13
store i32 0, i32* %14
br label %22
22: ; preds = %43, %20
%23 = load i32, i32* %14
%24 = load i32, i32* %8
%25 = icmp slt i32 %23, %24
br i1 %25, label %26, label %46
26: ; preds = %22
%27 = load double*, double** %9
%28 = load i32, i32* %11
%29 = load i32, i32* %8
%30 = mul nsw i32 %28, %29
%31 = load i32, i32* %14
%32 = add nsw i32 %30, %31
%33 = sext i32 %32 to i64
%34 = getelementptr inbounds double, double* %27, i64 %33
%35 = load double, double* %34
store double %35, double* %15
%36 = load double, double* %15
%37 = load double, double* %13
%38 = fcmp olt double %36, %37
br i1 %38, label %39, label %42
39: ; preds = %26
%40 = load i32, i32* %14
store i32 %40, i32* %12
%41 = load double, double* %15
store double %41, double* %13
br label %42
42: ; preds = %39, %26
br label %43
43: ; preds = %42
%44 = load i32, i32* %14
%45 = add nsw i32 %44, 1
store i32 %45, i32* %14
br label %22
46: ; preds = %22
%47 = load i32, i32* %12
%48 = load i32*, i32** %10
%49 = load i32, i32* %11
%50 = sext i32 %49 to i64
%51 = getelementptr inbounds i32, i32* %48, i64 %50
store i32 %47, i32* %51
br label %52
52: ; preds = %46
%53 = load i32, i32* %11
%54 = add nsw i32 %53, 1
store i32 %54, i32* %11
br label %16
55: ; preds = %16
ret void
}

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