#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 */
int *primes_under(int /*<<< orphan*/ *, int, size_t *);
int sqrt(int);
int *get_prime_squares(int max_triangle_row, size_t *count) {
  int limit = ((2) > (sqrt(max_triangle_row)) ? (2) : (sqrt(max_triangle_row)));
  int *prime_squares = primes_under(NULL, limit + 1, count);
  for (size_t i = 0; i < *count; i++)
    prime_squares[i] *= prime_squares[i];
  return prime_squares;
}


x86 assembly (clang, -O0):
<get_prime_squares>:
push   %rbp
mov    %rsp,%rbp
sub    $0x30,%rsp
mov    %edi,-0x4(%rbp)
mov    %rsi,-0x10(%rbp)
mov    -0x4(%rbp),%edi
call   1040 <sqrt@plt>
mov    %eax,%ecx
mov    $0x2,%eax
cmp    %ecx,%eax
jle    1153 <get_prime_squares+0x33>
mov    $0x2,%eax
mov    %eax,-0x2c(%rbp)
jmp    115e <get_prime_squares+0x3e>
mov    -0x4(%rbp),%edi
call   1040 <sqrt@plt>
mov    %eax,-0x2c(%rbp)
mov    -0x2c(%rbp),%eax
mov    %eax,-0x14(%rbp)
mov    -0x14(%rbp),%esi
add    $0x1,%esi
mov    -0x10(%rbp),%rdx
xor    %eax,%eax
mov    %eax,%edi
call   1030 <primes_under@plt>
mov    %rax,-0x20(%rbp)
movq   $0x0,-0x28(%rbp)
mov    -0x28(%rbp),%rax
mov    -0x10(%rbp),%rcx
cmp    (%rcx),%rax
jae    11bf <get_prime_squares+0x9f>
mov    -0x20(%rbp),%rax
mov    -0x28(%rbp),%rcx
mov    (%rax,%rcx,4),%edx
mov    -0x20(%rbp),%rax
mov    -0x28(%rbp),%rcx
imul   (%rax,%rcx,4),%edx
mov    %edx,(%rax,%rcx,4)
mov    -0x28(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x28(%rbp)
jmp    1183 <get_prime_squares+0x63>
mov    -0x20(%rbp),%rax
add    $0x30,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<get_prime_squares>:
push   %rbp
push   %rbx
push   %rax
mov    %rsi,%rbx
mov    %edi,%ebp
call   1040 <sqrt@plt>
mov    $0x3,%esi
cmp    $0x2,%eax
jl     1143 <get_prime_squares+0x23>
mov    %ebp,%edi
call   1040 <sqrt@plt>
mov    %eax,%esi
add    $0x1,%esi
xor    %edi,%edi
mov    %rbx,%rdx
call   1030 <primes_under@plt>
mov    (%rbx),%rcx
test   %rcx,%rcx
je     12a8 <get_prime_squares+0x188>
cmp    $0x8,%rcx
jae    1166 <get_prime_squares+0x46>
xor    %edx,%edx
jmp    1296 <get_prime_squares+0x176>
mov    %rcx,%rdx
and    $0xfffffffffffffff8,%rdx
lea    -0x8(%rdx),%rsi
mov    %rsi,%rbp
shr    $0x3,%rbp
add    $0x1,%rbp
test   %rsi,%rsi
je     1240 <get_prime_squares+0x120>
mov    %rbp,%rdi
and    $0xfffffffffffffffe,%rdi
xor    %esi,%esi
xchg   %ax,%ax
movdqu (%rax,%rsi,4),%xmm1
movdqu 0x10(%rax,%rsi,4),%xmm2
movdqu 0x20(%rax,%rsi,4),%xmm3
movdqu 0x30(%rax,%rsi,4),%xmm0
pshufd $0xf5,%xmm1,%xmm4
pmuludq %xmm1,%xmm1
pshufd $0xe8,%xmm1,%xmm1
pmuludq %xmm4,%xmm4
pshufd $0xe8,%xmm4,%xmm4
punpckldq %xmm4,%xmm1
pshufd $0xf5,%xmm2,%xmm4
pmuludq %xmm2,%xmm2
pshufd $0xe8,%xmm2,%xmm2
pmuludq %xmm4,%xmm4
pshufd $0xe8,%xmm4,%xmm4
punpckldq %xmm4,%xmm2
movdqu %xmm1,(%rax,%rsi,4)
movdqu %xmm2,0x10(%rax,%rsi,4)
pshufd $0xf5,%xmm3,%xmm1
pmuludq %xmm3,%xmm3
pshufd $0xe8,%xmm3,%xmm2
pmuludq %xmm1,%xmm1
pshufd $0xe8,%xmm1,%xmm1
punpckldq %xmm1,%xmm2
pshufd $0xf5,%xmm0,%xmm1
pmuludq %xmm0,%xmm0
pshufd $0xe8,%xmm0,%xmm0
pmuludq %xmm1,%xmm1
pshufd $0xe8,%xmm1,%xmm1
punpckldq %xmm1,%xmm0
movdqu %xmm2,0x20(%rax,%rsi,4)
movdqu %xmm0,0x30(%rax,%rsi,4)
add    $0x10,%rsi
add    $0xfffffffffffffffe,%rdi
jne    1190 <get_prime_squares+0x70>
test   $0x1,%bpl
jne    1248 <get_prime_squares+0x128>
jmp    12a3 <get_prime_squares+0x183>
xor    %esi,%esi
test   $0x1,%bpl
je     12a3 <get_prime_squares+0x183>
movdqu (%rax,%rsi,4),%xmm0
movdqu 0x10(%rax,%rsi,4),%xmm1
pshufd $0xf5,%xmm0,%xmm2
pmuludq %xmm0,%xmm0
pshufd $0xe8,%xmm0,%xmm0
pmuludq %xmm2,%xmm2
pshufd $0xe8,%xmm2,%xmm2
punpckldq %xmm2,%xmm0
pshufd $0xf5,%xmm1,%xmm2
pmuludq %xmm1,%xmm1
pshufd $0xe8,%xmm1,%xmm1
pmuludq %xmm2,%xmm2
pshufd $0xe8,%xmm2,%xmm2
punpckldq %xmm2,%xmm1
movdqu %xmm0,(%rax,%rsi,4)
movdqu %xmm1,0x10(%rax,%rsi,4)
jmp    12a3 <get_prime_squares+0x183>
mov    (%rax,%rdx,4),%esi
imul   %esi,%esi
mov    %esi,(%rax,%rdx,4)
add    $0x1,%rdx
cmp    %rdx,%rcx
jne    1296 <get_prime_squares+0x176>
add    $0x8,%rsp
pop    %rbx
pop    %rbp
ret

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

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