#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 */
typedef size_t uint32_t;

/* Variables and functions */
void neuralops_caffe_maxpool2d_bwd(
    size_t batch_sz, size_t in_width, size_t in_height, size_t chan,
    size_t out_width, size_t out_height, const uint32_t *mask_buf,
    const float *out_grad, float *in_grad, size_t kernel_w_, size_t kernel_h_,
    size_t stride_w_, size_t stride_h_, size_t pad_w_, size_t pad_h_) {
  size_t bottom_count = in_width * in_height * chan * batch_sz;
  for (size_t i = 0; i < bottom_count; i++) {
    in_grad[i] = 0.0f;
  }
  for (size_t n = 0; n < batch_sz; n++) {
    float *bottom_diff = in_grad + n * in_width * in_height * chan;
    const uint32_t *top_mask = mask_buf + n * out_width * out_height * chan;
    const float *top_diff = out_grad + n * out_width * out_height * chan;
    for (size_t c = 0; c < chan; ++c) {
      for (size_t ph = 0; ph < out_height; ++ph) {
        for (size_t pw = 0; pw < out_width; ++pw) {
          size_t index = ph * out_width + pw;
          uint32_t bottom_index = top_mask[index];
          bottom_diff[bottom_index] += top_diff[index];
        }
      }
      bottom_diff += in_width * in_height;
      top_mask += out_width * out_height;
      top_diff += out_width * out_height;
    }
  }
}


x86 assembly (clang, -O0):
<neuralops_caffe_maxpool2d_bwd>:
push   %rbp
mov    %rsp,%rbp
push   %rax
mov    0x50(%rbp),%rax
mov    0x48(%rbp),%rax
mov    0x40(%rbp),%rax
mov    0x38(%rbp),%rax
mov    0x30(%rbp),%rax
mov    0x28(%rbp),%rax
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
imul   -0x18(%rbp),%rax
imul   -0x20(%rbp),%rax
imul   -0x8(%rbp),%rax
mov    %rax,-0x38(%rbp)
movq   $0x0,-0x40(%rbp)
mov    -0x40(%rbp),%rax
cmp    -0x38(%rbp),%rax
jae    118f <neuralops_caffe_maxpool2d_bwd+0x8f>
mov    0x20(%rbp),%rax
mov    -0x40(%rbp),%rcx
xorps  %xmm0,%xmm0
movss  %xmm0,(%rax,%rcx,4)
mov    -0x40(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x40(%rbp)
jmp    1160 <neuralops_caffe_maxpool2d_bwd+0x60>
movq   $0x0,-0x48(%rbp)
mov    -0x48(%rbp),%rax
cmp    -0x8(%rbp),%rax
jae    1320 <neuralops_caffe_maxpool2d_bwd+0x220>
mov    0x20(%rbp),%rax
mov    -0x48(%rbp),%rcx
imul   -0x10(%rbp),%rcx
imul   -0x18(%rbp),%rcx
imul   -0x20(%rbp),%rcx
shl    $0x2,%rcx
add    %rcx,%rax
mov    %rax,-0x50(%rbp)
mov    0x10(%rbp),%rax
mov    -0x48(%rbp),%rcx
imul   -0x28(%rbp),%rcx
imul   -0x30(%rbp),%rcx
imul   -0x20(%rbp),%rcx
shl    $0x3,%rcx
add    %rcx,%rax
mov    %rax,-0x58(%rbp)
mov    0x18(%rbp),%rax
mov    -0x48(%rbp),%rcx
imul   -0x28(%rbp),%rcx
imul   -0x30(%rbp),%rcx
imul   -0x20(%rbp),%rcx
shl    $0x2,%rcx
add    %rcx,%rax
mov    %rax,-0x60(%rbp)
movq   $0x0,-0x68(%rbp)
mov    -0x68(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    130a <neuralops_caffe_maxpool2d_bwd+0x20a>
movq   $0x0,-0x70(%rbp)
mov    -0x70(%rbp),%rax
cmp    -0x30(%rbp),%rax
jae    12ba <neuralops_caffe_maxpool2d_bwd+0x1ba>
movq   $0x0,-0x78(%rbp)
mov    -0x78(%rbp),%rax
cmp    -0x28(%rbp),%rax
jae    12a4 <neuralops_caffe_maxpool2d_bwd+0x1a4>
mov    -0x70(%rbp),%rax
imul   -0x28(%rbp),%rax
add    -0x78(%rbp),%rax
mov    %rax,-0x80(%rbp)
mov    -0x58(%rbp),%rax
mov    -0x80(%rbp),%rcx
mov    (%rax,%rcx,8),%rax
mov    %rax,-0x88(%rbp)
mov    -0x60(%rbp),%rax
mov    -0x80(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm0
mov    -0x50(%rbp),%rax
mov    -0x88(%rbp),%rcx
addss  (%rax,%rcx,4),%xmm0
movss  %xmm0,(%rax,%rcx,4)
mov    -0x78(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x78(%rbp)
jmp    123f <neuralops_caffe_maxpool2d_bwd+0x13f>
jmp    12a9 <neuralops_caffe_maxpool2d_bwd+0x1a9>
mov    -0x70(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x70(%rbp)
jmp    1229 <neuralops_caffe_maxpool2d_bwd+0x129>
mov    -0x10(%rbp),%rax
imul   -0x18(%rbp),%rax
shl    $0x2,%rax
add    -0x50(%rbp),%rax
mov    %rax,-0x50(%rbp)
mov    -0x28(%rbp),%rax
imul   -0x30(%rbp),%rax
shl    $0x3,%rax
add    -0x58(%rbp),%rax
mov    %rax,-0x58(%rbp)
mov    -0x28(%rbp),%rax
imul   -0x30(%rbp),%rax
shl    $0x2,%rax
add    -0x60(%rbp),%rax
mov    %rax,-0x60(%rbp)
mov    -0x68(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x68(%rbp)
jmp    1213 <neuralops_caffe_maxpool2d_bwd+0x113>
jmp    130f <neuralops_caffe_maxpool2d_bwd+0x20f>
mov    -0x48(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x48(%rbp)
jmp    1197 <neuralops_caffe_maxpool2d_bwd+0x97>
add    $0x8,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<neuralops_caffe_maxpool2d_bwd>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
sub    $0x78,%rsp
mov    %r9,%r14
mov    %r8,%r13
mov    %rcx,%r15
mov    %rdi,%rbp
mov    0xc0(%rsp),%rdi
mov    %rdx,%rbx
imul   %rsi,%rbx
mov    %rcx,%rax
imul   %rbp,%rax
mov    %rbx,%rcx
mov    %rbx,0x28(%rsp)
imul   %rbx,%rax
test   %rax,%rax
je     1168 <neuralops_caffe_maxpool2d_bwd+0x58>
imul   %r15,%rdx
imul   %rsi,%rdx
imul   %rbp,%rdx
shl    $0x2,%rdx
xor    %esi,%esi
call   1030 <memset@plt>
test   %rbp,%rbp
je     1388 <neuralops_caffe_maxpool2d_bwd+0x278>
mov    0xb8(%rsp),%rax
mov    0xb0(%rsp),%rcx
mov    0x28(%rsp),%rdx
imul   %r15,%rdx
mov    %rdx,0x38(%rsp)
mov    %r14,%rdx
imul   %r13,%rdx
mov    %rdx,%rsi
imul   %r15,%rsi
mov    %r13,%rbx
and    $0xfffffffffffffffe,%rbx
add    $0x4,%rax
mov    %rax,0x20(%rsp)
lea    0x0(,%rsi,4),%rax
mov    %rax,0x48(%rsp)
lea    0x0(,%rdx,4),%rax
mov    %rax,0x60(%rsp)
lea    0x0(,%r13,4),%rdi
lea    0x8(%rcx),%rax
mov    %rax,0x18(%rsp)
mov    %rsi,0x30(%rsp)
lea    0x0(,%rsi,8),%rax
mov    %rax,0x40(%rsp)
mov    %rdx,0x68(%rsp)
lea    0x0(,%rdx,8),%rax
mov    %rax,0x58(%rsp)
lea    0x0(,%r13,8),%r11
xor    %eax,%eax
mov    %rax,(%rsp)
mov    %r15,0x70(%rsp)
mov    %rbp,0x50(%rsp)
jmp    1251 <neuralops_caffe_maxpool2d_bwd+0x141>
cs nopw 0x0(%rax,%rax,1)
mov    (%rsp),%rcx
add    $0x1,%rcx
mov    0x48(%rsp),%rax
add    %rax,0x20(%rsp)
mov    0x40(%rsp),%rax
add    %rax,0x18(%rsp)
mov    0x50(%rsp),%rbp
mov    %rcx,%rax
mov    %rcx,(%rsp)
cmp    %rbp,%rcx
je     1388 <neuralops_caffe_maxpool2d_bwd+0x278>
test   %r15,%r15
je     1220 <neuralops_caffe_maxpool2d_bwd+0x110>
mov    0x30(%rsp),%rax
mov    (%rsp),%rsi
imul   %rsi,%rax
mov    0xb8(%rsp),%rcx
lea    (%rcx,%rax,4),%rcx
mov    0xb0(%rsp),%rdx
lea    (%rdx,%rax,8),%r9
mov    0x38(%rsp),%rax
imul   %rsi,%rax
mov    0xc0(%rsp),%rdx
lea    (%rdx,%rax,4),%rax
mov    0x18(%rsp),%rdx
mov    %rdx,0x10(%rsp)
mov    0x20(%rsp),%rdx
mov    %rdx,0x8(%rsp)
xor    %ebp,%ebp
jmp    12f6 <neuralops_caffe_maxpool2d_bwd+0x1e6>
nopl   0x0(%rax,%rax,1)
mov    0x28(%rsp),%rdx
lea    (%rax,%rdx,4),%rax
mov    0x68(%rsp),%rdx
lea    (%r9,%rdx,8),%r9
lea    (%rcx,%rdx,4),%rcx
add    $0x1,%rbp
mov    0x8(%rsp),%rdx
add    0x60(%rsp),%rdx
mov    %rdx,0x8(%rsp)
mov    0x10(%rsp),%rdx
add    0x58(%rsp),%rdx
mov    %rdx,0x10(%rsp)
mov    0x70(%rsp),%r15
cmp    %r15,%rbp
je     1220 <neuralops_caffe_maxpool2d_bwd+0x110>
test   %r14,%r14
je     12b0 <neuralops_caffe_maxpool2d_bwd+0x1a0>
mov    0x10(%rsp),%r8
mov    0x8(%rsp),%r12
xor    %r15d,%r15d
jmp    131f <neuralops_caffe_maxpool2d_bwd+0x20f>
nopw   0x0(%rax,%rax,1)
add    $0x1,%r15
add    %rdi,%r12
add    %r11,%r8
cmp    %r14,%r15
je     12b0 <neuralops_caffe_maxpool2d_bwd+0x1a0>
test   %r13,%r13
je     1310 <neuralops_caffe_maxpool2d_bwd+0x200>
xor    %esi,%esi
cmp    $0x1,%r13
je     1363 <neuralops_caffe_maxpool2d_bwd+0x253>
nopl   0x0(%rax)
mov    -0x8(%r8,%rsi,8),%rdx
movss  -0x4(%r12,%rsi,4),%xmm0
addss  (%rax,%rdx,4),%xmm0
movss  %xmm0,(%rax,%rdx,4)
mov    (%r8,%rsi,8),%rdx
movss  (%r12,%rsi,4),%xmm0
addss  (%rax,%rdx,4),%xmm0
movss  %xmm0,(%rax,%rdx,4)
add    $0x2,%rsi
cmp    %rsi,%rbx
jne    1330 <neuralops_caffe_maxpool2d_bwd+0x220>
test   $0x1,%r13b
je     1310 <neuralops_caffe_maxpool2d_bwd+0x200>
mov    %r15,%r10
imul   %r13,%r10
add    %r10,%rsi
mov    (%r9,%rsi,8),%rdx
movss  (%rcx,%rsi,4),%xmm0
addss  (%rax,%rdx,4),%xmm0
movss  %xmm0,(%rax,%rdx,4)
jmp    1310 <neuralops_caffe_maxpool2d_bwd+0x200>
add    $0x78,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

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

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