#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 */
scalar_t__ sumsq_sse2_assist(short *, int);
long long sumsq_sse2(signed short *in, int cnt) {
  long long sum = 0;
  while (((int)in & 15) != 0 && cnt != 0) {
    sum += (long)in[0] * in[0];
    in++;
    cnt--;
  }
  sum += sumsq_sse2_assist(in, cnt);
  in += cnt & ~7;
  cnt &= 7;
  while (cnt != 0) {
    sum += (long)in[0] * in[0];
    in++;
    cnt--;
  }
  return sum;
}


x86 assembly (clang, -O0):
<sumsq_sse2>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %rdi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
movq   $0x0,-0x18(%rbp)
mov    -0x8(%rbp),%rax
mov    %eax,%ecx
and    $0xf,%ecx
xor    %eax,%eax
cmp    $0x0,%ecx
mov    %al,-0x19(%rbp)
je     1148 <sumsq_sse2+0x38>
cmpl   $0x0,-0xc(%rbp)
setne  %al
mov    %al,-0x19(%rbp)
mov    -0x19(%rbp),%al
test   $0x1,%al
jne    1158 <sumsq_sse2+0x48>
jmp    118e <sumsq_sse2+0x7e>
mov    -0x8(%rbp),%rax
movswq (%rax),%rax
mov    -0x8(%rbp),%rcx
movswq (%rcx),%rcx
imul   %rcx,%rax
add    -0x18(%rbp),%rax
mov    %rax,-0x18(%rbp)
mov    -0x8(%rbp),%rax
add    $0x2,%rax
mov    %rax,-0x8(%rbp)
mov    -0xc(%rbp),%eax
add    $0xffffffff,%eax
mov    %eax,-0xc(%rbp)
jmp    1127 <sumsq_sse2+0x17>
mov    -0x8(%rbp),%rdi
mov    -0xc(%rbp),%esi
call   1030 <sumsq_sse2_assist@plt>
add    -0x18(%rbp),%rax
mov    %rax,-0x18(%rbp)
mov    -0xc(%rbp),%ecx
and    $0xfffffff8,%ecx
mov    -0x8(%rbp),%rax
movslq %ecx,%rcx
shl    $0x1,%rcx
add    %rcx,%rax
mov    %rax,-0x8(%rbp)
mov    -0xc(%rbp),%eax
and    $0x7,%eax
mov    %eax,-0xc(%rbp)
cmpl   $0x0,-0xc(%rbp)
je     1203 <sumsq_sse2+0xf3>
mov    -0x8(%rbp),%rax
movswq (%rax),%rax
mov    -0x8(%rbp),%rcx
movswq (%rcx),%rcx
imul   %rcx,%rax
add    -0x18(%rbp),%rax
mov    %rax,-0x18(%rbp)
mov    -0x8(%rbp),%rax
add    $0x2,%rax
mov    %rax,-0x8(%rbp)
mov    -0xc(%rbp),%eax
add    $0xffffffff,%eax
mov    %eax,-0xc(%rbp)
jmp    11c3 <sumsq_sse2+0xb3>
mov    -0x18(%rbp),%rax
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<sumsq_sse2>:
push   %rbp
push   %r14
push   %rbx
mov    %rdi,%r14
xor    %ebx,%ebx
test   $0xf,%r14b
je     1156 <sumsq_sse2+0x46>
test   %esi,%esi
je     1156 <sumsq_sse2+0x46>
lea    0x2(%r14),%rax
xor    %ebx,%ebx
nopl   0x0(%rax)
mov    %rbx,%rcx
movswq (%r14),%rbx
imul   %rbx,%rbx
add    %rcx,%rbx
add    $0x2,%r14
lea    -0x1(%rsi),%ebp
test   $0xf,%al
je     1158 <sumsq_sse2+0x48>
add    $0x2,%rax
cmp    $0x1,%esi
mov    %ebp,%esi
jne    1130 <sumsq_sse2+0x20>
jmp    1158 <sumsq_sse2+0x48>
mov    %esi,%ebp
mov    %r14,%rdi
mov    %ebp,%esi
call   1030 <sumsq_sse2_assist@plt>
add    %rbx,%rax
mov    %ebp,%ecx
and    $0x7,%ecx
je     11ea <sumsq_sse2+0xda>
and    $0xfffffff8,%ebp
movslq %ebp,%rdx
movswq (%r14,%rdx,2),%rsi
imul   %rsi,%rsi
add    %rsi,%rax
cmp    $0x1,%ecx
je     11ea <sumsq_sse2+0xda>
movswq 0x2(%r14,%rdx,2),%rsi
imul   %rsi,%rsi
add    %rsi,%rax
cmp    $0x2,%ecx
je     11ea <sumsq_sse2+0xda>
movswq 0x4(%r14,%rdx,2),%rsi
imul   %rsi,%rsi
add    %rsi,%rax
cmp    $0x3,%ecx
je     11ea <sumsq_sse2+0xda>
movswq 0x6(%r14,%rdx,2),%rsi
imul   %rsi,%rsi
add    %rsi,%rax
cmp    $0x4,%ecx
je     11ea <sumsq_sse2+0xda>
movswq 0x8(%r14,%rdx,2),%rsi
imul   %rsi,%rsi
add    %rsi,%rax
cmp    $0x5,%ecx
je     11ea <sumsq_sse2+0xda>
movswq 0xa(%r14,%rdx,2),%rsi
imul   %rsi,%rsi
add    %rsi,%rax
cmp    $0x6,%ecx
je     11ea <sumsq_sse2+0xda>
movswq 0xc(%r14,%rdx,2),%rcx
imul   %rcx,%rcx
add    %rcx,%rax
pop    %rbx
pop    %r14
pop    %rbp
ret

LLVM IR (-O0):
define i64 @sumsq_sse2(i16* %0, i32 %1) {
%3 = alloca i16*
%4 = alloca i32
%5 = alloca i64
store i16* %0, i16** %3
store i32 %1, i32* %4
store i64 0, i64* %5
br label %6
6: ; preds = %16, %2
%7 = load i16*, i16** %3
%8 = ptrtoint i16* %7 to i32
%9 = and i32 %8, 15
%10 = icmp ne i32 %9, 0
br i1 %10, label %11, label %14
11: ; preds = %6
%12 = load i32, i32* %4
%13 = icmp ne i32 %12, 0
br label %14
14: ; preds = %11, %6
%15 = phi i1 [ false, %6 ], [ %13, %11 ]
br i1 %15, label %16, label %32
16: ; preds = %14
%17 = load i16*, i16** %3
%18 = getelementptr inbounds i16, i16* %17, i64 0
%19 = load i16, i16* %18
%20 = sext i16 %19 to i64
%21 = load i16*, i16** %3
%22 = getelementptr inbounds i16, i16* %21, i64 0
%23 = load i16, i16* %22
%24 = sext i16 %23 to i64
%25 = mul nsw i64 %20, %24
%26 = load i64, i64* %5
%27 = add nsw i64 %26, %25
store i64 %27, i64* %5
%28 = load i16*, i16** %3
%29 = getelementptr inbounds i16, i16* %28, i32 1
store i16* %29, i16** %3
%30 = load i32, i32* %4
%31 = add nsw i32 %30, -1
store i32 %31, i32* %4
br label %6
32: ; preds = %14
%33 = load i16*, i16** %3
%34 = load i32, i32* %4
%35 = call i64 @sumsq_sse2_assist(i16* %33, i32 %34)
%36 = load i64, i64* %5
%37 = add nsw i64 %36, %35
store i64 %37, i64* %5
%38 = load i32, i32* %4
%39 = and i32 %38, -8
%40 = load i16*, i16** %3
%41 = sext i32 %39 to i64
%42 = getelementptr inbounds i16, i16* %40, i64 %41
store i16* %42, i16** %3
%43 = load i32, i32* %4
%44 = and i32 %43, 7
store i32 %44, i32* %4
br label %45
45: ; preds = %48, %32
%46 = load i32, i32* %4
%47 = icmp ne i32 %46, 0
br i1 %47, label %48, label %64
48: ; preds = %45
%49 = load i16*, i16** %3
%50 = getelementptr inbounds i16, i16* %49, i64 0
%51 = load i16, i16* %50
%52 = sext i16 %51 to i64
%53 = load i16*, i16** %3
%54 = getelementptr inbounds i16, i16* %53, i64 0
%55 = load i16, i16* %54
%56 = sext i16 %55 to i64
%57 = mul nsw i64 %52, %56
%58 = load i64, i64* %5
%59 = add nsw i64 %58, %57
store i64 %59, i64* %5
%60 = load i16*, i16** %3
%61 = getelementptr inbounds i16, i16* %60, i32 1
store i16* %61, i16** %3
%62 = load i32, i32* %4
%63 = add nsw i32 %62, -1
store i32 %63, i32* %4
br label %45
64: ; preds = %45
%65 = load i64, i64* %5
ret i64 %65
}

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