#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 */
float INFINITY;
int /*<<< orphan*/ free(float *);
scalar_t__ malloc(int);
int *get_partition(int *medoids, float *dm, float *dataset, int n_objects,
                   int n_attributes) {
  int *partition = (int *)malloc(sizeof(int) * n_objects);
  for (int i = 0; i < n_objects; i++) {
    float closest_dist = INFINITY;
    int closest_index = -1;
    for (int j = 0; j < n_objects; j++) {
      if (medoids[j] == 1) {
        if (dm[i * n_objects + j] < closest_dist) {
          closest_dist = dm[i * n_objects + j];
          closest_index = j;
        }
      }
      partition[i] = closest_index;
    }
  }
  free(dm);
  return partition;
}


x86 assembly (clang, -O0):
<get_partition>:
push   %rbp
mov    %rsp,%rbp
sub    $0x40,%rsp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %rdx,-0x18(%rbp)
mov    %ecx,-0x1c(%rbp)
mov    %r8d,-0x20(%rbp)
movslq -0x1c(%rbp),%rax
shl    $0x2,%rax
mov    %eax,%edi
call   1040 <malloc@plt>
mov    %rax,-0x28(%rbp)
movl   $0x0,-0x2c(%rbp)
mov    -0x2c(%rbp),%eax
cmp    -0x1c(%rbp),%eax
jge    1216 <get_partition+0xf6>
mov    0x2e80(%rip),%rax
movss  (%rax),%xmm0
movss  %xmm0,-0x30(%rbp)
movl   $0xffffffff,-0x34(%rbp)
movl   $0x0,-0x38(%rbp)
mov    -0x38(%rbp),%eax
cmp    -0x1c(%rbp),%eax
jge    1203 <get_partition+0xe3>
mov    -0x8(%rbp),%rax
movslq -0x38(%rbp),%rcx
cmpl   $0x1,(%rax,%rcx,4)
jne    11e7 <get_partition+0xc7>
mov    -0x10(%rbp),%rax
mov    -0x2c(%rbp),%ecx
imul   -0x1c(%rbp),%ecx
add    -0x38(%rbp),%ecx
movslq %ecx,%rcx
movss  (%rax,%rcx,4),%xmm1
movss  -0x30(%rbp),%xmm0
ucomiss %xmm1,%xmm0
jbe    11e2 <get_partition+0xc2>
mov    -0x10(%rbp),%rax
mov    -0x2c(%rbp),%ecx
imul   -0x1c(%rbp),%ecx
add    -0x38(%rbp),%ecx
movslq %ecx,%rcx
movss  (%rax,%rcx,4),%xmm0
movss  %xmm0,-0x30(%rbp)
mov    -0x38(%rbp),%eax
mov    %eax,-0x34(%rbp)
jmp    11e7 <get_partition+0xc7>
mov    -0x34(%rbp),%edx
mov    -0x28(%rbp),%rax
movslq -0x2c(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x38(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x38(%rbp)
jmp    117f <get_partition+0x5f>
jmp    1208 <get_partition+0xe8>
mov    -0x2c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x2c(%rbp)
jmp    1155 <get_partition+0x35>
mov    -0x10(%rbp),%rdi
call   1030 <free@plt>
mov    -0x28(%rbp),%rax
add    $0x40,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<get_partition>:
push   %r15
push   %r14
push   %r12
push   %rbx
push   %rax
mov    %ecx,%r15d
mov    %rsi,%r14
mov    %rdi,%r12
lea    0x0(,%r15,4),%edi
call   1040 <malloc@plt>
mov    %rax,%rbx
test   %r15d,%r15d
jle    1200 <get_partition+0xe0>
mov    0x2e97(%rip),%rax
movss  (%rax),%xmm0
mov    %r15d,%r9d
mov    %r9d,%r10d
and    $0xfffffffe,%r10d
lea    0x4(%r14),%rdx
lea    0x0(,%r9,4),%r8
xor    %ecx,%ecx
jmp    1183 <get_partition+0x63>
nop
mov    %esi,(%rbx,%rcx,4)
add    $0x1,%rcx
add    %r8,%rdx
cmp    %r9,%rcx
je     1200 <get_partition+0xe0>
mov    $0xffffffff,%esi
xor    %eax,%eax
movaps %xmm0,%xmm1
cmp    $0x1,%r15d
jne    11cc <get_partition+0xac>
test   $0x1,%r9b
je     1173 <get_partition+0x53>
cmpl   $0x1,(%r12,%rax,4)
jne    1170 <get_partition+0x50>
mov    %rcx,%rdi
imul   %r9,%rdi
add    %rax,%rdi
ucomiss (%r14,%rdi,4),%xmm1
jbe    1170 <get_partition+0x50>
mov    %eax,%esi
jmp    1170 <get_partition+0x50>
cs nopw 0x0(%rax,%rax,1)
nop
mov    %esi,(%rbx,%rcx,4)
add    $0x2,%rax
cmp    %rax,%r10
je     1193 <get_partition+0x73>
cmpl   $0x1,(%r12,%rax,4)
jne    11e3 <get_partition+0xc3>
movss  -0x4(%rdx,%rax,4),%xmm2
ucomiss %xmm2,%xmm1
jbe    11e3 <get_partition+0xc3>
movaps %xmm2,%xmm1
mov    %eax,%esi
mov    %esi,(%rbx,%rcx,4)
cmpl   $0x1,0x4(%r12,%rax,4)
jne    11c0 <get_partition+0xa0>
movss  (%rdx,%rax,4),%xmm2
ucomiss %xmm2,%xmm1
jbe    11c0 <get_partition+0xa0>
lea    0x1(%rax),%esi
movaps %xmm2,%xmm1
jmp    11c0 <get_partition+0xa0>
mov    %r14,%rdi
call   1030 <free@plt>
mov    %rbx,%rax
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
ret

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

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