#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 neuralops_batchnorm2d_fwd_mean(size_t batch_sz, size_t width,
                                    size_t height, size_t chan,
                                    const float *in_buf, float *mean) {
  for (size_t a = 0; a < chan; a += 1) {
    mean[a] = 0.0f;
  }
  size_t p = 0;
  for (size_t idx = 0; idx < batch_sz; idx += 1) {
    for (size_t a = 0; a < chan; a += 1) {
      for (size_t y = 0; y < height; y += 1) {
        for (size_t x = 0; x < width; x += 1) {
          mean[a] += in_buf[p];
          p += 1;
        }
      }
    }
  }
  for (size_t a = 0; a < chan; a += 1) {
    mean[a] /= (float)(width * height * batch_sz);
  }
}


x86 assembly (clang, -O0):
<neuralops_batchnorm2d_fwd_mean>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %rdx,-0x18(%rbp)
mov    %rcx,-0x20(%rbp)
mov    %r8,-0x28(%rbp)
mov    %r9,-0x30(%rbp)
movq   $0x0,-0x38(%rbp)
mov    -0x38(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    1153 <neuralops_batchnorm2d_fwd_mean+0x53>
mov    -0x30(%rbp),%rax
mov    -0x38(%rbp),%rcx
xorps  %xmm0,%xmm0
movss  %xmm0,(%rax,%rcx,4)
mov    -0x38(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x38(%rbp)
jmp    1124 <neuralops_batchnorm2d_fwd_mean+0x24>
movq   $0x0,-0x40(%rbp)
movq   $0x0,-0x48(%rbp)
mov    -0x48(%rbp),%rax
cmp    -0x8(%rbp),%rax
jae    1231 <neuralops_batchnorm2d_fwd_mean+0x131>
movq   $0x0,-0x50(%rbp)
mov    -0x50(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    121b <neuralops_batchnorm2d_fwd_mean+0x11b>
movq   $0x0,-0x58(%rbp)
mov    -0x58(%rbp),%rax
cmp    -0x18(%rbp),%rax
jae    1205 <neuralops_batchnorm2d_fwd_mean+0x105>
movq   $0x0,-0x60(%rbp)
mov    -0x60(%rbp),%rax
cmp    -0x10(%rbp),%rax
jae    11ef <neuralops_batchnorm2d_fwd_mean+0xef>
mov    -0x28(%rbp),%rax
mov    -0x40(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm0
mov    -0x30(%rbp),%rax
mov    -0x50(%rbp),%rcx
addss  (%rax,%rcx,4),%xmm0
movss  %xmm0,(%rax,%rcx,4)
mov    -0x40(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x40(%rbp)
mov    -0x60(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x60(%rbp)
jmp    11a5 <neuralops_batchnorm2d_fwd_mean+0xa5>
jmp    11f4 <neuralops_batchnorm2d_fwd_mean+0xf4>
mov    -0x58(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x58(%rbp)
jmp    118f <neuralops_batchnorm2d_fwd_mean+0x8f>
jmp    120a <neuralops_batchnorm2d_fwd_mean+0x10a>
mov    -0x50(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x50(%rbp)
jmp    1179 <neuralops_batchnorm2d_fwd_mean+0x79>
jmp    1220 <neuralops_batchnorm2d_fwd_mean+0x120>
mov    -0x48(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x48(%rbp)
jmp    1163 <neuralops_batchnorm2d_fwd_mean+0x63>
movq   $0x0,-0x68(%rbp)
mov    -0x68(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    12c0 <neuralops_batchnorm2d_fwd_mean+0x1c0>
mov    -0x10(%rbp),%rax
mov    -0x18(%rbp),%rcx
imul   %rcx,%rax
mov    -0x8(%rbp),%rcx
imul   %rcx,%rax
mov    %rax,%rdx
shr    %rdx
mov    %eax,%ecx
and    $0x1,%ecx
or     %rdx,%rcx
cvtsi2ss %rcx,%xmm0
addss  %xmm0,%xmm0
cvtsi2ss %rax,%xmm1
movss  %xmm1,-0x70(%rbp)
test   %rax,%rax
movss  %xmm0,-0x6c(%rbp)
js     1294 <neuralops_batchnorm2d_fwd_mean+0x194>
movss  -0x70(%rbp),%xmm0
movss  %xmm0,-0x6c(%rbp)
movss  -0x6c(%rbp),%xmm1
mov    -0x30(%rbp),%rax
mov    -0x68(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm0
divss  %xmm1,%xmm0
movss  %xmm0,(%rax,%rcx,4)
mov    -0x68(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x68(%rbp)
jmp    1239 <neuralops_batchnorm2d_fwd_mean+0x139>
pop    %rbp
ret

x86 assembly (clang, -O2):
<neuralops_batchnorm2d_fwd_mean>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
push   %rax
mov    %r9,%rbx
mov    %r8,%rbp
mov    %rcx,%r14
mov    %rdx,%r12
mov    %rsi,%r15
mov    %rdi,%r13
test   %rcx,%rcx
je     1144 <neuralops_batchnorm2d_fwd_mean+0x34>
lea    0x0(,%r14,4),%rdx
mov    %rbx,%rdi
xor    %esi,%esi
call   1030 <memset@plt>
test   %r13,%r13
je     1227 <neuralops_batchnorm2d_fwd_mean+0x117>
lea    -0x1(%r15),%r9
mov    %r15d,%ecx
and    $0x3,%ecx
mov    %r15,%r10
and    $0xfffffffffffffffc,%r10
xor    %r8d,%r8d
xor    %edi,%edi
jmp    117d <neuralops_batchnorm2d_fwd_mean+0x6d>
cs nopw 0x0(%rax,%rax,1)
nop
add    $0x1,%r8
cmp    %r13,%r8
je     1227 <neuralops_batchnorm2d_fwd_mean+0x117>
test   %r14,%r14
je     1170 <neuralops_batchnorm2d_fwd_mean+0x60>
xor    %esi,%esi
jmp    1199 <neuralops_batchnorm2d_fwd_mean+0x89>
cs nopw 0x0(%rax,%rax,1)
add    $0x1,%rsi
cmp    %r14,%rsi
je     1170 <neuralops_batchnorm2d_fwd_mean+0x60>
test   %r12,%r12
je     1190 <neuralops_batchnorm2d_fwd_mean+0x80>
xor    %eax,%eax
jmp    11b9 <neuralops_batchnorm2d_fwd_mean+0xa9>
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax)
add    $0x1,%rax
cmp    %r12,%rax
je     1190 <neuralops_batchnorm2d_fwd_mean+0x80>
test   %r15,%r15
je     11b0 <neuralops_batchnorm2d_fwd_mean+0xa0>
movss  (%rbx,%rsi,4),%xmm0
mov    %r10,%rdx
cmp    $0x3,%r9
jb     1206 <neuralops_batchnorm2d_fwd_mean+0xf6>
nopl   0x0(%rax)
addss  0x0(%rbp,%rdi,4),%xmm0
movss  %xmm0,(%rbx,%rsi,4)
addss  0x4(%rbp,%rdi,4),%xmm0
movss  %xmm0,(%rbx,%rsi,4)
addss  0x8(%rbp,%rdi,4),%xmm0
movss  %xmm0,(%rbx,%rsi,4)
addss  0xc(%rbp,%rdi,4),%xmm0
movss  %xmm0,(%rbx,%rsi,4)
add    $0x4,%rdi
add    $0xfffffffffffffffc,%rdx
jne    11d0 <neuralops_batchnorm2d_fwd_mean+0xc0>
test   %rcx,%rcx
je     11b0 <neuralops_batchnorm2d_fwd_mean+0xa0>
mov    %rcx,%rdx
xchg   %ax,%ax
addss  0x0(%rbp,%rdi,4),%xmm0
movss  %xmm0,(%rbx,%rsi,4)
add    $0x1,%rdi
add    $0xffffffffffffffff,%rdx
jne    1210 <neuralops_batchnorm2d_fwd_mean+0x100>
jmp    11b0 <neuralops_batchnorm2d_fwd_mean+0xa0>
test   %r14,%r14
je     1307 <neuralops_batchnorm2d_fwd_mean+0x1f7>
imul   %r13,%r15
imul   %r12,%r15
test   %r15,%r15
js     12c9 <neuralops_batchnorm2d_fwd_mean+0x1b9>
xorps  %xmm0,%xmm0
cvtsi2ss %r15,%xmm0
cmp    $0x4,%r14
jb     12ec <neuralops_batchnorm2d_fwd_mean+0x1dc>
mov    %r14,%rax
and    $0xfffffffffffffffc,%rax
movaps %xmm0,%xmm1
shufps $0x0,%xmm0,%xmm1
lea    -0x4(%rax),%rdx
mov    %rdx,%rcx
shr    $0x2,%rcx
add    $0x1,%rcx
test   %rdx,%rdx
je     1316 <neuralops_batchnorm2d_fwd_mean+0x206>
mov    %rcx,%rsi
and    $0xfffffffffffffffe,%rsi
xor    %edx,%edx
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax)
movups (%rbx,%rdx,4),%xmm2
movups 0x10(%rbx,%rdx,4),%xmm3
divps  %xmm1,%xmm2
movups %xmm2,(%rbx,%rdx,4)
divps  %xmm1,%xmm3
movups %xmm3,0x10(%rbx,%rdx,4)
add    $0x8,%rdx
add    $0xfffffffffffffffe,%rsi
jne    1290 <neuralops_batchnorm2d_fwd_mean+0x180>
test   $0x1,%cl
je     12c2 <neuralops_batchnorm2d_fwd_mean+0x1b2>
movups (%rbx,%rdx,4),%xmm2
divps  %xmm1,%xmm2
movups %xmm2,(%rbx,%rdx,4)
cmp    %r14,%rax
jne    12f0 <neuralops_batchnorm2d_fwd_mean+0x1e0>
jmp    1307 <neuralops_batchnorm2d_fwd_mean+0x1f7>
mov    %r15,%rax
shr    %rax
and    $0x1,%r15d
or     %rax,%r15
xorps  %xmm0,%xmm0
cvtsi2ss %r15,%xmm0
addss  %xmm0,%xmm0
cmp    $0x4,%r14
jae    1253 <neuralops_batchnorm2d_fwd_mean+0x143>
xor    %eax,%eax
xchg   %ax,%ax
movss  (%rbx,%rax,4),%xmm1
divss  %xmm0,%xmm1
movss  %xmm1,(%rbx,%rax,4)
add    $0x1,%rax
cmp    %rax,%r14
jne    12f0 <neuralops_batchnorm2d_fwd_mean+0x1e0>
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret
xor    %edx,%edx
test   $0x1,%cl
jne    12b7 <neuralops_batchnorm2d_fwd_mean+0x1a7>
jmp    12c2 <neuralops_batchnorm2d_fwd_mean+0x1b2>

LLVM IR (-O0):
define void @neuralops_batchnorm2d_fwd_mean(i64 %0, i64 %1, i64 %2, i64 %3, float* %4, float* %5) {
%7 = alloca i64
%8 = alloca i64
%9 = alloca i64
%10 = alloca i64
%11 = alloca float*
%12 = alloca float*
%13 = alloca i64
%14 = alloca i64
%15 = alloca i64
%16 = alloca i64
%17 = alloca i64
%18 = alloca i64
%19 = alloca i64
store i64 %0, i64* %7
store i64 %1, i64* %8
store i64 %2, i64* %9
store i64 %3, i64* %10
store float* %4, float** %11
store float* %5, float** %12
store i64 0, i64* %13
br label %20
20: ; preds = %28, %6
%21 = load i64, i64* %13
%22 = load i64, i64* %10
%23 = icmp ult i64 %21, %22
br i1 %23, label %24, label %31
24: ; preds = %20
%25 = load float*, float** %12
%26 = load i64, i64* %13
%27 = getelementptr inbounds float, float* %25, i64 %26
store float 0.000000e+00, float* %27
br label %28
28: ; preds = %24
%29 = load i64, i64* %13
%30 = add i64 %29, 1
store i64 %30, i64* %13
br label %20
31: ; preds = %20
store i64 0, i64* %14
store i64 0, i64* %15
br label %32
32: ; preds = %75, %31
%33 = load i64, i64* %15
%34 = load i64, i64* %7
%35 = icmp ult i64 %33, %34
br i1 %35, label %36, label %78
36: ; preds = %32
store i64 0, i64* %16
br label %37
37: ; preds = %71, %36
%38 = load i64, i64* %16
%39 = load i64, i64* %10
%40 = icmp ult i64 %38, %39
br i1 %40, label %41, label %74
41: ; preds = %37
store i64 0, i64* %17
br label %42
42: ; preds = %67, %41
%43 = load i64, i64* %17
%44 = load i64, i64* %9
%45 = icmp ult i64 %43, %44
br i1 %45, label %46, label %70
46: ; preds = %42
store i64 0, i64* %18
br label %47
47: ; preds = %63, %46
%48 = load i64, i64* %18
%49 = load i64, i64* %8
%50 = icmp ult i64 %48, %49
br i1 %50, label %51, label %66
51: ; preds = %47
%52 = load float*, float** %11
%53 = load i64, i64* %14
%54 = getelementptr inbounds float, float* %52, i64 %53
%55 = load float, float* %54
%56 = load float*, float** %12
%57 = load i64, i64* %16
%58 = getelementptr inbounds float, float* %56, i64 %57
%59 = load float, float* %58
%60 = fadd float %59, %55
store float %60, float* %58
%61 = load i64, i64* %14
%62 = add i64 %61, 1
store i64 %62, i64* %14
br label %63
63: ; preds = %51
%64 = load i64, i64* %18
%65 = add i64 %64, 1
store i64 %65, i64* %18
br label %47
66: ; preds = %47
br label %67
67: ; preds = %66
%68 = load i64, i64* %17
%69 = add i64 %68, 1
store i64 %69, i64* %17
br label %42
70: ; preds = %42
br label %71
71: ; preds = %70
%72 = load i64, i64* %16
%73 = add i64 %72, 1
store i64 %73, i64* %16
br label %37
74: ; preds = %37
br label %75
75: ; preds = %74
%76 = load i64, i64* %15
%77 = add i64 %76, 1
store i64 %77, i64* %15
br label %32
78: ; preds = %32
store i64 0, i64* %19
br label %79
79: ; preds = %95, %78
%80 = load i64, i64* %19
%81 = load i64, i64* %10
%82 = icmp ult i64 %80, %81
br i1 %82, label %83, label %98
83: ; preds = %79
%84 = load i64, i64* %8
%85 = load i64, i64* %9
%86 = mul i64 %84, %85
%87 = load i64, i64* %7
%88 = mul i64 %86, %87
%89 = uitofp i64 %88 to float
%90 = load float*, float** %12
%91 = load i64, i64* %19
%92 = getelementptr inbounds float, float* %90, i64 %91
%93 = load float, float* %92
%94 = fdiv float %93, %89
store float %94, float* %92
br label %95
95: ; preds = %83
%96 = load i64, i64* %19
%97 = add i64 %96, 1
store i64 %97, i64* %19
br label %79
98: ; preds = %79
ret void
}

LLVM IR (-O2):
define void @neuralops_batchnorm2d_fwd_mean(i64 %0, i64 %1, i64 %2, i64 %3, float* nocapture %4, float* nocapture %5) {
%7 = icmp eq i64 %3, 0
br i1 %7, label %11, label %8
8: ; preds = %6
%9 = bitcast float* %5 to i8*
%10 = shl nuw i64 %3, 2
br label %11
11: ; preds = %8, %6
%12 = icmp eq i64 %0, 0
br i1 %12, label %24, label %13
13: ; preds = %11
%14 = icmp eq i64 %2, 0
%15 = icmp eq i64 %1, 0
%16 = add i64 %1, -1
%17 = and i64 %1, 3
%18 = icmp ult i64 %16, 3
%19 = and i64 %1, -4
%20 = icmp eq i64 %17, 0
br label %21
21: ; preds = %13, %76
%22 = phi i64 [ 0, %13 ], [ %78, %76 ]
%23 = phi i64 [ 0, %13 ], [ %77, %76 ]
br i1 %7, label %76, label %71
24: ; preds = %76, %11
br i1 %7, label %129, label %25
25: ; preds = %24
%26 = mul i64 %1, %0
%27 = mul i64 %26, %2
%28 = uitofp i64 %27 to float
%29 = icmp ult i64 %3, 4
br i1 %29, label %69, label %30
30: ; preds = %25
%31 = and i64 %3, -4
%32 = insertelement <4 x float> poison, float %28, i64 0
%33 = shufflevector <4 x float> %32, <4 x float> poison, <4 x i32> zeroinitializer
%34 = add i64 %31, -4
%35 = lshr exact i64 %34, 2
%36 = add nuw nsw i64 %35, 1
%37 = and i64 %36, 1
%38 = icmp eq i64 %34, 0
br i1 %38, label %58, label %39
39: ; preds = %30
%40 = and i64 %36, 9223372036854775806
br label %41
41: ; preds = %41, %39
%42 = phi i64 [ 0, %39 ], [ %55, %41 ]
%43 = phi i64 [ 0, %39 ], [ %56, %41 ]
%44 = getelementptr inbounds float, float* %5, i64 %42
%45 = bitcast float* %44 to <4 x float>*
%46 = load <4 x float>, <4 x float>* %45
%47 = fdiv <4 x float> %46, %33
%48 = bitcast float* %44 to <4 x float>*
store <4 x float> %47, <4 x float>* %48
%49 = or i64 %42, 4
%50 = getelementptr inbounds float, float* %5, i64 %49
%51 = bitcast float* %50 to <4 x float>*
%52 = load <4 x float>, <4 x float>* %51
%53 = fdiv <4 x float> %52, %33
%54 = bitcast float* %50 to <4 x float>*
store <4 x float> %53, <4 x float>* %54
%55 = add nuw i64 %42, 8
%56 = add i64 %43, 2
%57 = icmp eq i64 %56, %40
br i1 %57, label %58, label %41
58: ; preds = %41, %30
%59 = phi i64 [ 0, %30 ], [ %55, %41 ]
%60 = icmp eq i64 %37, 0
br i1 %60, label %67, label %61
61: ; preds = %58
%62 = getelementptr inbounds float, float* %5, i64 %59
%63 = bitcast float* %62 to <4 x float>*
%64 = load <4 x float>, <4 x float>* %63
%65 = fdiv <4 x float> %64, %33
%66 = bitcast float* %62 to <4 x float>*
store <4 x float> %65, <4 x float>* %66
br label %67
67: ; preds = %58, %61
%68 = icmp eq i64 %31, %3
br i1 %68, label %129, label %69
69: ; preds = %25, %67
%70 = phi i64 [ 0, %25 ], [ %31, %67 ]
br label %130
71: ; preds = %21, %85
%72 = phi i64 [ %87, %85 ], [ 0, %21 ]
%73 = phi i64 [ %86, %85 ], [ %23, %21 ]
br i1 %14, label %85, label %74
74: ; preds = %71
%75 = getelementptr inbounds float, float* %5, i64 %72
br label %80
76: ; preds = %85, %21
%77 = phi i64 [ %23, %21 ], [ %86, %85 ]
%78 = add nuw i64 %22, 1
%79 = icmp eq i64 %78, %0
br i1 %79, label %24, label %21
80: ; preds = %74, %103
%81 = phi i64 [ 0, %74 ], [ %105, %103 ]
%82 = phi i64 [ %73, %74 ], [ %104, %103 ]
br i1 %15, label %103, label %83
83: ; preds = %80
%84 = load float, float* %75
br i1 %18, label %89, label %107
85: ; preds = %103, %71
%86 = phi i64 [ %73, %71 ], [ %104, %103 ]
%87 = add nuw i64 %72, 1
%88 = icmp eq i64 %87, %3
br i1 %88, label %76, label %71
89: ; preds = %107, %83
%90 = phi i64 [ undef, %83 ], [ %126, %107 ]
%91 = phi float [ %84, %83 ], [ %125, %107 ]
%92 = phi i64 [ %82, %83 ], [ %126, %107 ]
br i1 %20, label %103, label %93
93: ; preds = %89, %93
%94 = phi float [ %99, %93 ], [ %91, %89 ]
%95 = phi i64 [ %100, %93 ], [ %92, %89 ]
%96 = phi i64 [ %101, %93 ], [ 0, %89 ]
%97 = getelementptr inbounds float, float* %4, i64 %95
%98 = load float, float* %97
%99 = fadd float %98, %94
store float %99, float* %75
%100 = add i64 %95, 1
%101 = add i64 %96, 1
%102 = icmp eq i64 %101, %17
br i1 %102, label %103, label %93
103: ; preds = %89, %93, %80
%104 = phi i64 [ %82, %80 ], [ %90, %89 ], [ %100, %93 ]
%105 = add nuw i64 %81, 1
%106 = icmp eq i64 %105, %2
br i1 %106, label %85, label %80
107: ; preds = %83, %107
%108 = phi float [ %125, %107 ], [ %84, %83 ]
%109 = phi i64 [ %126, %107 ], [ %82, %83 ]
%110 = phi i64 [ %127, %107 ], [ 0, %83 ]
%111 = getelementptr inbounds float, float* %4, i64 %109
%112 = load float, float* %111
%113 = fadd float %112, %108
store float %113, float* %75
%114 = add i64 %109, 1
%115 = getelementptr inbounds float, float* %4, i64 %114
%116 = load float, float* %115
%117 = fadd float %116, %113
store float %117, float* %75
%118 = add i64 %109, 2
%119 = getelementptr inbounds float, float* %4, i64 %118
%120 = load float, float* %119
%121 = fadd float %120, %117
store float %121, float* %75
%122 = add i64 %109, 3
%123 = getelementptr inbounds float, float* %4, i64 %122
%124 = load float, float* %123
%125 = fadd float %124, %121
store float %125, float* %75
%126 = add i64 %109, 4
%127 = add i64 %110, 4
%128 = icmp eq i64 %127, %19
br i1 %128, label %89, label %107
129: ; preds = %130, %67, %24
ret void
130: ; preds = %69, %130
%131 = phi i64 [ %135, %130 ], [ %70, %69 ]
%132 = getelementptr inbounds float, float* %5, i64 %131
%133 = load float, float* %132
%134 = fdiv float %133, %28
store float %134, float* %132
%135 = add nuw i64 %131, 1
%136 = icmp eq i64 %135, %3
br i1 %136, label %129, label %130
}