#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 setidmat(double *a, int n) {
  int i;
  vzero(a, n * n);
  for (i = 0; i < n; i++) {
    a[i * n + i] = 1.0;
  }
}


x86 assembly (clang, -O0):
<setidmat>:
push   %rbp
mov    %rsp,%rbp
sub    $0x10,%rsp
mov    %rdi,-0x8(%rbp)
mov    %esi,-0xc(%rbp)
mov    -0x8(%rbp),%rdi
mov    -0xc(%rbp),%esi
imul   -0xc(%rbp),%esi
call   1030 <vzero@plt>
movl   $0x0,-0x10(%rbp)
mov    -0x10(%rbp),%eax
cmp    -0xc(%rbp),%eax
jge    116e <setidmat+0x5e>
mov    -0x8(%rbp),%rax
mov    -0x10(%rbp),%ecx
imul   -0xc(%rbp),%ecx
add    -0x10(%rbp),%ecx
movslq %ecx,%rcx
movsd  0xea5(%rip),%xmm0
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x10(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x10(%rbp)
jmp    1136 <setidmat+0x26>
add    $0x10,%rsp
pop    %rbp
ret

x86 assembly (clang, -O2):
<setidmat>:
push   %rbp
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
sub    $0x18,%rsp
mov    %esi,%ebx
mov    %rdi,%r15
imul   %esi,%esi
call   1030 <vzero@plt>
test   %ebx,%ebx
jle    1216 <setidmat+0x106>
mov    %ebx,%r11d
lea    -0x1(%r11),%rax
mov    %r11,%rsi
and    $0x3,%r11d
xor    %edx,%edx
movabs $0x3ff0000000000000,%r9
cmp    $0x3,%rax
jb     11ee <setidmat+0xde>
lea    (%rbx,%rbx,1),%edi
lea    0x0(,%rbx,4),%r10d
lea    (%rbx,%rbx,2),%ecx
and    $0xfffffffc,%esi
neg    %rsi
mov    %rsi,0x10(%rsp)
mov    $0x2,%r12d
mov    $0x1,%r13d
mov    $0x3,%r8d
xor    %esi,%esi
mov    %rbx,0x8(%rsp)
mov    %ebx,%eax
xor    %ebp,%ebp
nopl   (%rax)
movslq %edi,%rdi
cltq
movslq %edx,%rdx
lea    (%rdx,%rbp,1),%r14
mov    %r9,(%r15,%r14,8)
lea    0x0(%r13,%rax,1),%rbx
mov    %r9,(%r15,%rbx,8)
lea    (%r12,%rdi,1),%rbx
movslq %ecx,%rcx
mov    %r9,(%r15,%rbx,8)
lea    (%r8,%rcx,1),%rbx
mov    %r9,(%r15,%rbx,8)
add    $0x4,%rbp
add    %r10d,%edx
add    $0x4,%r12
add    %r10d,%edi
add    $0x4,%r13
add    %r10d,%eax
add    $0x4,%r8
add    %r10d,%ecx
add    $0xfffffffffffffffc,%rsi
cmp    %rsi,0x10(%rsp)
jne    1190 <setidmat+0x80>
neg    %rsi
mov    %rsi,%rdx
mov    0x8(%rsp),%rbx
test   %r11,%r11
je     1216 <setidmat+0x106>
mov    %ebx,%eax
imul   %edx,%eax
nopl   0x0(%rax,%rax,1)
cltq
lea    (%rdx,%rax,1),%rcx
mov    %r9,(%r15,%rcx,8)
add    $0x1,%rdx
add    %ebx,%eax
add    $0xffffffffffffffff,%r11
jne    1200 <setidmat+0xf0>
add    $0x18,%rsp
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
pop    %rbp
ret

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

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