#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_scale_bias_bwd(size_t batch_sz, size_t out_width,
                                     size_t out_height, size_t out_chan,
                                     const float *in_buf, const float *scale,
                                     const float *out_grad, float *scale_grad,
                                     float *bias_grad, float *in_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) {
      float s = scale[a];
      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];
          scale_grad[a] += g * in_buf[p];
          bias_grad[a] += g;
          in_grad[p] = g * s;
          p += 1;
        }
      }
    }
  }
}


x86 assembly (clang, -O0):
<neuralops_conv2d_scale_bias_bwd>:
push   %rbp
mov    %rsp,%rbp
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)
movq   $0x0,-0x38(%rbp)
movq   $0x0,-0x40(%rbp)
mov    -0x40(%rbp),%rax
cmp    -0x8(%rbp),%rax
jae    1269 <neuralops_conv2d_scale_bias_bwd+0x169>
movq   $0x0,-0x48(%rbp)
mov    -0x48(%rbp),%rax
cmp    -0x20(%rbp),%rax
jae    1253 <neuralops_conv2d_scale_bias_bwd+0x153>
mov    -0x30(%rbp),%rax
mov    -0x48(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm0
movss  %xmm0,-0x4c(%rbp)
movq   $0x0,-0x58(%rbp)
mov    -0x58(%rbp),%rax
cmp    -0x18(%rbp),%rax
jae    123d <neuralops_conv2d_scale_bias_bwd+0x13d>
movq   $0x0,-0x60(%rbp)
mov    -0x60(%rbp),%rax
cmp    -0x10(%rbp),%rax
jae    1227 <neuralops_conv2d_scale_bias_bwd+0x127>
mov    0x10(%rbp),%rax
mov    -0x38(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm0
movss  %xmm0,-0x64(%rbp)
movss  -0x64(%rbp),%xmm0
mov    -0x28(%rbp),%rax
mov    -0x38(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm2
mov    0x18(%rbp),%rax
mov    -0x48(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm1
mulss  %xmm2,%xmm0
addss  %xmm1,%xmm0
movss  %xmm0,(%rax,%rcx,4)
movss  -0x64(%rbp),%xmm0
mov    0x20(%rbp),%rax
mov    -0x48(%rbp),%rcx
addss  (%rax,%rcx,4),%xmm0
movss  %xmm0,(%rax,%rcx,4)
movss  -0x64(%rbp),%xmm0
mulss  -0x4c(%rbp),%xmm0
mov    0x28(%rbp),%rax
mov    -0x38(%rbp),%rcx
movss  %xmm0,(%rax,%rcx,4)
mov    -0x38(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x38(%rbp)
mov    -0x60(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x60(%rbp)
jmp    1190 <neuralops_conv2d_scale_bias_bwd+0x90>
jmp    122c <neuralops_conv2d_scale_bias_bwd+0x12c>
mov    -0x58(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x58(%rbp)
jmp    117a <neuralops_conv2d_scale_bias_bwd+0x7a>
jmp    1242 <neuralops_conv2d_scale_bias_bwd+0x142>
mov    -0x48(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x48(%rbp)
jmp    1152 <neuralops_conv2d_scale_bias_bwd+0x52>
jmp    1258 <neuralops_conv2d_scale_bias_bwd+0x158>
mov    -0x40(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x40(%rbp)
jmp    113c <neuralops_conv2d_scale_bias_bwd+0x3c>
pop    %rbp
ret

x86 assembly (clang, -O2):
<neuralops_conv2d_scale_bias_bwd>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
test   %rdi,%rdi
je     11d1 <neuralops_conv2d_scale_bias_bwd+0xd1>
mov    0x50(%rsp),%r10
mov    0x48(%rsp),%r11
mov    0x40(%rsp),%r12
mov    0x38(%rsp),%r14
xor    %ebx,%ebx
xor    %r15d,%r15d
jmp    113d <neuralops_conv2d_scale_bias_bwd+0x3d>
xchg   %ax,%ax
add    $0x1,%r15
cmp    %rdi,%r15
je     11d1 <neuralops_conv2d_scale_bias_bwd+0xd1>
test   %rcx,%rcx
je     1130 <neuralops_conv2d_scale_bias_bwd+0x30>
xor    %eax,%eax
jmp    1159 <neuralops_conv2d_scale_bias_bwd+0x59>
cs nopw 0x0(%rax,%rax,1)
add    $0x1,%rax
cmp    %rcx,%rax
je     1130 <neuralops_conv2d_scale_bias_bwd+0x30>
test   %rdx,%rdx
je     1150 <neuralops_conv2d_scale_bias_bwd+0x50>
movss  (%r9,%rax,4),%xmm0
xor    %r13d,%r13d
jmp    1179 <neuralops_conv2d_scale_bias_bwd+0x79>
nopl   0x0(%rax)
add    $0x1,%r13
cmp    %rdx,%r13
je     1150 <neuralops_conv2d_scale_bias_bwd+0x50>
mov    %rsi,%rbp
test   %rsi,%rsi
je     1170 <neuralops_conv2d_scale_bias_bwd+0x70>
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax,%rax,1)
movss  (%r14,%rbx,4),%xmm1
movss  (%r8,%rbx,4),%xmm2
mulss  %xmm1,%xmm2
addss  (%r12,%rax,4),%xmm2
movss  %xmm2,(%r12,%rax,4)
movaps %xmm0,%xmm2
mulss  %xmm1,%xmm2
addss  (%r11,%rax,4),%xmm1
movss  %xmm1,(%r11,%rax,4)
movss  %xmm2,(%r10,%rbx,4)
add    $0x1,%rbx
add    $0xffffffffffffffff,%rbp
jne    1190 <neuralops_conv2d_scale_bias_bwd+0x90>
jmp    1170 <neuralops_conv2d_scale_bias_bwd+0x70>
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

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

LLVM IR (-O2):
define void @neuralops_conv2d_scale_bias_bwd(i64 %0, i64 %1, i64 %2, i64 %3, float* nocapture %4, float* nocapture %5, float* nocapture %6, float* nocapture %7, float* nocapture %8, float* nocapture writeonly %9) {
%11 = icmp eq i64 %0, 0
br i1 %11, label %19, label %12
12: ; preds = %10
%13 = icmp eq i64 %3, 0
%14 = icmp eq i64 %2, 0
%15 = icmp eq i64 %1, 0
br label %16
16: ; preds = %12, %20
%17 = phi i64 [ 0, %12 ], [ %21, %20 ]
%18 = phi i64 [ 0, %12 ], [ %22, %20 ]
br i1 %13, label %20, label %24
19: ; preds = %20, %10
ret void
20: ; preds = %35, %16
%21 = phi i64 [ %17, %16 ], [ %36, %35 ]
%22 = add nuw i64 %18, 1
%23 = icmp eq i64 %22, %0
br i1 %23, label %19, label %16
24: ; preds = %16, %35
%25 = phi i64 [ %36, %35 ], [ %17, %16 ]
%26 = phi i64 [ %37, %35 ], [ 0, %16 ]
%27 = getelementptr inbounds float, float* %5, i64 %26
%28 = load float, float* %27
br i1 %14, label %35, label %29
29: ; preds = %24
%30 = getelementptr inbounds float, float* %7, i64 %26
%31 = getelementptr inbounds float, float* %8, i64 %26
br label %32
32: ; preds = %29, %39
%33 = phi i64 [ %25, %29 ], [ %40, %39 ]
%34 = phi i64 [ 0, %29 ], [ %41, %39 ]
br i1 %15, label %39, label %43
35: ; preds = %39, %24
%36 = phi i64 [ %25, %24 ], [ %40, %39 ]
%37 = add nuw i64 %26, 1
%38 = icmp eq i64 %37, %3
br i1 %38, label %20, label %24
39: ; preds = %43, %32
%40 = phi i64 [ %33, %32 ], [ %56, %43 ]
%41 = add nuw i64 %34, 1
%42 = icmp eq i64 %41, %2
br i1 %42, label %35, label %32
43: ; preds = %32, %43
%44 = phi i64 [ %56, %43 ], [ %33, %32 ]
%45 = phi i64 [ %57, %43 ], [ 0, %32 ]
%46 = getelementptr inbounds float, float* %6, i64 %44
%47 = load float, float* %46
%48 = getelementptr inbounds float, float* %4, i64 %44
%49 = load float, float* %48
%50 = load float, float* %30
%51 = tail call float @llvm.fmuladd.f32(float %47, float %49, float %50)
store float %51, float* %30
%52 = load float, float* %31
%53 = fadd float %47, %52
store float %53, float* %31
%54 = fmul float %28, %47
%55 = getelementptr inbounds float, float* %9, i64 %44
store float %54, float* %55
%56 = add i64 %44, 1
%57 = add nuw i64 %45, 1
%58 = icmp eq i64 %57, %1
br i1 %58, label %39, label %43
}