#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 /*<<< orphan*/ vzero(double *, int);
void setdiag(double *a, double *diag, int n) {
  int i, k;
  vzero(a, n * n);
  for (i = 0; i < n; i++) {
    k = i * n + i;
    a[k] = diag[i];
  }
}


x86 assembly (clang, -O0):
<setdiag>:
push   %rbp
mov    %rsp,%rbp
sub    $0x20,%rsp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %edx,-0x14(%rbp)
mov    -0x8(%rbp),%rdi
mov    -0x14(%rbp),%esi
imul   -0x14(%rbp),%esi
call   1030 <vzero@plt>
movl   $0x0,-0x18(%rbp)
mov    -0x18(%rbp),%eax
cmp    -0x14(%rbp),%eax
jge    117b <setdiag+0x6b>
mov    -0x18(%rbp),%eax
imul   -0x14(%rbp),%eax
add    -0x18(%rbp),%eax
mov    %eax,-0x1c(%rbp)
mov    -0x10(%rbp),%rax
movslq -0x18(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
mov    -0x8(%rbp),%rax
movslq -0x1c(%rbp),%rcx
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x18(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x18(%rbp)
jmp    113a <setdiag+0x2a>
add    $0x20,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<setdiag>:
push   %r15
push   %r14
push   %r12
push   %rbx
push   %rax
mov    %edx,%r15d
mov    %rsi,%r14
mov    %rdi,%r12
mov    %edx,%esi
imul   %edx,%esi
call   1030 <vzero@plt>
test   %r15d,%r15d
jle    11b6 <setdiag+0xa6>
mov    %r15d,%r8d
xor    %ecx,%ecx
cmp    $0x1,%r15d
jne    1143 <setdiag+0x33>
xor    %edx,%edx
jmp    119a <setdiag+0x8a>
mov    %r8d,%r9d
and    $0xfffffffe,%r9d
lea    (%r15,%r15,1),%r10d
mov    $0x1,%eax
mov    %r15d,%esi
xor    %edx,%edx
nopl   0x0(%rax,%rax,1)
movslq %esi,%rsi
lea    (%rax,%rsi,1),%rbx
movslq %ecx,%rcx
lea    (%rdx,%rcx,1),%rdi
movsd  (%r14,%rdx,8),%xmm0
movsd  %xmm0,(%r12,%rdi,8)
movsd  0x8(%r14,%rdx,8),%xmm0
movsd  %xmm0,(%r12,%rbx,8)
add    $0x2,%rdx
add    %r10d,%ecx
add    $0x2,%rax
add    %r10d,%esi
cmp    %rdx,%r9
jne    1160 <setdiag+0x50>
test   $0x1,%r8b
je     11b6 <setdiag+0xa6>
imul   %edx,%r15d
movslq %r15d,%rax
add    %rdx,%rax
movsd  (%r14,%rdx,8),%xmm0
movsd  %xmm0,(%r12,%rax,8)
add    $0x8,%rsp
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
ret

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

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