#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 */
scalar_t__ log(double);
double k_loss(double *e, double *y, int n, int k) {
  int i, j;
  double s = 0.0;
  for (i = 0; i < n; i++) {
    for (j = 0; j < k; j++) {
      if (y[i] == j) {
        s -= log(e[j * n + i]);
      }
    }
  }
  return s;
}


x86 assembly (clang, -O0):
<k_loss>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
mov    %ecx,-0x18(%rbp)
xorps  %xmm0,%xmm0
movsd  %xmm0,-0x28(%rbp)
movl   $0x0,-0x1c(%rbp)
mov    -0x1c(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    11ca <k_loss+0xba>
movl   $0x0,-0x20(%rbp)
mov    -0x20(%rbp),%eax
cmp    -0x18(%rbp),%eax
jge    11b7 <k_loss+0xa7>
mov    -0x10(%rbp),%rax
movslq -0x1c(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
cvtsi2sdl -0x20(%rbp),%xmm1
ucomisd %xmm1,%xmm0
jne    11a4 <k_loss+0x94>
jp     11a4 <k_loss+0x94>
mov    -0x8(%rbp),%rax
mov    -0x20(%rbp),%ecx
imul   -0x14(%rbp),%ecx
add    -0x1c(%rbp),%ecx
movslq %ecx,%rcx
movsd  (%rax,%rcx,8),%xmm0
call   1030 <log@plt>
cvtsi2sd %rax,%xmm1
movsd  -0x28(%rbp),%xmm0
subsd  %xmm1,%xmm0
movsd  %xmm0,-0x28(%rbp)
jmp    11a9 <k_loss+0x99>
mov    -0x20(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x20(%rbp)
jmp    1148 <k_loss+0x38>
jmp    11bc <k_loss+0xac>
mov    -0x1c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x1c(%rbp)
jmp    1135 <k_loss+0x25>
movsd  -0x28(%rbp),%xmm0
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<k_loss>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
sub    $0x28,%rsp
test   %edx,%edx
jle    11c8 <k_loss+0xb8>
mov    %rsi,%r13
mov    %rdi,%r12
mov    %edx,%eax
mov    %ecx,%ebp
mov    %rax,0x10(%rsp)
lea    0x0(,%rax,8),%rbx
xorpd  %xmm2,%xmm2
xor    %r14d,%r14d
mov    %ecx,0xc(%rsp)
jmp    1168 <k_loss+0x58>
nopw   0x0(%rax,%rax,1)
add    $0x1,%r14
mov    0x18(%rsp),%r12
add    $0x8,%r12
cmp    0x10(%rsp),%r14
mov    0xc(%rsp),%ecx
je     11cc <k_loss+0xbc>
test   %ecx,%ecx
mov    %r12,0x18(%rsp)
jle    1150 <k_loss+0x40>
xor    %r15d,%r15d
jmp    118c <k_loss+0x7c>
cs nopw 0x0(%rax,%rax,1)
add    $0x1,%r15
add    %rbx,%r12
cmp    %r15,%rbp
je     1150 <k_loss+0x40>
movsd  0x0(%r13,%r14,8),%xmm0
xorps  %xmm1,%xmm1
cvtsi2sd %r15d,%xmm1
ucomisd %xmm1,%xmm0
jne    1180 <k_loss+0x70>
jp     1180 <k_loss+0x70>
movsd  (%r12),%xmm0
movsd  %xmm2,0x20(%rsp)
call   1030 <log@plt>
movsd  0x20(%rsp),%xmm2
xorps  %xmm0,%xmm0
cvtsi2sd %rax,%xmm0
subsd  %xmm0,%xmm2
jmp    1180 <k_loss+0x70>
xorpd  %xmm2,%xmm2
movapd %xmm2,%xmm0
add    $0x28,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

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

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