#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_logistic_bwd(size_t batch_sz, size_t dim, const float *out_buf,
                            const float *out_grad, float *in_grad) {
#pragma omp parallel for
  for (size_t p = 0; p < batch_sz * dim; p += 1) {
    float y = out_buf[p];
    float dy = out_grad[p];
    in_grad[p] = y * (1.0f - y) * dy;
  }
}


x86 assembly (clang, -O0):
<neuralops_logistic_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)
movq   $0x0,-0x30(%rbp)
mov    -0x30(%rbp),%rax
mov    -0x8(%rbp),%rcx
imul   -0x10(%rbp),%rcx
cmp    %rcx,%rax
jae    1193 <neuralops_logistic_bwd+0x93>
mov    -0x18(%rbp),%rax
mov    -0x30(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm0
movss  %xmm0,-0x34(%rbp)
mov    -0x20(%rbp),%rax
mov    -0x30(%rbp),%rcx
movss  (%rax,%rcx,4),%xmm0
movss  %xmm0,-0x38(%rbp)
movss  -0x34(%rbp),%xmm0
movss  0xe99(%rip),%xmm1
subss  -0x34(%rbp),%xmm1
mulss  %xmm1,%xmm0
mulss  -0x38(%rbp),%xmm0
mov    -0x28(%rbp),%rax
mov    -0x30(%rbp),%rcx
movss  %xmm0,(%rax,%rcx,4)
mov    -0x30(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x30(%rbp)
jmp    1120 <neuralops_logistic_bwd+0x20>
pop    %rbp
ret

x86 assembly (clang, -O2):
<neuralops_logistic_bwd>:
imul   %rsi,%rdi
test   %rdi,%rdi
je     12df <neuralops_logistic_bwd+0x1df>
cmp    $0x8,%rdi
jae    11a7 <neuralops_logistic_bwd+0xa7>
xor    %eax,%eax
mov    %rax,%rsi
or     $0x1,%rsi
test   $0x1,%dil
je     1149 <neuralops_logistic_bwd+0x49>
movss  (%rdx,%rax,4),%xmm0
movss  0xedd(%rip),%xmm1
subss  %xmm0,%xmm1
mulss  %xmm0,%xmm1
mulss  (%rcx,%rax,4),%xmm1
movss  %xmm1,(%r8,%rax,4)
mov    %rsi,%rax
cmp    %rsi,%rdi
je     12df <neuralops_logistic_bwd+0x1df>
movss  0xeb6(%rip),%xmm0
nopw   0x0(%rax,%rax,1)
movss  (%rdx,%rax,4),%xmm1
movaps %xmm0,%xmm2
subss  %xmm1,%xmm2
mulss  %xmm1,%xmm2
mulss  (%rcx,%rax,4),%xmm2
movss  %xmm2,(%r8,%rax,4)
movss  0x4(%rdx,%rax,4),%xmm1
movaps %xmm0,%xmm2
subss  %xmm1,%xmm2
mulss  %xmm1,%xmm2
mulss  0x4(%rcx,%rax,4),%xmm2
movss  %xmm2,0x4(%r8,%rax,4)
add    $0x2,%rax
cmp    %rax,%rdi
jne    1160 <neuralops_logistic_bwd+0x60>
jmp    12df <neuralops_logistic_bwd+0x1df>
lea    (%r8,%rdi,4),%rax
lea    (%rdx,%rdi,4),%rsi
cmp    %r8,%rsi
seta   %r10b
lea    (%rcx,%rdi,4),%rsi
cmp    %rdx,%rax
seta   %r11b
cmp    %r8,%rsi
seta   %sil
cmp    %rcx,%rax
seta   %r9b
xor    %eax,%eax
test   %r11b,%r10b
jne    1119 <neuralops_logistic_bwd+0x19>
and    %r9b,%sil
jne    1119 <neuralops_logistic_bwd+0x19>
mov    %rdi,%rax
and    $0xfffffffffffffff8,%rax
lea    -0x8(%rax),%rsi
mov    %rsi,%r9
shr    $0x3,%r9
add    $0x1,%r9
test   %rsi,%rsi
je     12e0 <neuralops_logistic_bwd+0x1e0>
mov    %r9,%r10
and    $0xfffffffffffffffe,%r10
xor    %esi,%esi
movaps 0xdee(%rip),%xmm0
cs nopw 0x0(%rax,%rax,1)
nopl   0x0(%rax)
movups (%rdx,%rsi,4),%xmm1
movups 0x10(%rdx,%rsi,4),%xmm2
movups (%rcx,%rsi,4),%xmm3
movups 0x10(%rcx,%rsi,4),%xmm4
movaps %xmm0,%xmm5
subps  %xmm1,%xmm5
movaps %xmm0,%xmm6
subps  %xmm2,%xmm6
mulps  %xmm1,%xmm5
mulps  %xmm3,%xmm5
mulps  %xmm2,%xmm6
mulps  %xmm4,%xmm6
movups %xmm5,(%r8,%rsi,4)
movups %xmm6,0x10(%r8,%rsi,4)
movups 0x20(%rdx,%rsi,4),%xmm1
movups 0x30(%rdx,%rsi,4),%xmm2
movups 0x20(%rcx,%rsi,4),%xmm3
movups 0x30(%rcx,%rsi,4),%xmm4
movaps %xmm0,%xmm5
subps  %xmm1,%xmm5
movaps %xmm0,%xmm6
subps  %xmm2,%xmm6
mulps  %xmm1,%xmm5
mulps  %xmm3,%xmm5
mulps  %xmm2,%xmm6
mulps  %xmm4,%xmm6
movups %xmm5,0x20(%r8,%rsi,4)
movups %xmm6,0x30(%r8,%rsi,4)
add    $0x10,%rsi
add    $0xfffffffffffffffe,%r10
jne    1220 <neuralops_logistic_bwd+0x120>
test   $0x1,%r9b
je     12d6 <neuralops_logistic_bwd+0x1d6>
movups (%rdx,%rsi,4),%xmm0
movups 0x10(%rdx,%rsi,4),%xmm1
movups (%rcx,%rsi,4),%xmm2
movups 0x10(%rcx,%rsi,4),%xmm3
movaps 0xd4a(%rip),%xmm4
movaps %xmm4,%xmm5
subps  %xmm0,%xmm5
subps  %xmm1,%xmm4
mulps  %xmm0,%xmm5
mulps  %xmm2,%xmm5
mulps  %xmm1,%xmm4
mulps  %xmm3,%xmm4
movups %xmm5,(%r8,%rsi,4)
movups %xmm4,0x10(%r8,%rsi,4)
cmp    %rax,%rdi
jne    1119 <neuralops_logistic_bwd+0x19>
ret
xor    %esi,%esi
test   $0x1,%r9b
jne    129d <neuralops_logistic_bwd+0x19d>
jmp    12d6 <neuralops_logistic_bwd+0x1d6>

LLVM IR (-O0):
define void @neuralops_logistic_bwd(i64 %0, i64 %1, float* %2, float* %3, float* %4) {
%6 = alloca i64
%7 = alloca i64
%8 = alloca float*
%9 = alloca float*
%10 = alloca float*
%11 = alloca i64
%12 = alloca float
%13 = alloca float
store i64 %0, i64* %6
store i64 %1, i64* %7
store float* %2, float** %8
store float* %3, float** %9
store float* %4, float** %10
store i64 0, i64* %11
br label %14
14: ; preds = %38, %5
%15 = load i64, i64* %11
%16 = load i64, i64* %6
%17 = load i64, i64* %7
%18 = mul i64 %16, %17
%19 = icmp ult i64 %15, %18
br i1 %19, label %20, label %41
20: ; preds = %14
%21 = load float*, float** %8
%22 = load i64, i64* %11
%23 = getelementptr inbounds float, float* %21, i64 %22
%24 = load float, float* %23
store float %24, float* %12
%25 = load float*, float** %9
%26 = load i64, i64* %11
%27 = getelementptr inbounds float, float* %25, i64 %26
%28 = load float, float* %27
store float %28, float* %13
%29 = load float, float* %12
%30 = load float, float* %12
%31 = fsub float 1.000000e+00, %30
%32 = fmul float %29, %31
%33 = load float, float* %13
%34 = fmul float %32, %33
%35 = load float*, float** %10
%36 = load i64, i64* %11
%37 = getelementptr inbounds float, float* %35, i64 %36
store float %34, float* %37
br label %38
38: ; preds = %20
%39 = load i64, i64* %11
%40 = add i64 %39, 1
store i64 %40, i64* %11
br label %14
41: ; preds = %14
ret void
}

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