#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 distance(int, float);
scalar_t__ malloc(int);
void knn_cluster(int *const in[], int *const out[], int in_node_count, int k) {
  int i;
  int j;
  int m;
  const double eps = 0.1;
  int *parent = (int *)malloc(sizeof(int) * in_node_count);
  float *classes = (float *)malloc(sizeof(float) * in_node_count);
  for (m = 0; m < k; m++) {
    for (i = 0; i < in_node_count; i++) {
      for (j = 0; j < in_node_count; j++) {
        if (distance(in[i][j], classes[i]) < 0.1) {
          parent[j] = i;
        }
      }
    }
  }
}


x86 assembly (clang, -O0):
<knn_cluster>:
push   %rbp
mov    %rsp,%rbp
sub    $0x40,%rsp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
mov    %ecx,-0x18(%rbp)
movsd  0xec2(%rip),%xmm0
movsd  %xmm0,-0x30(%rbp)
movslq -0x14(%rbp),%rax
shl    $0x2,%rax
mov    %eax,%edi
call   1030 <malloc@plt>
mov    %rax,-0x38(%rbp)
movslq -0x14(%rbp),%rax
shl    $0x2,%rax
mov    %eax,%edi
call   1030 <malloc@plt>
mov    %rax,-0x40(%rbp)
movl   $0x0,-0x24(%rbp)
mov    -0x24(%rbp),%eax
cmp    -0x18(%rbp),%eax
jge    1223 <knn_cluster+0x103>
movl   $0x0,-0x1c(%rbp)
mov    -0x1c(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    1210 <knn_cluster+0xf0>
movl   $0x0,-0x20(%rbp)
mov    -0x20(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    11fd <knn_cluster+0xdd>
mov    -0x8(%rbp),%rax
movslq -0x1c(%rbp),%rcx
mov    (%rax,%rcx,8),%rax
movslq -0x20(%rbp),%rcx
mov    (%rax,%rcx,4),%edi
mov    -0x40(%rbp),%rax
movslq -0x1c(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm0
call   1040 <distance@plt>
movaps %xmm0,%xmm1
movsd  0xe2e(%rip),%xmm0
ucomisd %xmm1,%xmm0
jbe    11ea <knn_cluster+0xca>
mov    -0x1c(%rbp),%edx
mov    -0x38(%rbp),%rax
movslq -0x20(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
jmp    11ef <knn_cluster+0xcf>
mov    -0x20(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x20(%rbp)
jmp    1196 <knn_cluster+0x76>
jmp    1202 <knn_cluster+0xe2>
mov    -0x1c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x1c(%rbp)
jmp    1183 <knn_cluster+0x63>
jmp    1215 <knn_cluster+0xf5>
mov    -0x24(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x24(%rbp)
jmp    1170 <knn_cluster+0x50>
add    $0x40,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<knn_cluster>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
sub    $0x18,%rsp
mov    %ecx,%ebp
mov    %rdi,%r13
mov    %rdx,0x8(%rsp)
lea    0x0(,%rdx,4),%ebx
mov    %ebx,%edi
call   1030 <malloc@plt>
mov    %rax,%r12
mov    %ebx,%edi
call   1030 <malloc@plt>
mov    %ebp,0x4(%rsp)
test   %ebp,%ebp
jle    11d1 <knn_cluster+0xb1>
mov    %rax,%rbx
mov    0x8(%rsp),%ebp
xor    %eax,%eax
jmp    117e <knn_cluster+0x5e>
cs nopw 0x0(%rax,%rax,1)
nopl   (%rax)
mov    0x10(%rsp),%rax
add    $0x1,%eax
cmp    0x4(%rsp),%eax
je     11d1 <knn_cluster+0xb1>
mov    %rax,0x10(%rsp)
cmpl   $0x0,0x8(%rsp)
jle    1170 <knn_cluster+0x50>
xor    %r15d,%r15d
jmp    1199 <knn_cluster+0x79>
nop
add    $0x1,%r15
cmp    %rbp,%r15
je     1170 <knn_cluster+0x50>
xor    %r14d,%r14d
jmp    11a9 <knn_cluster+0x89>
xchg   %ax,%ax
add    $0x1,%r14
cmp    %r14,%rbp
je     1190 <knn_cluster+0x70>
mov    0x0(%r13,%r15,8),%rax
mov    (%rax,%r14,4),%edi
movss  (%rbx,%r15,4),%xmm0
call   1040 <distance@plt>
movsd  0xe3b(%rip),%xmm1
ucomisd %xmm0,%xmm1
jbe    11a0 <knn_cluster+0x80>
mov    %r15d,(%r12,%r14,4)
jmp    11a0 <knn_cluster+0x80>
add    $0x18,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define void @knn_cluster(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
%12 = alloca double
%13 = alloca i32*
%14 = alloca float*
store i32** %0, i32*** %5
store i32** %1, i32*** %6
store i32 %2, i32* %7
store i32 %3, i32* %8
store double 1.000000e-01, double* %12
%15 = load i32, i32* %7
%16 = sext i32 %15 to i64
%17 = mul i64 4, %16
%18 = trunc i64 %17 to i32
%19 = call i64 @malloc(i32 %18)
%20 = inttoptr i64 %19 to i32*
store i32* %20, i32** %13
%21 = load i32, i32* %7
%22 = sext i32 %21 to i64
%23 = mul i64 4, %22
%24 = trunc i64 %23 to i32
%25 = call i64 @malloc(i32 %24)
%26 = inttoptr i64 %25 to float*
store float* %26, float** %14
store i32 0, i32* %11
br label %27
27: ; preds = %73, %4
%28 = load i32, i32* %11
%29 = load i32, i32* %8
%30 = icmp slt i32 %28, %29
br i1 %30, label %31, label %76
31: ; preds = %27
store i32 0, i32* %9
br label %32
32: ; preds = %69, %31
%33 = load i32, i32* %9
%34 = load i32, i32* %7
%35 = icmp slt i32 %33, %34
br i1 %35, label %36, label %72
36: ; preds = %32
store i32 0, i32* %10
br label %37
37: ; preds = %65, %36
%38 = load i32, i32* %10
%39 = load i32, i32* %7
%40 = icmp slt i32 %38, %39
br i1 %40, label %41, label %68
41: ; preds = %37
%42 = load i32**, i32*** %5
%43 = load i32, i32* %9
%44 = sext i32 %43 to i64
%45 = getelementptr inbounds i32*, i32** %42, i64 %44
%46 = load i32*, i32** %45
%47 = load i32, i32* %10
%48 = sext i32 %47 to i64
%49 = getelementptr inbounds i32, i32* %46, i64 %48
%50 = load i32, i32* %49
%51 = load float*, float** %14
%52 = load i32, i32* %9
%53 = sext i32 %52 to i64
%54 = getelementptr inbounds float, float* %51, i64 %53
%55 = load float, float* %54
%56 = call double @distance(i32 %50, float %55)
%57 = fcmp olt double %56, 1.000000e-01
br i1 %57, label %58, label %64
58: ; preds = %41
%59 = load i32, i32* %9
%60 = load i32*, i32** %13
%61 = load i32, i32* %10
%62 = sext i32 %61 to i64
%63 = getelementptr inbounds i32, i32* %60, i64 %62
store i32 %59, i32* %63
br label %64
64: ; preds = %58, %41
br label %65
65: ; preds = %64
%66 = load i32, i32* %10
%67 = add nsw i32 %66, 1
store i32 %67, i32* %10
br label %37
68: ; preds = %37
br label %69
69: ; preds = %68
%70 = load i32, i32* %9
%71 = add nsw i32 %70, 1
store i32 %71, i32* %9
br label %32
72: ; preds = %32
br label %73
73: ; preds = %72
%74 = load i32, i32* %11
%75 = add nsw i32 %74, 1
store i32 %75, i32* %11
br label %27
76: ; preds = %27
ret void
}

LLVM IR (-O2):
define void @knn_cluster(i32** nocapture %0, i32** nocapture readnone %1, i32 %2, i32 %3) {
%5 = shl i32 %2, 2
%6 = tail call i64 @malloc(i32 %5)
%7 = inttoptr i64 %6 to i32*
%8 = tail call i64 @malloc(i32 %5)
%9 = inttoptr i64 %8 to float*
%10 = icmp sgt i32 %3, 0
br i1 %10, label %11, label %40
11: ; preds = %4
%12 = icmp sgt i32 %2, 0
%13 = zext i32 %2 to i64
br label %14
14: ; preds = %11, %37
%15 = phi i32 [ 0, %11 ], [ %38, %37 ]
br i1 %12, label %16, label %37
16: ; preds = %14, %34
%17 = phi i64 [ %35, %34 ], [ 0, %14 ]
%18 = getelementptr inbounds i32*, i32** %0, i64 %17
%19 = getelementptr inbounds float, float* %9, i64 %17
%20 = trunc i64 %17 to i32
br label %21
21: ; preds = %16, %31
%22 = phi i64 [ 0, %16 ], [ %32, %31 ]
%23 = load i32*, i32** %18
%24 = getelementptr inbounds i32, i32* %23, i64 %22
%25 = load i32, i32* %24
%26 = load float, float* %19
%27 = tail call double @distance(i32 %25, float %26)
%28 = fcmp olt double %27, 1.000000e-01
br i1 %28, label %29, label %31
29: ; preds = %21
%30 = getelementptr inbounds i32, i32* %7, i64 %22
store i32 %20, i32* %30
br label %31
31: ; preds = %21, %29
%32 = add nuw nsw i64 %22, 1
%33 = icmp eq i64 %32, %13
br i1 %33, label %34, label %21
34: ; preds = %31
%35 = add nuw nsw i64 %17, 1
%36 = icmp eq i64 %35, %13
br i1 %36, label %37, label %16
37: ; preds = %34, %14
%38 = add nuw nsw i32 %15, 1
%39 = icmp eq i32 %38, %3
br i1 %39, label %40, label %14
40: ; preds = %37, %4
ret void
}