#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 */
void upsample_cpu(float *in, int w, int h, int c, int batch, int stride,
                  int forward, float scale, float *out) {
  int i, j, k, b;
  for (b = 0; b < batch; ++b) {
    for (k = 0; k < c; ++k) {
      for (j = 0; j < h * stride; ++j) {
        for (i = 0; i < w * stride; ++i) {
          int in_index =
              b * w * h * c + k * w * h + (j / stride) * w + i / stride;
          int out_index = b * w * h * c * stride * stride +
                          k * w * h * stride * stride + j * w * stride + i;
          if (forward)
            out[out_index] = scale * in[in_index];
          else
            in[in_index] += scale * out[out_index];
        }
      }
    }
  }
}


x86 assembly (clang, -O0):
<upsample_cpu>:
push   %rbp
mov    %rsp,%rbp
mov    0x18(%rbp),%rax
mov    0x10(%rbp),%eax
mov    %rdi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
mov    %edx,-0x10(%rbp)
mov    %ecx,-0x14(%rbp)
mov    %r8d,-0x18(%rbp)
mov    %r9d,-0x1c(%rbp)
movss  %xmm0,-0x20(%rbp)
movl   $0x0,-0x30(%rbp)
mov    -0x30(%rbp),%eax
cmp    -0x18(%rbp),%eax
jge    12a7 <upsample_cpu+0x1a7>
movl   $0x0,-0x2c(%rbp)
mov    -0x2c(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    1294 <upsample_cpu+0x194>
movl   $0x0,-0x28(%rbp)
mov    -0x28(%rbp),%eax
mov    -0x10(%rbp),%ecx
imul   -0x1c(%rbp),%ecx
cmp    %ecx,%eax
jge    1281 <upsample_cpu+0x181>
movl   $0x0,-0x24(%rbp)
mov    -0x24(%rbp),%eax
mov    -0xc(%rbp),%ecx
imul   -0x1c(%rbp),%ecx
cmp    %ecx,%eax
jge    126e <upsample_cpu+0x16e>
mov    -0x30(%rbp),%eax
imul   -0xc(%rbp),%eax
imul   -0x10(%rbp),%eax
imul   -0x14(%rbp),%eax
mov    -0x2c(%rbp),%ecx
imul   -0xc(%rbp),%ecx
imul   -0x10(%rbp),%ecx
add    %ecx,%eax
mov    %eax,-0x40(%rbp)
mov    -0x28(%rbp),%eax
cltd
idivl  -0x1c(%rbp)
mov    %eax,%ecx
mov    -0x40(%rbp),%eax
imul   -0xc(%rbp),%ecx
add    %ecx,%eax
mov    %eax,-0x3c(%rbp)
mov    -0x24(%rbp),%eax
cltd
idivl  -0x1c(%rbp)
mov    %eax,%ecx
mov    -0x3c(%rbp),%eax
add    %ecx,%eax
mov    %eax,-0x34(%rbp)
mov    -0x30(%rbp),%eax
imul   -0xc(%rbp),%eax
imul   -0x10(%rbp),%eax
imul   -0x14(%rbp),%eax
imul   -0x1c(%rbp),%eax
imul   -0x1c(%rbp),%eax
mov    -0x2c(%rbp),%ecx
imul   -0xc(%rbp),%ecx
imul   -0x10(%rbp),%ecx
imul   -0x1c(%rbp),%ecx
imul   -0x1c(%rbp),%ecx
add    %ecx,%eax
mov    -0x28(%rbp),%ecx
imul   -0xc(%rbp),%ecx
imul   -0x1c(%rbp),%ecx
add    %ecx,%eax
add    -0x24(%rbp),%eax
mov    %eax,-0x38(%rbp)
cmpl   $0x0,0x10(%rbp)
je     122f <upsample_cpu+0x12f>
movss  -0x20(%rbp),%xmm0
mov    -0x8(%rbp),%rax
movslq -0x34(%rbp),%rcx
mulss  (%rax,%rcx,4),%xmm0
mov    0x18(%rbp),%rax
movslq -0x38(%rbp),%rcx
movss  %xmm0,(%rax,%rcx,4)
jmp    125b <upsample_cpu+0x15b>
movss  -0x20(%rbp),%xmm0
mov    0x18(%rbp),%rax
movslq -0x38(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm2
mov    -0x8(%rbp),%rax
movslq -0x34(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm1
mulss  %xmm2,%xmm0
addss  %xmm1,%xmm0
movss  %xmm0,(%rax,%rcx,4)
jmp    1260 <upsample_cpu+0x160>
mov    -0x24(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x24(%rbp)
jmp    116b <upsample_cpu+0x6b>
jmp    1273 <upsample_cpu+0x173>
mov    -0x28(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x28(%rbp)
jmp    1152 <upsample_cpu+0x52>
jmp    1286 <upsample_cpu+0x186>
mov    -0x2c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x2c(%rbp)
jmp    113f <upsample_cpu+0x3f>
jmp    1299 <upsample_cpu+0x199>
mov    -0x30(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x30(%rbp)
jmp    112c <upsample_cpu+0x2c>
pop    %rbp
ret

x86 assembly (clang, -O2):
<upsample_cpu>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
mov    %ecx,-0x2c(%rsp)
mov    %r8d,-0x14(%rsp)
test   %r8d,%r8d
jle    127e <upsample_cpu+0x17e>
mov    0x38(%rsp),%r12d
mov    %r9d,%r15d
imul   %edx,%r15d
mov    %r9d,%ebp
imul   %esi,%ebp
mov    %edx,%eax
imul   %esi,%eax
mov    %eax,-0x10(%rsp)
mov    %r9d,%eax
imul   %r9d,%eax
mov    %eax,%ecx
imul   -0x2c(%rsp),%ecx
imul   %edx,%ecx
imul   %esi,%ecx
mov    %ecx,-0x18(%rsp)
imul   %edx,%eax
imul   %esi,%eax
mov    %eax,-0xc(%rsp)
xor    %eax,%eax
mov    %rax,-0x20(%rsp)
movl   $0x0,-0x28(%rsp)
jmp    1193 <upsample_cpu+0x93>
nopw   0x0(%rax,%rax,1)
mov    -0x28(%rsp),%ecx
add    $0x1,%ecx
mov    -0x20(%rsp),%rax
add    -0x18(%rsp),%eax
mov    %rax,-0x20(%rsp)
mov    %ecx,-0x28(%rsp)
cmp    -0x14(%rsp),%ecx
je     127e <upsample_cpu+0x17e>
cmpl   $0x0,-0x2c(%rsp)
jle    1170 <upsample_cpu+0x70>
mov    -0x28(%rsp),%eax
imul   -0x2c(%rsp),%eax
mov    %rax,-0x8(%rsp)
xor    %r8d,%r8d
mov    -0x20(%rsp),%rax
mov    %eax,-0x24(%rsp)
jmp    11d7 <upsample_cpu+0xd7>
cs nopw 0x0(%rax,%rax,1)
add    $0x1,%r8d
mov    -0x24(%rsp),%eax
add    -0xc(%rsp),%eax
mov    %eax,-0x24(%rsp)
cmp    -0x2c(%rsp),%r8d
je     1170 <upsample_cpu+0x70>
test   %r15d,%r15d
jle    11c0 <upsample_cpu+0xc0>
mov    -0x8(%rsp),%rax
lea    (%r8,%rax,1),%ecx
imul   -0x10(%rsp),%ecx
mov    -0x24(%rsp),%r13d
xor    %r10d,%r10d
jmp    120c <upsample_cpu+0x10c>
cs nopw 0x0(%rax,%rax,1)
xchg   %ax,%ax
add    $0x1,%r10
add    %ebp,%r13d
cmp    %r15,%r10
je     11c0 <upsample_cpu+0xc0>
test   %ebp,%ebp
jle    1200 <upsample_cpu+0x100>
movslq %r13d,%rax
mov    0x40(%rsp),%rdx
lea    (%rdx,%rax,4),%r14
mov    %r10d,%eax
cltd
idiv   %r9d
mov    %eax,%r11d
imul   %esi,%r11d
add    %ecx,%r11d
xor    %ebx,%ebx
jmp    125d <upsample_cpu+0x15d>
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax,%rax,1)
movss  (%r14,%rbx,4),%xmm1
mulss  %xmm0,%xmm1
addss  (%rdi,%rax,4),%xmm1
movss  %xmm1,(%rdi,%rax,4)
add    $0x1,%rbx
cmp    %rbx,%rbp
je     1200 <upsample_cpu+0x100>
mov    %ebx,%eax
cltd
idiv   %r9d
add    %r11d,%eax
cltq
test   %r12d,%r12d
je     1240 <upsample_cpu+0x140>
movss  (%rdi,%rax,4),%xmm1
mulss  %xmm0,%xmm1
movss  %xmm1,(%r14,%rbx,4)
jmp    1254 <upsample_cpu+0x154>
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

LLVM IR (-O0):
define void @upsample_cpu(float* %0, i32 %1, i32 %2, i32 %3, i32 %4, i32 %5, i32 %6, float %7, float* %8) {
%10 = alloca float*
%11 = alloca i32
%12 = alloca i32
%13 = alloca i32
%14 = alloca i32
%15 = alloca i32
%16 = alloca i32
%17 = alloca float
%18 = alloca float*
%19 = alloca i32
%20 = alloca i32
%21 = alloca i32
%22 = alloca i32
%23 = alloca i32
%24 = alloca i32
store float* %0, float** %10
store i32 %1, i32* %11
store i32 %2, i32* %12
store i32 %3, i32* %13
store i32 %4, i32* %14
store i32 %5, i32* %15
store i32 %6, i32* %16
store float %7, float* %17
store float* %8, float** %18
store i32 0, i32* %22
br label %25
25: ; preds = %141, %9
%26 = load i32, i32* %22
%27 = load i32, i32* %14
%28 = icmp slt i32 %26, %27
br i1 %28, label %29, label %144
29: ; preds = %25
store i32 0, i32* %21
br label %30
30: ; preds = %137, %29
%31 = load i32, i32* %21
%32 = load i32, i32* %13
%33 = icmp slt i32 %31, %32
br i1 %33, label %34, label %140
34: ; preds = %30
store i32 0, i32* %20
br label %35
35: ; preds = %133, %34
%36 = load i32, i32* %20
%37 = load i32, i32* %12
%38 = load i32, i32* %15
%39 = mul nsw i32 %37, %38
%40 = icmp slt i32 %36, %39
br i1 %40, label %41, label %136
41: ; preds = %35
store i32 0, i32* %19
br label %42
42: ; preds = %129, %41
%43 = load i32, i32* %19
%44 = load i32, i32* %11
%45 = load i32, i32* %15
%46 = mul nsw i32 %44, %45
%47 = icmp slt i32 %43, %46
br i1 %47, label %48, label %132
48: ; preds = %42
%49 = load i32, i32* %22
%50 = load i32, i32* %11
%51 = mul nsw i32 %49, %50
%52 = load i32, i32* %12
%53 = mul nsw i32 %51, %52
%54 = load i32, i32* %13
%55 = mul nsw i32 %53, %54
%56 = load i32, i32* %21
%57 = load i32, i32* %11
%58 = mul nsw i32 %56, %57
%59 = load i32, i32* %12
%60 = mul nsw i32 %58, %59
%61 = add nsw i32 %55, %60
%62 = load i32, i32* %20
%63 = load i32, i32* %15
%64 = sdiv i32 %62, %63
%65 = load i32, i32* %11
%66 = mul nsw i32 %64, %65
%67 = add nsw i32 %61, %66
%68 = load i32, i32* %19
%69 = load i32, i32* %15
%70 = sdiv i32 %68, %69
%71 = add nsw i32 %67, %70
store i32 %71, i32* %23
%72 = load i32, i32* %22
%73 = load i32, i32* %11
%74 = mul nsw i32 %72, %73
%75 = load i32, i32* %12
%76 = mul nsw i32 %74, %75
%77 = load i32, i32* %13
%78 = mul nsw i32 %76, %77
%79 = load i32, i32* %15
%80 = mul nsw i32 %78, %79
%81 = load i32, i32* %15
%82 = mul nsw i32 %80, %81
%83 = load i32, i32* %21
%84 = load i32, i32* %11
%85 = mul nsw i32 %83, %84
%86 = load i32, i32* %12
%87 = mul nsw i32 %85, %86
%88 = load i32, i32* %15
%89 = mul nsw i32 %87, %88
%90 = load i32, i32* %15
%91 = mul nsw i32 %89, %90
%92 = add nsw i32 %82, %91
%93 = load i32, i32* %20
%94 = load i32, i32* %11
%95 = mul nsw i32 %93, %94
%96 = load i32, i32* %15
%97 = mul nsw i32 %95, %96
%98 = add nsw i32 %92, %97
%99 = load i32, i32* %19
%100 = add nsw i32 %98, %99
store i32 %100, i32* %24
%101 = load i32, i32* %16
%102 = icmp ne i32 %101, 0
br i1 %102, label %103, label %115
103: ; preds = %48
%104 = load float, float* %17
%105 = load float*, float** %10
%106 = load i32, i32* %23
%107 = sext i32 %106 to i64
%108 = getelementptr inbounds float, float* %105, i64 %107
%109 = load float, float* %108
%110 = fmul float %104, %109
%111 = load float*, float** %18
%112 = load i32, i32* %24
%113 = sext i32 %112 to i64
%114 = getelementptr inbounds float, float* %111, i64 %113
store float %110, float* %114
br label %128
115: ; preds = %48
%116 = load float, float* %17
%117 = load float*, float** %18
%118 = load i32, i32* %24
%119 = sext i32 %118 to i64
%120 = getelementptr inbounds float, float* %117, i64 %119
%121 = load float, float* %120
%122 = load float*, float** %10
%123 = load i32, i32* %23
%124 = sext i32 %123 to i64
%125 = getelementptr inbounds float, float* %122, i64 %124
%126 = load float, float* %125
%127 = call float @llvm.fmuladd.f32(float %116, float %121, float %126)
store float %127, float* %125
br label %128
128: ; preds = %115, %103
br label %129
129: ; preds = %128
%130 = load i32, i32* %19
%131 = add nsw i32 %130, 1
store i32 %131, i32* %19
br label %42
132: ; preds = %42
br label %133
133: ; preds = %132
%134 = load i32, i32* %20
%135 = add nsw i32 %134, 1
store i32 %135, i32* %20
br label %35
136: ; preds = %35
br label %137
137: ; preds = %136
%138 = load i32, i32* %21
%139 = add nsw i32 %138, 1
store i32 %139, i32* %21
br label %30
140: ; preds = %30
br label %141
141: ; preds = %140
%142 = load i32, i32* %22
%143 = add nsw i32 %142, 1
store i32 %143, i32* %22
br label %25
144: ; preds = %25
ret void
}

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