#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 int uint16_t;

/* Variables and functions */
void lwe_key_round(uint16_t *vec, const size_t length, const int b) {
  size_t i;
  uint16_t negmask = ~((1 << b) - 1);
  uint16_t half = b > 0 ? 1 << (b - 1) : 0;
  for (i = 0; i < length; i++) {
    vec[i] = (vec[i] + half) & negmask;
  }
}


x86 assembly (clang, -O0):
<lwe_key_round>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
mov    -0x14(%rbp),%ecx
mov    $0x1,%eax
shl    %cl,%eax
sub    $0x1,%eax
xor    $0xffffffff,%eax
mov    %eax,-0x24(%rbp)
cmpl   $0x0,-0x14(%rbp)
jle    1141 <lwe_key_round+0x41>
mov    -0x14(%rbp),%ecx
sub    $0x1,%ecx
mov    $0x1,%eax
shl    %cl,%eax
mov    %eax,-0x2c(%rbp)
jmp    114b <lwe_key_round+0x4b>
xor    %eax,%eax
mov    %eax,-0x2c(%rbp)
jmp    114b <lwe_key_round+0x4b>
mov    -0x2c(%rbp),%eax
mov    %eax,-0x28(%rbp)
movq   $0x0,-0x20(%rbp)
mov    -0x20(%rbp),%rax
cmp    -0x10(%rbp),%rax
jae    1194 <lwe_key_round+0x94>
mov    -0x8(%rbp),%rax
mov    -0x20(%rbp),%rcx
mov    (%rax,%rcx,4),%edx
add    -0x28(%rbp),%edx
and    -0x24(%rbp),%edx
mov    -0x8(%rbp),%rax
mov    -0x20(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x20(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x20(%rbp)
jmp    1159 <lwe_key_round+0x59>
pop    %rbp
ret

x86 assembly (clang, -O2):
<lwe_key_round>:
mov    $0xffffffff,%r8d
mov    %edx,%ecx
shl    %cl,%r8d
lea    -0x1(%rdx),%ecx
mov    $0x1,%eax
shl    %cl,%eax
xor    %r10d,%r10d
test   %edx,%edx
cmovg  %eax,%r10d
test   %rsi,%rsi
je     1215 <lwe_key_round+0x115>
cmp    $0x8,%rsi
jae    1134 <lwe_key_round+0x34>
xor    %edx,%edx
jmp    1200 <lwe_key_round+0x100>
mov    %rsi,%rdx
and    $0xfffffffffffffff8,%rdx
movd   %r10d,%xmm0
pshufd $0x0,%xmm0,%xmm1
movd   %r8d,%xmm0
pshufd $0x0,%xmm0,%xmm0
lea    -0x8(%rdx),%rax
mov    %rax,%r9
shr    $0x3,%r9
add    $0x1,%r9
test   %rax,%rax
je     1216 <lwe_key_round+0x116>
mov    %r9,%rcx
and    $0xfffffffffffffffe,%rcx
xor    %eax,%eax
movdqu (%rdi,%rax,4),%xmm2
movdqu 0x10(%rdi,%rax,4),%xmm3
movdqu 0x20(%rdi,%rax,4),%xmm4
movdqu 0x30(%rdi,%rax,4),%xmm5
paddd  %xmm1,%xmm2
paddd  %xmm1,%xmm3
pand   %xmm0,%xmm2
pand   %xmm0,%xmm3
movdqu %xmm2,(%rdi,%rax,4)
movdqu %xmm3,0x10(%rdi,%rax,4)
paddd  %xmm1,%xmm4
paddd  %xmm1,%xmm5
pand   %xmm0,%xmm4
pand   %xmm0,%xmm5
movdqu %xmm4,0x20(%rdi,%rax,4)
movdqu %xmm5,0x30(%rdi,%rax,4)
add    $0x10,%rax
add    $0xfffffffffffffffe,%rcx
jne    1170 <lwe_key_round+0x70>
test   $0x1,%r9b
je     11f4 <lwe_key_round+0xf4>
movdqu (%rdi,%rax,4),%xmm2
movdqu 0x10(%rdi,%rax,4),%xmm3
paddd  %xmm1,%xmm2
paddd  %xmm1,%xmm3
pand   %xmm0,%xmm2
pand   %xmm0,%xmm3
movdqu %xmm2,(%rdi,%rax,4)
movdqu %xmm3,0x10(%rdi,%rax,4)
cmp    %rsi,%rdx
je     1215 <lwe_key_round+0x115>
nopl   0x0(%rax)
mov    (%rdi,%rdx,4),%eax
add    %r10d,%eax
and    %r8d,%eax
mov    %eax,(%rdi,%rdx,4)
add    $0x1,%rdx
cmp    %rdx,%rsi
jne    1200 <lwe_key_round+0x100>
ret
xor    %eax,%eax
test   $0x1,%r9b
jne    11ce <lwe_key_round+0xce>
jmp    11f4 <lwe_key_round+0xf4>

LLVM IR (-O0):
define void @lwe_key_round(i32* %0, i64 %1, i32 %2) {
%4 = alloca i32*
%5 = alloca i64
%6 = alloca i32
%7 = alloca i64
%8 = alloca i32
%9 = alloca i32
store i32* %0, i32** %4
store i64 %1, i64* %5
store i32 %2, i32* %6
%10 = load i32, i32* %6
%11 = shl i32 1, %10
%12 = sub nsw i32 %11, 1
%13 = xor i32 %12, -1
store i32 %13, i32* %8
%14 = load i32, i32* %6
%15 = icmp sgt i32 %14, 0
br i1 %15, label %16, label %20
16: ; preds = %3
%17 = load i32, i32* %6
%18 = sub nsw i32 %17, 1
%19 = shl i32 1, %18
br label %21
20: ; preds = %3
br label %21
21: ; preds = %20, %16
%22 = phi i32 [ %19, %16 ], [ 0, %20 ]
store i32 %22, i32* %9
store i64 0, i64* %7
br label %23
23: ; preds = %39, %21
%24 = load i64, i64* %7
%25 = load i64, i64* %5
%26 = icmp ult i64 %24, %25
br i1 %26, label %27, label %42
27: ; preds = %23
%28 = load i32*, i32** %4
%29 = load i64, i64* %7
%30 = getelementptr inbounds i32, i32* %28, i64 %29
%31 = load i32, i32* %30
%32 = load i32, i32* %9
%33 = add nsw i32 %31, %32
%34 = load i32, i32* %8
%35 = and i32 %33, %34
%36 = load i32*, i32** %4
%37 = load i64, i64* %7
%38 = getelementptr inbounds i32, i32* %36, i64 %37
store i32 %35, i32* %38
br label %39
39: ; preds = %27
%40 = load i64, i64* %7
%41 = add i64 %40, 1
store i64 %41, i64* %7
br label %23
42: ; preds = %23
ret void
}

LLVM IR (-O2):
define void @lwe_key_round(i32* nocapture %0, i64 %1, i32 %2) {
%4 = shl nsw i32 -1, %2
%5 = icmp sgt i32 %2, 0
%6 = add nsw i32 %2, -1
%7 = shl nuw i32 1, %6
%8 = select i1 %5, i32 %7, i32 0
%9 = icmp eq i64 %1, 0
br i1 %9, label %88, label %10
10: ; preds = %3
%11 = icmp ult i64 %1, 8
br i1 %11, label %78, label %12
12: ; preds = %10
%13 = and i64 %1, -8
%14 = insertelement <4 x i32> poison, i32 %8, i64 0
%15 = shufflevector <4 x i32> %14, <4 x i32> poison, <4 x i32> zeroinitializer
%16 = insertelement <4 x i32> poison, i32 %8, i64 0
%17 = shufflevector <4 x i32> %16, <4 x i32> poison, <4 x i32> zeroinitializer
%18 = insertelement <4 x i32> poison, i32 %4, i64 0
%19 = shufflevector <4 x i32> %18, <4 x i32> poison, <4 x i32> zeroinitializer
%20 = insertelement <4 x i32> poison, i32 %4, i64 0
%21 = shufflevector <4 x i32> %20, <4 x i32> poison, <4 x i32> zeroinitializer
%22 = add i64 %13, -8
%23 = lshr exact i64 %22, 3
%24 = add nuw nsw i64 %23, 1
%25 = and i64 %24, 1
%26 = icmp eq i64 %22, 0
br i1 %26, label %60, label %27
27: ; preds = %12
%28 = and i64 %24, 4611686018427387902
br label %29
29: ; preds = %29, %27
%30 = phi i64 [ 0, %27 ], [ %57, %29 ]
%31 = phi i64 [ 0, %27 ], [ %58, %29 ]
%32 = getelementptr inbounds i32, i32* %0, i64 %30
%33 = bitcast i32* %32 to <4 x i32>*
%34 = load <4 x i32>, <4 x i32>* %33
%35 = getelementptr inbounds i32, i32* %32, i64 4
%36 = bitcast i32* %35 to <4 x i32>*
%37 = load <4 x i32>, <4 x i32>* %36
%38 = add nsw <4 x i32> %34, %15
%39 = add nsw <4 x i32> %37, %17
%40 = and <4 x i32> %38, %19
%41 = and <4 x i32> %39, %21
%42 = bitcast i32* %32 to <4 x i32>*
store <4 x i32> %40, <4 x i32>* %42
%43 = bitcast i32* %35 to <4 x i32>*
store <4 x i32> %41, <4 x i32>* %43
%44 = or i64 %30, 8
%45 = getelementptr inbounds i32, i32* %0, i64 %44
%46 = bitcast i32* %45 to <4 x i32>*
%47 = load <4 x i32>, <4 x i32>* %46
%48 = getelementptr inbounds i32, i32* %45, i64 4
%49 = bitcast i32* %48 to <4 x i32>*
%50 = load <4 x i32>, <4 x i32>* %49
%51 = add nsw <4 x i32> %47, %15
%52 = add nsw <4 x i32> %50, %17
%53 = and <4 x i32> %51, %19
%54 = and <4 x i32> %52, %21
%55 = bitcast i32* %45 to <4 x i32>*
store <4 x i32> %53, <4 x i32>* %55
%56 = bitcast i32* %48 to <4 x i32>*
store <4 x i32> %54, <4 x i32>* %56
%57 = add nuw i64 %30, 16
%58 = add i64 %31, 2
%59 = icmp eq i64 %58, %28
br i1 %59, label %60, label %29
60: ; preds = %29, %12
%61 = phi i64 [ 0, %12 ], [ %57, %29 ]
%62 = icmp eq i64 %25, 0
br i1 %62, label %76, label %63
63: ; preds = %60
%64 = getelementptr inbounds i32, i32* %0, i64 %61
%65 = bitcast i32* %64 to <4 x i32>*
%66 = load <4 x i32>, <4 x i32>* %65
%67 = getelementptr inbounds i32, i32* %64, i64 4
%68 = bitcast i32* %67 to <4 x i32>*
%69 = load <4 x i32>, <4 x i32>* %68
%70 = add nsw <4 x i32> %66, %15
%71 = add nsw <4 x i32> %69, %17
%72 = and <4 x i32> %70, %19
%73 = and <4 x i32> %71, %21
%74 = bitcast i32* %64 to <4 x i32>*
store <4 x i32> %72, <4 x i32>* %74
%75 = bitcast i32* %67 to <4 x i32>*
store <4 x i32> %73, <4 x i32>* %75
br label %76
76: ; preds = %60, %63
%77 = icmp eq i64 %13, %1
br i1 %77, label %88, label %78
78: ; preds = %10, %76
%79 = phi i64 [ 0, %10 ], [ %13, %76 ]
br label %80
80: ; preds = %78, %80
%81 = phi i64 [ %86, %80 ], [ %79, %78 ]
%82 = getelementptr inbounds i32, i32* %0, i64 %81
%83 = load i32, i32* %82
%84 = add nsw i32 %83, %8
%85 = and i32 %84, %4
store i32 %85, i32* %82
%86 = add nuw i64 %81, 1
%87 = icmp eq i64 %86, %1
br i1 %87, label %88, label %80
88: ; preds = %80, %76, %3
ret void
}