#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_avgpool2d_bwd(size_t batch_sz, size_t in_width, size_t in_height,
                             size_t chan, const float *in_buf,
                             const float *out_grad, float *in_grad,
                             size_t pool_w, size_t pool_h) {
  size_t out_width = (in_width + pool_w - 1) / pool_w;
  size_t out_height = (in_height + pool_h - 1) / pool_h;
  float normalizer = 1.0f / ((float)(pool_w * pool_h));
  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 < in_height; y += 1) {
        for (size_t x = 0; x < in_width; x += 1) {
          in_grad[p] = normalizer *
                       out_grad[x / pool_w +
                                out_width * (y / pool_h +
                                             out_height * (a + chan * idx))];
          p += 1;
        }
      }
    }
  }
}


x86 assembly (clang, -O0):
<neuralops_avgpool2d_bwd>:
push   %rbp
mov    %rsp,%rbp
sub    $0x10,%rsp
mov    0x20(%rbp),%rax
mov    0x18(%rbp),%rax
mov    0x10(%rbp),%rax
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %rdx,-0x18(%rbp)
mov    %rcx,-0x20(%rbp)
mov    %r8,-0x28(%rbp)
mov    %r9,-0x30(%rbp)
mov    -0x10(%rbp),%rax
mov    0x18(%rbp),%rcx
lea    -0x1(%rax,%rcx,1),%rax
xor    %edx,%edx
mov    %rdx,-0x80(%rbp)
div    %rcx
mov    -0x80(%rbp),%rdx
mov    %rax,-0x38(%rbp)
mov    -0x18(%rbp),%rax
mov    0x20(%rbp),%rcx
lea    -0x1(%rax,%rcx,1),%rax
div    %rcx
mov    %rax,-0x40(%rbp)
mov    0x18(%rbp),%rax
mov    0x20(%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,-0x78(%rbp)
test   %rax,%rax
movss  %xmm0,-0x74(%rbp)
js     11a3 <neuralops_avgpool2d_bwd+0xa3>
movss  -0x78(%rbp),%xmm0
movss  %xmm0,-0x74(%rbp)
movss  -0x74(%rbp),%xmm1
movss  0xe50(%rip),%xmm0
divss  %xmm1,%xmm0
movss  %xmm0,-0x44(%rbp)
movq   $0x0,-0x50(%rbp)
movq   $0x0,-0x58(%rbp)
mov    -0x58(%rbp),%rax
cmp    -0x8(%rbp),%rax
jae    12f1 <neuralops_avgpool2d_bwd+0x1f1>
movq   $0x0,-0x60(%rbp)
mov    -0x60(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    12db <neuralops_avgpool2d_bwd+0x1db>
movq   $0x0,-0x68(%rbp)
mov    -0x68(%rbp),%rax
cmp    -0x18(%rbp),%rax
jae    12c5 <neuralops_avgpool2d_bwd+0x1c5>
movq   $0x0,-0x70(%rbp)
mov    -0x70(%rbp),%rax
cmp    -0x10(%rbp),%rax
jae    12af <neuralops_avgpool2d_bwd+0x1af>
movss  -0x44(%rbp),%xmm0
mov    -0x30(%rbp),%rax
mov    %rax,-0x88(%rbp)
mov    -0x70(%rbp),%rax
xor    %ecx,%ecx
mov    %ecx,%edx
divq   0x18(%rbp)
mov    %rax,%rcx
mov    -0x38(%rbp),%rax
mov    %rax,-0x90(%rbp)
mov    -0x68(%rbp),%rax
xor    %edx,%edx
divq   0x20(%rbp)
mov    -0x90(%rbp),%rdx
mov    %rax,%rsi
mov    -0x88(%rbp),%rax
mov    -0x40(%rbp),%rdi
mov    -0x60(%rbp),%r8
mov    -0x20(%rbp),%r9
imul   -0x58(%rbp),%r9
add    %r9,%r8
imul   %r8,%rdi
add    %rdi,%rsi
imul   %rsi,%rdx
add    %rdx,%rcx
mulss  (%rax,%rcx,4),%xmm0
mov    0x10(%rbp),%rax
mov    -0x50(%rbp),%rcx
movss  %xmm0,(%rax,%rcx,4)
mov    -0x50(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x50(%rbp)
mov    -0x70(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x70(%rbp)
jmp    120b <neuralops_avgpool2d_bwd+0x10b>
jmp    12b4 <neuralops_avgpool2d_bwd+0x1b4>
mov    -0x68(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x68(%rbp)
jmp    11f5 <neuralops_avgpool2d_bwd+0xf5>
jmp    12ca <neuralops_avgpool2d_bwd+0x1ca>
mov    -0x60(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x60(%rbp)
jmp    11df <neuralops_avgpool2d_bwd+0xdf>
jmp    12e0 <neuralops_avgpool2d_bwd+0x1e0>
mov    -0x58(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x58(%rbp)
jmp    11c9 <neuralops_avgpool2d_bwd+0xc9>
add    $0x10,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<neuralops_avgpool2d_bwd>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
mov    %rdx,%r8
mov    %rsi,%r11
mov    %rdi,-0x30(%rsp)
mov    0x48(%rsp),%r12
mov    0x40(%rsp),%r13
lea    (%rsi,%r13,1),%rax
add    $0xffffffffffffffff,%rax
mov    %rax,%rdx
or     %r13,%rdx
shr    $0x20,%rdx
je     113d <neuralops_avgpool2d_bwd+0x3d>
xor    %edx,%edx
div    %r13
mov    %rax,%r10
jmp    1145 <neuralops_avgpool2d_bwd+0x45>
xor    %edx,%edx
div    %r13d
mov    %eax,%r10d
lea    (%r8,%r12,1),%rax
add    $0xffffffffffffffff,%rax
mov    %rax,%rdx
or     %r12,%rdx
shr    $0x20,%rdx
je     1160 <neuralops_avgpool2d_bwd+0x60>
xor    %edx,%edx
div    %r12
jmp    1165 <neuralops_avgpool2d_bwd+0x65>
xor    %edx,%edx
div    %r12d
mov    %rax,-0x28(%rsp)
mov    %r12,%rax
imul   %r13,%rax
test   %rax,%rax
js     1188 <neuralops_avgpool2d_bwd+0x88>
cvtsi2ss %rax,%xmm1
cmpq   $0x0,-0x30(%rsp)
jne    11a9 <neuralops_avgpool2d_bwd+0xa9>
jmp    1389 <neuralops_avgpool2d_bwd+0x289>
mov    %rax,%rdx
shr    %rdx
and    $0x1,%eax
or     %rdx,%rax
cvtsi2ss %rax,%xmm1
addss  %xmm1,%xmm1
cmpq   $0x0,-0x30(%rsp)
je     1389 <neuralops_avgpool2d_bwd+0x289>
mov    0x38(%rsp),%rax
movss  0xe4a(%rip),%xmm0
divss  %xmm1,%xmm0
mov    %r11,%rsi
and    $0xfffffffffffffffe,%rsi
add    $0x4,%rax
mov    %rax,-0x8(%rsp)
xor    %eax,%eax
mov    %rax,-0x38(%rsp)
xor    %ebp,%ebp
mov    %rcx,-0x18(%rsp)
jmp    11fc <neuralops_avgpool2d_bwd+0xfc>
nopw   0x0(%rax,%rax,1)
mov    -0x38(%rsp),%rdx
add    $0x1,%rdx
mov    %rdx,%rax
mov    %rdx,-0x38(%rsp)
cmp    -0x30(%rsp),%rdx
je     1389 <neuralops_avgpool2d_bwd+0x289>
test   %rcx,%rcx
je     11e0 <neuralops_avgpool2d_bwd+0xe0>
mov    -0x38(%rsp),%rax
imul   %rcx,%rax
mov    %rax,-0x20(%rsp)
xor    %edx,%edx
jmp    1233 <neuralops_avgpool2d_bwd+0x133>
cs nopw 0x0(%rax,%rax,1)
nopl   (%rax)
mov    -0x10(%rsp),%rdx
add    $0x1,%rdx
mov    -0x18(%rsp),%rcx
cmp    %rcx,%rdx
je     11e0 <neuralops_avgpool2d_bwd+0xe0>
test   %r8,%r8
mov    %rdx,-0x10(%rsp)
je     1220 <neuralops_avgpool2d_bwd+0x120>
mov    -0x20(%rsp),%rax
lea    (%rdx,%rax,1),%rdi
imul   -0x28(%rsp),%rdi
xor    %r14d,%r14d
jmp    127d <neuralops_avgpool2d_bwd+0x17d>
mov    %rbx,%rax
xor    %edx,%edx
div    %r13
add    %rax,%r15
movss  (%r9,%r15,4),%xmm1
mulss  %xmm0,%xmm1
mov    0x38(%rsp),%rax
movss  %xmm1,(%rax,%rbp,4)
add    $0x1,%rbp
add    $0x1,%r14
cmp    %r8,%r14
je     1220 <neuralops_avgpool2d_bwd+0x120>
test   %r11,%r11
je     1274 <neuralops_avgpool2d_bwd+0x174>
mov    %r14,%rax
or     %r12,%rax
shr    $0x20,%rax
je     12a0 <neuralops_avgpool2d_bwd+0x1a0>
mov    %r14,%rax
xor    %edx,%edx
div    %r12
mov    %rax,%r15
jmp    12ab <neuralops_avgpool2d_bwd+0x1ab>
nopl   0x0(%rax,%rax,1)
mov    %r14d,%eax
xor    %edx,%edx
div    %r12d
mov    %eax,%r15d
add    %rdi,%r15
imul   %r10,%r15
cmp    $0x1,%r11
jne    12d0 <neuralops_avgpool2d_bwd+0x1d0>
xor    %ebx,%ebx
test   $0x1,%r11b
je     1274 <neuralops_avgpool2d_bwd+0x174>
jmp    136d <neuralops_avgpool2d_bwd+0x26d>
cs nopw 0x0(%rax,%rax,1)
nop
mov    -0x8(%rsp),%rax
lea    (%rax,%rbp,4),%rcx
xor    %ebx,%ebx
jmp    1300 <neuralops_avgpool2d_bwd+0x200>
nopl   (%rax)
xor    %edx,%edx
div    %r13
add    %r15,%rax
movss  (%r9,%rax,4),%xmm1
mulss  %xmm0,%xmm1
movss  %xmm1,(%rcx,%rbx,4)
add    $0x2,%rbx
cmp    %rbx,%rsi
je     1360 <neuralops_avgpool2d_bwd+0x260>
mov    %rbx,%rax
or     %r13,%rax
shr    $0x20,%rax
je     1320 <neuralops_avgpool2d_bwd+0x220>
mov    %rbx,%rax
xor    %edx,%edx
div    %r13
jmp    1327 <neuralops_avgpool2d_bwd+0x227>
cs nopw 0x0(%rax,%rax,1)
mov    %ebx,%eax
xor    %edx,%edx
div    %r13d
add    %r15,%rax
movss  (%r9,%rax,4),%xmm1
mulss  %xmm0,%xmm1
movss  %xmm1,-0x4(%rcx,%rbx,4)
lea    0x1(%rbx),%rax
mov    %rax,%rdx
or     %r13,%rdx
shr    $0x20,%rdx
jne    12e0 <neuralops_avgpool2d_bwd+0x1e0>
xor    %edx,%edx
div    %r13d
jmp    12e5 <neuralops_avgpool2d_bwd+0x1e5>
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax,%rax,1)
add    %rbx,%rbp
test   $0x1,%r11b
je     1274 <neuralops_avgpool2d_bwd+0x174>
mov    %rbx,%rax
or     %r13,%rax
shr    $0x20,%rax
jne    1251 <neuralops_avgpool2d_bwd+0x151>
mov    %ebx,%eax
xor    %edx,%edx
div    %r13d
jmp    1259 <neuralops_avgpool2d_bwd+0x159>
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

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

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