#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_conv2d_bias_bwd(size_t batch_sz, size_t out_width,
                               size_t out_height, size_t out_chan,
                               const float *out_grad, float *bias_grad) {
  size_t p = 0;
  for (size_t idx = 0; idx < batch_sz; idx += 1) {
    for (size_t a = 0; a < out_chan; a += 1) {
      for (size_t y = 0; y < out_height; y += 1) {
        for (size_t x = 0; x < out_width; x += 1) {
          float g = out_grad[p];
          bias_grad[a] += g;
          p += 1;
        }
      }
    }
  }
}


x86 assembly (clang, -O0):
<neuralops_conv2d_bias_bwd>:
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)
movq   $0x0,-0x40(%rbp)
mov    -0x40(%rbp),%rax
cmp    -0x8(%rbp),%rax
jae    1204 <neuralops_conv2d_bias_bwd+0x104>
movq   $0x0,-0x48(%rbp)
mov    -0x48(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    11ee <neuralops_conv2d_bias_bwd+0xee>
movq   $0x0,-0x50(%rbp)
mov    -0x50(%rbp),%rax
cmp    -0x18(%rbp),%rax
jae    11d8 <neuralops_conv2d_bias_bwd+0xd8>
movq   $0x0,-0x58(%rbp)
mov    -0x58(%rbp),%rax
cmp    -0x10(%rbp),%rax
jae    11c2 <neuralops_conv2d_bias_bwd+0xc2>
mov    -0x28(%rbp),%rax
mov    -0x38(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm0
movss  %xmm0,-0x5c(%rbp)
movss  -0x5c(%rbp),%xmm0
mov    -0x30(%rbp),%rax
mov    -0x48(%rbp),%rcx
addss  (%rax,%rcx,4),%xmm0
movss  %xmm0,(%rax,%rcx,4)
mov    -0x38(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x38(%rbp)
mov    -0x58(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x58(%rbp)
jmp    116e <neuralops_conv2d_bias_bwd+0x6e>
jmp    11c7 <neuralops_conv2d_bias_bwd+0xc7>
mov    -0x50(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x50(%rbp)
jmp    1158 <neuralops_conv2d_bias_bwd+0x58>
jmp    11dd <neuralops_conv2d_bias_bwd+0xdd>
mov    -0x48(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x48(%rbp)
jmp    1142 <neuralops_conv2d_bias_bwd+0x42>
jmp    11f3 <neuralops_conv2d_bias_bwd+0xf3>
mov    -0x40(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x40(%rbp)
jmp    112c <neuralops_conv2d_bias_bwd+0x2c>
pop    %rbp
ret

x86 assembly (clang, -O2):
<neuralops_conv2d_bias_bwd>:
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
test   %rdi,%rdi
je     11fb <neuralops_conv2d_bias_bwd+0xfb>
lea    -0x1(%rsi),%r10
mov    %esi,%r15d
and    $0x3,%r15d
mov    %rsi,%r11
and    $0xfffffffffffffffc,%r11
xor    %ebx,%ebx
xor    %r14d,%r14d
jmp    113d <neuralops_conv2d_bias_bwd+0x3d>
nopl   0x0(%rax,%rax,1)
add    $0x1,%r14
cmp    %rdi,%r14
je     11fb <neuralops_conv2d_bias_bwd+0xfb>
test   %rcx,%rcx
je     1130 <neuralops_conv2d_bias_bwd+0x30>
xor    %r13d,%r13d
jmp    1159 <neuralops_conv2d_bias_bwd+0x59>
nopw   0x0(%rax,%rax,1)
add    $0x1,%r13
cmp    %rcx,%r13
je     1130 <neuralops_conv2d_bias_bwd+0x30>
test   %rdx,%rdx
je     1150 <neuralops_conv2d_bias_bwd+0x50>
xor    %r12d,%r12d
jmp    1179 <neuralops_conv2d_bias_bwd+0x79>
cs nopw 0x0(%rax,%rax,1)
nopl   (%rax)
add    $0x1,%r12
cmp    %rdx,%r12
je     1150 <neuralops_conv2d_bias_bwd+0x50>
test   %rsi,%rsi
je     1170 <neuralops_conv2d_bias_bwd+0x70>
movss  (%r9,%r13,4),%xmm0
mov    %r11,%rax
cmp    $0x3,%r10
jb     11cd <neuralops_conv2d_bias_bwd+0xcd>
nopl   (%rax)
addss  (%r8,%rbx,4),%xmm0
movss  %xmm0,(%r9,%r13,4)
addss  0x4(%r8,%rbx,4),%xmm0
movss  %xmm0,(%r9,%r13,4)
addss  0x8(%r8,%rbx,4),%xmm0
movss  %xmm0,(%r9,%r13,4)
addss  0xc(%r8,%rbx,4),%xmm0
movss  %xmm0,(%r9,%r13,4)
add    $0x4,%rbx
add    $0xfffffffffffffffc,%rax
jne    1190 <neuralops_conv2d_bias_bwd+0x90>
test   %r15,%r15
je     1170 <neuralops_conv2d_bias_bwd+0x70>
mov    %r15,%rax
cs nopw 0x0(%rax,%rax,1)
nop
addss  (%r8,%rbx,4),%xmm0
movss  %xmm0,(%r9,%r13,4)
add    $0x1,%rbx
add    $0xffffffffffffffff,%rax
jne    11e0 <neuralops_conv2d_bias_bwd+0xe0>
jmp    1170 <neuralops_conv2d_bias_bwd+0x70>
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
ret

LLVM IR (-O0):
define void @neuralops_conv2d_bias_bwd(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 float
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
store i64 0, i64* %14
br label %19
19: ; preds = %63, %6
%20 = load i64, i64* %14
%21 = load i64, i64* %7
%22 = icmp ult i64 %20, %21
br i1 %22, label %23, label %66
23: ; preds = %19
store i64 0, i64* %15
br label %24
24: ; preds = %59, %23
%25 = load i64, i64* %15
%26 = load i64, i64* %10
%27 = icmp ult i64 %25, %26
br i1 %27, label %28, label %62
28: ; preds = %24
store i64 0, i64* %16
br label %29
29: ; preds = %55, %28
%30 = load i64, i64* %16
%31 = load i64, i64* %9
%32 = icmp ult i64 %30, %31
br i1 %32, label %33, label %58
33: ; preds = %29
store i64 0, i64* %17
br label %34
34: ; preds = %51, %33
%35 = load i64, i64* %17
%36 = load i64, i64* %8
%37 = icmp ult i64 %35, %36
br i1 %37, label %38, label %54
38: ; preds = %34
%39 = load float*, float** %11
%40 = load i64, i64* %13
%41 = getelementptr inbounds float, float* %39, i64 %40
%42 = load float, float* %41
store float %42, float* %18
%43 = load float, float* %18
%44 = load float*, float** %12
%45 = load i64, i64* %15
%46 = getelementptr inbounds float, float* %44, i64 %45
%47 = load float, float* %46
%48 = fadd float %47, %43
store float %48, float* %46
%49 = load i64, i64* %13
%50 = add i64 %49, 1
store i64 %50, i64* %13
br label %51
51: ; preds = %38
%52 = load i64, i64* %17
%53 = add i64 %52, 1
store i64 %53, i64* %17
br label %34
54: ; preds = %34
br label %55
55: ; preds = %54
%56 = load i64, i64* %16
%57 = add i64 %56, 1
store i64 %57, i64* %16
br label %29
58: ; preds = %29
br label %59
59: ; preds = %58
%60 = load i64, i64* %15
%61 = add i64 %60, 1
store i64 %61, i64* %15
br label %24
62: ; preds = %24
br label %63
63: ; preds = %62
%64 = load i64, i64* %14
%65 = add i64 %64, 1
store i64 %65, i64* %14
br label %19
66: ; preds = %19
ret void
}

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