void __ComputeHKWayLoadImbalance(int ncon, int nparts, float *npwgts,
                                 float *lbvec) {
  int i, j;
  float max;
  for (i = 0; i < ncon; i++) {
    max = 0.0;
    for (j = 0; j < nparts; j++) {
      if (npwgts[j * ncon + i] > max)
        max = npwgts[j * ncon + i];
    }
    lbvec[i] = max * nparts;
  }
}


x86 assembly (clang, -O0):
<__ComputeHKWayLoadImbalance>:
push   %rbp
mov    %rsp,%rbp
mov    %edi,-0x4(%rbp)
mov    %esi,-0x8(%rbp)
mov    %rdx,-0x10(%rbp)
mov    %rcx,-0x18(%rbp)
movl   $0x0,-0x1c(%rbp)
mov    -0x1c(%rbp),%eax
cmp    -0x4(%rbp),%eax
jge    11b7 <__ComputeHKWayLoadImbalance+0xb7>
xorps  %xmm0,%xmm0
movss  %xmm0,-0x24(%rbp)
movl   $0x0,-0x20(%rbp)
mov    -0x20(%rbp),%eax
cmp    -0x8(%rbp),%eax
jge    118e <__ComputeHKWayLoadImbalance+0x8e>
mov    -0x10(%rbp),%rax
mov    -0x20(%rbp),%ecx
imul   -0x4(%rbp),%ecx
add    -0x1c(%rbp),%ecx
movslq %ecx,%rcx
movss  (%rax,%rcx,4),%xmm0
ucomiss -0x24(%rbp),%xmm0
jbe    117b <__ComputeHKWayLoadImbalance+0x7b>
mov    -0x10(%rbp),%rax
mov    -0x20(%rbp),%ecx
imul   -0x4(%rbp),%ecx
add    -0x1c(%rbp),%ecx
movslq %ecx,%rcx
movss  (%rax,%rcx,4),%xmm0
movss  %xmm0,-0x24(%rbp)
jmp    1180 <__ComputeHKWayLoadImbalance+0x80>
mov    -0x20(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x20(%rbp)
jmp    1134 <__ComputeHKWayLoadImbalance+0x34>
movss  -0x24(%rbp),%xmm0
cvtsi2ssl -0x8(%rbp),%xmm1
mulss  %xmm1,%xmm0
mov    -0x18(%rbp),%rax
movslq -0x1c(%rbp),%rcx
movss  %xmm0,(%rax,%rcx,4)
mov    -0x1c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x1c(%rbp)
jmp    1119 <__ComputeHKWayLoadImbalance+0x19>
pop    %rbp
ret

x86 assembly (clang, -O2):
<__ComputeHKWayLoadImbalance>:
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
test   %edi,%edi
jle    11f9 <__ComputeHKWayLoadImbalance+0xf9>
cvtsi2ss %esi,%xmm0
mov    %edi,%eax
mov    %esi,%r10d
lea    -0x1(%r10),%r8
mov    %r10d,%r9d
and    $0x3,%r9d
and    $0xfffffffc,%r10d
mov    %rax,%r11
shl    $0x4,%r11
lea    0x0(,%rax,4),%r13
lea    0x0(,%r13,2),%r12
add    %r13,%r12
xor    %r14d,%r14d
mov    %rdx,%r15
jmp    116e <__ComputeHKWayLoadImbalance+0x6e>
nopl   0x0(%rax,%rax,1)
xorps  %xmm2,%xmm2
mulss  %xmm0,%xmm2
movss  %xmm2,(%rcx,%r14,4)
add    $0x1,%r14
add    $0x4,%r15
cmp    %rax,%r14
je     11f9 <__ComputeHKWayLoadImbalance+0xf9>
test   %esi,%esi
jle    1150 <__ComputeHKWayLoadImbalance+0x50>
xorps  %xmm1,%xmm1
cmp    $0x3,%r8
jae    1180 <__ComputeHKWayLoadImbalance+0x80>
xor    %ebx,%ebx
jmp    11c0 <__ComputeHKWayLoadImbalance+0xc0>
nop
mov    %r15,%rdi
xor    %ebx,%ebx
cs nopw 0x0(%rax,%rax,1)
nop
movss  (%rdi),%xmm2
maxss  %xmm1,%xmm2
movss  (%rdi,%rax,4),%xmm1
maxss  %xmm2,%xmm1
movss  (%rdi,%rax,8),%xmm2
maxss  %xmm1,%xmm2
movss  (%rdi,%r12,1),%xmm1
maxss  %xmm2,%xmm1
add    $0x4,%rbx
add    %r11,%rdi
cmp    %rbx,%r10
jne    1190 <__ComputeHKWayLoadImbalance+0x90>
movaps %xmm1,%xmm2
test   %r9,%r9
je     1153 <__ComputeHKWayLoadImbalance+0x53>
imul   %rax,%rbx
add    %r14,%rbx
lea    (%rdx,%rbx,4),%rdi
mov    %r9,%rbx
cs nopw 0x0(%rax,%rax,1)
movss  (%rdi),%xmm2
maxss  %xmm1,%xmm2
add    %r13,%rdi
movaps %xmm2,%xmm1
add    $0xffffffffffffffff,%rbx
jne    11e0 <__ComputeHKWayLoadImbalance+0xe0>
jmp    1153 <__ComputeHKWayLoadImbalance+0x53>
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
ret

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

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