#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_2x2_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 out_width = (in_width + 1) / 2;
  size_t out_height = (in_height + 1) / 2;
  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] =
              0.25f *
              out_grad[x / 2 +
                       out_width * (y / 2 + out_height * (a + chan * idx))];
          p += 1;
        }
      }
    }
  }
}


x86 assembly (clang, -O0):
<neuralops_avgpool2d_2x2_bwd>:
push   %rbp
mov    %rsp,%rbp
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
add    $0x1,%rax
shr    $0x1,%rax
mov    %rax,-0x38(%rbp)
mov    -0x18(%rbp),%rax
add    $0x1,%rax
shr    $0x1,%rax
mov    %rax,-0x40(%rbp)
movq   $0x0,-0x48(%rbp)
movq   $0x0,-0x50(%rbp)
mov    -0x50(%rbp),%rax
cmp    -0x8(%rbp),%rax
jae    1253 <neuralops_avgpool2d_2x2_bwd+0x153>
movq   $0x0,-0x58(%rbp)
mov    -0x58(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    123d <neuralops_avgpool2d_2x2_bwd+0x13d>
movq   $0x0,-0x60(%rbp)
mov    -0x60(%rbp),%rax
cmp    -0x18(%rbp),%rax
jae    1227 <neuralops_avgpool2d_2x2_bwd+0x127>
movq   $0x0,-0x68(%rbp)
mov    -0x68(%rbp),%rax
cmp    -0x10(%rbp),%rax
jae    1211 <neuralops_avgpool2d_2x2_bwd+0x111>
mov    -0x30(%rbp),%rax
mov    -0x68(%rbp),%rcx
shr    $0x1,%rcx
mov    -0x38(%rbp),%rdx
mov    -0x60(%rbp),%rsi
shr    $0x1,%rsi
mov    -0x40(%rbp),%rdi
mov    -0x58(%rbp),%r8
mov    -0x20(%rbp),%r9
imul   -0x50(%rbp),%r9
add    %r9,%r8
imul   %r8,%rdi
add    %rdi,%rsi
imul   %rsi,%rdx
add    %rdx,%rcx
movss  0xe1e(%rip),%xmm0
mulss  (%rax,%rcx,4),%xmm0
mov    0x10(%rbp),%rax
mov    -0x48(%rbp),%rcx
movss  %xmm0,(%rax,%rcx,4)
mov    -0x48(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x48(%rbp)
mov    -0x68(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x68(%rbp)
jmp    1192 <neuralops_avgpool2d_2x2_bwd+0x92>
jmp    1216 <neuralops_avgpool2d_2x2_bwd+0x116>
mov    -0x60(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x60(%rbp)
jmp    117c <neuralops_avgpool2d_2x2_bwd+0x7c>
jmp    122c <neuralops_avgpool2d_2x2_bwd+0x12c>
mov    -0x58(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x58(%rbp)
jmp    1166 <neuralops_avgpool2d_2x2_bwd+0x66>
jmp    1242 <neuralops_avgpool2d_2x2_bwd+0x142>
mov    -0x50(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x50(%rbp)
jmp    1150 <neuralops_avgpool2d_2x2_bwd+0x50>
pop    %rbp
ret

x86 assembly (clang, -O2):
<neuralops_avgpool2d_2x2_bwd>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
mov    %rdi,-0x28(%rsp)
test   %rdi,%rdi
je     1281 <neuralops_avgpool2d_2x2_bwd+0x181>
mov    0x38(%rsp),%rdi
lea    0x1(%rsi),%r11
shr    %r11
lea    0x1(%rdx),%rbp
shr    %rbp
mov    %rsi,%rax
and    $0xfffffffffffffffe,%rax
mov    %rcx,%rbx
mov    %rbp,-0x38(%rsp)
imul   %rbp,%rbx
mov    %rbx,-0x30(%rsp)
lea    0x4(%rdi),%r13
xor    %r15d,%r15d
movss  0xeae(%rip),%xmm0
xor    %edi,%edi
xor    %ebx,%ebx
mov    %rcx,-0x8(%rsp)
jmp    117e <neuralops_avgpool2d_2x2_bwd+0x7e>
nopl   (%rax)
mov    -0x20(%rsp),%rdi
add    $0x1,%rdi
mov    -0x18(%rsp),%r15
add    -0x30(%rsp),%r15
cmp    -0x28(%rsp),%rdi
je     1281 <neuralops_avgpool2d_2x2_bwd+0x181>
test   %rcx,%rcx
mov    %r15,-0x18(%rsp)
mov    %rdi,-0x20(%rsp)
je     1160 <neuralops_avgpool2d_2x2_bwd+0x60>
imul   %rcx,%rdi
mov    %rdi,-0x10(%rsp)
xor    %edi,%edi
jmp    11b3 <neuralops_avgpool2d_2x2_bwd+0xb3>
nopw   0x0(%rax,%rax,1)
add    $0x1,%rdi
add    -0x38(%rsp),%r15
mov    -0x8(%rsp),%rcx
cmp    %rcx,%rdi
je     1160 <neuralops_avgpool2d_2x2_bwd+0x60>
test   %rdx,%rdx
je     11a0 <neuralops_avgpool2d_2x2_bwd+0xa0>
mov    -0x10(%rsp),%rcx
lea    (%rdi,%rcx,1),%r14
imul   -0x38(%rsp),%r14
xor    %r12d,%r12d
jmp    11d9 <neuralops_avgpool2d_2x2_bwd+0xd9>
nopl   0x0(%rax)
add    $0x1,%r12
cmp    %rdx,%r12
je     11a0 <neuralops_avgpool2d_2x2_bwd+0xa0>
test   %rsi,%rsi
je     11d0 <neuralops_avgpool2d_2x2_bwd+0xd0>
mov    %r12,%rbp
shr    %rbp
cmp    $0x1,%rsi
jne    1200 <neuralops_avgpool2d_2x2_bwd+0x100>
xor    %r10d,%r10d
test   $0x1,%sil
je     11d0 <neuralops_avgpool2d_2x2_bwd+0xd0>
jmp    1257 <neuralops_avgpool2d_2x2_bwd+0x157>
cs nopw 0x0(%rax,%rax,1)
nop
lea    (%r15,%rbp,1),%rcx
imul   %r11,%rcx
lea    (%r9,%rcx,4),%rcx
lea    0x0(,%rbx,4),%r8
add    %r13,%r8
xor    %r10d,%r10d
nopw   0x0(%rax,%rax,1)
movss  (%rcx,%r10,2),%xmm1
mulss  %xmm0,%xmm1
movss  %xmm1,-0x4(%r8,%r10,4)
movss  (%rcx,%r10,2),%xmm1
mulss  %xmm0,%xmm1
movss  %xmm1,(%r8,%r10,4)
add    $0x2,%r10
cmp    %r10,%rax
jne    1220 <neuralops_avgpool2d_2x2_bwd+0x120>
add    %r10,%rbx
test   $0x1,%sil
je     11d0 <neuralops_avgpool2d_2x2_bwd+0xd0>
add    %r14,%rbp
imul   %r11,%rbp
shr    %r10
add    %rbp,%r10
movss  (%r9,%r10,4),%xmm1
mulss  %xmm0,%xmm1
mov    0x38(%rsp),%rcx
movss  %xmm1,(%rcx,%rbx,4)
add    $0x1,%rbx
jmp    11d0 <neuralops_avgpool2d_2x2_bwd+0xd0>
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

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

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