void inv_diag(int n, const double *b, double *x, void *inv_param) {
  double *a = (double *)inv_param;
  int i;
  for (i = 0; i < n; i++) {
    if (a[i * n + i] != 0.0)
      x[i] = b[i] / a[i * n + i];
    else
      x[i] = b[i];
  }
}


x86 assembly (clang, -O0):
<inv_diag>:
push   %rbp
mov    %rsp,%rbp
mov    %edi,-0x4(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %rdx,-0x18(%rbp)
mov    %rcx,-0x20(%rbp)
mov    -0x20(%rbp),%rax
mov    %rax,-0x28(%rbp)
movl   $0x0,-0x2c(%rbp)
mov    -0x2c(%rbp),%eax
cmp    -0x4(%rbp),%eax
jge    11be <inv_diag+0xbe>
mov    -0x28(%rbp),%rax
mov    -0x2c(%rbp),%ecx
imul   -0x4(%rbp),%ecx
add    -0x2c(%rbp),%ecx
movslq %ecx,%rcx
movsd  (%rax,%rcx,8),%xmm0
xorps  %xmm1,%xmm1
ucomisd %xmm1,%xmm0
jne    115c <inv_diag+0x5c>
jp     115c <inv_diag+0x5c>
jmp    1191 <inv_diag+0x91>
mov    -0x10(%rbp),%rax
movslq -0x2c(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
mov    -0x28(%rbp),%rax
mov    -0x2c(%rbp),%ecx
imul   -0x4(%rbp),%ecx
add    -0x2c(%rbp),%ecx
movslq %ecx,%rcx
divsd  (%rax,%rcx,8),%xmm0
mov    -0x18(%rbp),%rax
movslq -0x2c(%rbp),%rcx
movsd  %xmm0,(%rax,%rcx,8)
jmp    11ab <inv_diag+0xab>
mov    -0x10(%rbp),%rax
movslq -0x2c(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
mov    -0x18(%rbp),%rax
movslq -0x2c(%rbp),%rcx
movsd  %xmm0,(%rax,%rcx,8)
jmp    11b0 <inv_diag+0xb0>
mov    -0x2c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x2c(%rbp)
jmp    1122 <inv_diag+0x22>
pop    %rbp
ret

x86 assembly (clang, -O2):
<inv_diag>:
test   %edi,%edi
jle    1161 <inv_diag+0x61>
mov    %edi,%r8d
shl    $0x3,%r8
xor    %r11d,%r11d
xorpd  %xmm0,%xmm0
movsd  0xee6(%rip),%xmm1
xor    %r9d,%r9d
xor    %r10d,%r10d
movslq %r11d,%r11
lea    (%r10,%r11,1),%rax
movsd  (%rcx,%rax,8),%xmm2
movsd  (%rsi,%r9,1),%xmm3
movapd %xmm2,%xmm4
cmpneqsd %xmm0,%xmm4
andpd  %xmm4,%xmm2
andnpd %xmm1,%xmm4
orpd   %xmm2,%xmm4
divsd  %xmm4,%xmm3
movsd  %xmm3,(%rdx,%r9,1)
add    $0x1,%r10
add    $0x8,%r9
add    %edi,%r11d
cmp    %r9,%r8
jne    1120 <inv_diag+0x20>
ret

LLVM IR (-O0):
define void @inv_diag(i32 %0, double* %1, double* %2, i8* %3) {
%5 = alloca i32
%6 = alloca double*
%7 = alloca double*
%8 = alloca i8*
%9 = alloca double*
%10 = alloca i32
store i32 %0, i32* %5
store double* %1, double** %6
store double* %2, double** %7
store i8* %3, i8** %8
%11 = load i8*, i8** %8
%12 = bitcast i8* %11 to double*
store double* %12, double** %9
store i32 0, i32* %10
br label %13
13: ; preds = %59, %4
%14 = load i32, i32* %10
%15 = load i32, i32* %5
%16 = icmp slt i32 %14, %15
br i1 %16, label %17, label %62
17: ; preds = %13
%18 = load double*, double** %9
%19 = load i32, i32* %10
%20 = load i32, i32* %5
%21 = mul nsw i32 %19, %20
%22 = load i32, i32* %10
%23 = add nsw i32 %21, %22
%24 = sext i32 %23 to i64
%25 = getelementptr inbounds double, double* %18, i64 %24
%26 = load double, double* %25
%27 = fcmp une double %26, 0.000000e+00
br i1 %27, label %28, label %48
28: ; preds = %17
%29 = load double*, double** %6
%30 = load i32, i32* %10
%31 = sext i32 %30 to i64
%32 = getelementptr inbounds double, double* %29, i64 %31
%33 = load double, double* %32
%34 = load double*, double** %9
%35 = load i32, i32* %10
%36 = load i32, i32* %5
%37 = mul nsw i32 %35, %36
%38 = load i32, i32* %10
%39 = add nsw i32 %37, %38
%40 = sext i32 %39 to i64
%41 = getelementptr inbounds double, double* %34, i64 %40
%42 = load double, double* %41
%43 = fdiv double %33, %42
%44 = load double*, double** %7
%45 = load i32, i32* %10
%46 = sext i32 %45 to i64
%47 = getelementptr inbounds double, double* %44, i64 %46
store double %43, double* %47
br label %58
48: ; preds = %17
%49 = load double*, double** %6
%50 = load i32, i32* %10
%51 = sext i32 %50 to i64
%52 = getelementptr inbounds double, double* %49, i64 %51
%53 = load double, double* %52
%54 = load double*, double** %7
%55 = load i32, i32* %10
%56 = sext i32 %55 to i64
%57 = getelementptr inbounds double, double* %54, i64 %56
store double %53, double* %57
br label %58
58: ; preds = %48, %28
br label %59
59: ; preds = %58
%60 = load i32, i32* %10
%61 = add nsw i32 %60, 1
store i32 %61, i32* %10
br label %13
62: ; preds = %13
ret void
}

LLVM IR (-O2):
define void @inv_diag(i32 %0, double* nocapture %1, double* nocapture writeonly %2, i8* nocapture %3) {
%5 = bitcast i8* %3 to double*
%6 = icmp sgt i32 %0, 0
br i1 %6, label %7, label %25
7: ; preds = %4
%8 = zext i32 %0 to i64
br label %9
9: ; preds = %7, %9
%10 = phi i64 [ 0, %7 ], [ %23, %9 ]
%11 = trunc i64 %10 to i32
%12 = mul nsw i32 %11, %0
%13 = sext i32 %12 to i64
%14 = add nsw i64 %10, %13
%15 = getelementptr inbounds double, double* %5, i64 %14
%16 = load double, double* %15
%17 = fcmp une double %16, 0.000000e+00
%18 = getelementptr inbounds double, double* %1, i64 %10
%19 = load double, double* %18
%20 = select i1 %17, double %16, double 1.000000e+00
%21 = fdiv double %19, %20
%22 = getelementptr inbounds double, double* %2, i64 %10
store double %21, double* %22
%23 = add nuw nsw i64 %10, 1
%24 = icmp eq i64 %23, %8
br i1 %24, label %25, label %9
25: ; preds = %9, %4
ret void
}