void __attribute__((no, noclone)) foo(int n, int *a, int workgroup_size) {
  int i;
#pragma omp target
#pragma omp teams thread_limit(workgroup_size)
#pragma omp distribute parallel for shared(a) firstprivate(n) private(i)
  for (i = 0; i < n; i++)
    a[i]++;
}


x86 assembly (clang, -O0):
<foo>:
push   %rbp
mov    %rsp,%rbp
mov    %edi,-0x4(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
movl   $0x0,-0x18(%rbp)
mov    -0x18(%rbp),%eax
cmp    -0x4(%rbp),%eax
jge    1140 <foo+0x40>
mov    -0x10(%rbp),%rax
movslq -0x18(%rbp),%rcx
mov    (%rax,%rcx,4),%edx
add    $0x1,%edx
mov    %edx,(%rax,%rcx,4)
mov    -0x18(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x18(%rbp)
jmp    1115 <foo+0x15>
pop    %rbp
ret

x86 assembly (clang, -O2):
<foo>:
test   %edi,%edi
jle    11cd <foo+0xcd>
mov    %edi,%eax
cmp    $0x8,%edi
jae    1116 <foo+0x16>
xor    %ecx,%ecx
jmp    11c0 <foo+0xc0>
mov    %eax,%ecx
and    $0xfffffff8,%ecx
lea    -0x8(%rcx),%rdx
mov    %rdx,%r8
shr    $0x3,%r8
add    $0x1,%r8
test   %rdx,%rdx
je     11ce <foo+0xce>
mov    %r8,%rdi
and    $0xfffffffffffffffe,%rdi
xor    %edx,%edx
pcmpeqd %xmm0,%xmm0
movdqu (%rsi,%rdx,4),%xmm1
movdqu 0x10(%rsi,%rdx,4),%xmm2
movdqu 0x20(%rsi,%rdx,4),%xmm3
movdqu 0x30(%rsi,%rdx,4),%xmm4
psubd  %xmm0,%xmm1
psubd  %xmm0,%xmm2
movdqu %xmm1,(%rsi,%rdx,4)
movdqu %xmm2,0x10(%rsi,%rdx,4)
psubd  %xmm0,%xmm3
psubd  %xmm0,%xmm4
movdqu %xmm3,0x20(%rsi,%rdx,4)
movdqu %xmm4,0x30(%rsi,%rdx,4)
add    $0x10,%rdx
add    $0xfffffffffffffffe,%rdi
jne    1140 <foo+0x40>
test   $0x1,%r8b
je     11b0 <foo+0xb0>
movdqu (%rsi,%rdx,4),%xmm0
movdqu 0x10(%rsi,%rdx,4),%xmm1
pcmpeqd %xmm2,%xmm2
psubd  %xmm2,%xmm0
psubd  %xmm2,%xmm1
movdqu %xmm0,(%rsi,%rdx,4)
movdqu %xmm1,0x10(%rsi,%rdx,4)
cmp    %rax,%rcx
je     11cd <foo+0xcd>
cs nopw 0x0(%rax,%rax,1)
nop
addl   $0x1,(%rsi,%rcx,4)
add    $0x1,%rcx
cmp    %rcx,%rax
jne    11c0 <foo+0xc0>
ret
xor    %edx,%edx
test   $0x1,%r8b
jne    118e <foo+0x8e>
jmp    11b0 <foo+0xb0>

LLVM IR (-O0):
define void @foo(i32 %0, i32* %1, i32 %2) {
%4 = alloca i32
%5 = alloca i32*
%6 = alloca i32
%7 = alloca i32
store i32 %0, i32* %4
store i32* %1, i32** %5
store i32 %2, i32* %6
store i32 0, i32* %7
br label %8
8: ; preds = %19, %3
%9 = load i32, i32* %7
%10 = load i32, i32* %4
%11 = icmp slt i32 %9, %10
br i1 %11, label %12, label %22
12: ; preds = %8
%13 = load i32*, i32** %5
%14 = load i32, i32* %7
%15 = sext i32 %14 to i64
%16 = getelementptr inbounds i32, i32* %13, i64 %15
%17 = load i32, i32* %16
%18 = add nsw i32 %17, 1
store i32 %18, i32* %16
br label %19
19: ; preds = %12
%20 = load i32, i32* %7
%21 = add nsw i32 %20, 1
store i32 %21, i32* %7
br label %8
22: ; preds = %8
ret void
}

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