#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 **new_matrix2i(int, int);
int **new_diag_matrix2i(int *diag, int n) {
  int **mat = new_matrix2i(n, n);
  int i;
  for (i = 0; i < n; ++i) {
    mat[i][i] = diag[i];
  }
  return mat;
}


x86 assembly (clang, -O0):
<new_diag_matrix2i>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %rdi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
mov    -0xc(%rbp),%edi
mov    -0xc(%rbp),%esi
call   1030 <new_matrix2i@plt>
mov    %rax,-0x18(%rbp)
movl   $0x0,-0x1c(%rbp)
mov    -0x1c(%rbp),%eax
cmp    -0xc(%rbp),%eax
jge    116d <new_diag_matrix2i+0x5d>
mov    -0x8(%rbp),%rax
movslq -0x1c(%rbp),%rcx
mov    (%rax,%rcx,4),%edx
mov    -0x18(%rbp),%rax
movslq -0x1c(%rbp),%rcx
mov    (%rax,%rcx,8),%rax
movslq -0x1c(%rbp),%rcx
mov    %edx,(%rax,%rcx,4)
mov    -0x1c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x1c(%rbp)
jmp    1135 <new_diag_matrix2i+0x25>
mov    -0x18(%rbp),%rax
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<new_diag_matrix2i>:
push   %rbp
push   %rbx
push   %rax
mov    %esi,%ebp
mov    %rdi,%rbx
mov    %esi,%edi
call   1030 <new_matrix2i@plt>
test   %ebp,%ebp
jle    1194 <new_diag_matrix2i+0x84>
mov    %ebp,%esi
lea    -0x1(%rsi),%rdx
mov    %esi,%ecx
and    $0x3,%ecx
cmp    $0x3,%rdx
jae    1138 <new_diag_matrix2i+0x28>
xor    %edx,%edx
jmp    117b <new_diag_matrix2i+0x6b>
and    $0xfffffffc,%esi
xor    %edx,%edx
nopl   (%rax)
mov    (%rbx,%rdx,4),%edi
mov    (%rax,%rdx,8),%rbp
mov    %edi,0x0(%rbp,%rdx,4)
mov    0x4(%rbx,%rdx,4),%edi
mov    0x8(%rax,%rdx,8),%rbp
mov    %edi,0x4(%rbp,%rdx,4)
mov    0x8(%rbx,%rdx,4),%edi
mov    0x10(%rax,%rdx,8),%rbp
mov    %edi,0x8(%rbp,%rdx,4)
mov    0xc(%rbx,%rdx,4),%edi
mov    0x18(%rax,%rdx,8),%rbp
mov    %edi,0xc(%rbp,%rdx,4)
add    $0x4,%rdx
cmp    %rdx,%rsi
jne    1140 <new_diag_matrix2i+0x30>
test   %rcx,%rcx
je     1194 <new_diag_matrix2i+0x84>
mov    (%rbx,%rdx,4),%esi
mov    (%rax,%rdx,8),%rdi
mov    %esi,(%rdi,%rdx,4)
add    $0x1,%rdx
add    $0xffffffffffffffff,%rcx
jne    1180 <new_diag_matrix2i+0x70>
add    $0x8,%rsp
pop    %rbx
pop    %rbp
ret

LLVM IR (-O0):
define i32** @new_diag_matrix2i(i32* %0, i32 %1) {
%3 = alloca i32*
%4 = alloca i32
%5 = alloca i32**
%6 = alloca i32
store i32* %0, i32** %3
store i32 %1, i32* %4
%7 = load i32, i32* %4
%8 = load i32, i32* %4
%9 = call i32** @new_matrix2i(i32 %7, i32 %8)
store i32** %9, i32*** %5
store i32 0, i32* %6
br label %10
10: ; preds = %28, %2
%11 = load i32, i32* %6
%12 = load i32, i32* %4
%13 = icmp slt i32 %11, %12
br i1 %13, label %14, label %31
14: ; preds = %10
%15 = load i32*, i32** %3
%16 = load i32, i32* %6
%17 = sext i32 %16 to i64
%18 = getelementptr inbounds i32, i32* %15, i64 %17
%19 = load i32, i32* %18
%20 = load i32**, i32*** %5
%21 = load i32, i32* %6
%22 = sext i32 %21 to i64
%23 = getelementptr inbounds i32*, i32** %20, i64 %22
%24 = load i32*, i32** %23
%25 = load i32, i32* %6
%26 = sext i32 %25 to i64
%27 = getelementptr inbounds i32, i32* %24, i64 %26
store i32 %19, i32* %27
br label %28
28: ; preds = %14
%29 = load i32, i32* %6
%30 = add nsw i32 %29, 1
store i32 %30, i32* %6
br label %10
31: ; preds = %10
%32 = load i32**, i32*** %5
ret i32** %32
}

LLVM IR (-O2):
define i32** @new_diag_matrix2i(i32* nocapture %0, i32 %1) {
%3 = tail call i32** @new_matrix2i(i32 %1, i32 %1)
%4 = icmp sgt i32 %1, 0
br i1 %4, label %5, label %55
5: ; preds = %2
%6 = zext i32 %1 to i64
%7 = add nsw i64 %6, -1
%8 = and i64 %6, 3
%9 = icmp ult i64 %7, 3
br i1 %9, label %41, label %10
10: ; preds = %5
%11 = and i64 %6, 4294967292
br label %12
12: ; preds = %12, %10
%13 = phi i64 [ 0, %10 ], [ %38, %12 ]
%14 = phi i64 [ 0, %10 ], [ %39, %12 ]
%15 = getelementptr inbounds i32, i32* %0, i64 %13
%16 = load i32, i32* %15
%17 = getelementptr inbounds i32*, i32** %3, i64 %13
%18 = load i32*, i32** %17
%19 = getelementptr inbounds i32, i32* %18, i64 %13
store i32 %16, i32* %19
%20 = or i64 %13, 1
%21 = getelementptr inbounds i32, i32* %0, i64 %20
%22 = load i32, i32* %21
%23 = getelementptr inbounds i32*, i32** %3, i64 %20
%24 = load i32*, i32** %23
%25 = getelementptr inbounds i32, i32* %24, i64 %20
store i32 %22, i32* %25
%26 = or i64 %13, 2
%27 = getelementptr inbounds i32, i32* %0, i64 %26
%28 = load i32, i32* %27
%29 = getelementptr inbounds i32*, i32** %3, i64 %26
%30 = load i32*, i32** %29
%31 = getelementptr inbounds i32, i32* %30, i64 %26
store i32 %28, i32* %31
%32 = or i64 %13, 3
%33 = getelementptr inbounds i32, i32* %0, i64 %32
%34 = load i32, i32* %33
%35 = getelementptr inbounds i32*, i32** %3, i64 %32
%36 = load i32*, i32** %35
%37 = getelementptr inbounds i32, i32* %36, i64 %32
store i32 %34, i32* %37
%38 = add nuw nsw i64 %13, 4
%39 = add i64 %14, 4
%40 = icmp eq i64 %39, %11
br i1 %40, label %41, label %12
41: ; preds = %12, %5
%42 = phi i64 [ 0, %5 ], [ %38, %12 ]
%43 = icmp eq i64 %8, 0
br i1 %43, label %55, label %44
44: ; preds = %41, %44
%45 = phi i64 [ %52, %44 ], [ %42, %41 ]
%46 = phi i64 [ %53, %44 ], [ 0, %41 ]
%47 = getelementptr inbounds i32, i32* %0, i64 %45
%48 = load i32, i32* %47
%49 = getelementptr inbounds i32*, i32** %3, i64 %45
%50 = load i32*, i32** %49
%51 = getelementptr inbounds i32, i32* %50, i64 %45
store i32 %48, i32* %51
%52 = add nuw nsw i64 %45, 1
%53 = add i64 %46, 1
%54 = icmp eq i64 %53, %8
br i1 %54, label %55, label %44
55: ; preds = %41, %44, %2
ret i32** %3
}