void unscale_beta(double *beta2, double *beta1, double *mean, double *sd, int p,
                  int K) {
  int j, k;
  double t, s;
  for (k = 0; k < K; k++) {
    s = 0;
    for (j = 1; j < p; j++) {
      t = beta1[p * k + j] * mean[j];
      beta2[p * k + j] = beta1[p * k + j];
      if (sd[j] != 0) {
        t /= sd[j];
        beta2[p * k + j] /= sd[j];
      }
      s += t;
    }
    beta2[p * k] = beta1[p * k] - s;
  }
}


x86 assembly (clang, -O0):
<unscale_beta>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %rdx,-0x18(%rbp)
mov    %rcx,-0x20(%rbp)
mov    %r8d,-0x24(%rbp)
mov    %r9d,-0x28(%rbp)
movl   $0x0,-0x30(%rbp)
mov    -0x30(%rbp),%eax
cmp    -0x28(%rbp),%eax
jge    1260 <unscale_beta+0x160>
xorps  %xmm0,%xmm0
movsd  %xmm0,-0x40(%rbp)
movl   $0x1,-0x2c(%rbp)
mov    -0x2c(%rbp),%eax
cmp    -0x24(%rbp),%eax
jge    1227 <unscale_beta+0x127>
mov    -0x10(%rbp),%rax
mov    -0x24(%rbp),%ecx
imul   -0x30(%rbp),%ecx
add    -0x2c(%rbp),%ecx
movslq %ecx,%rcx
movsd  (%rax,%rcx,8),%xmm0
mov    -0x18(%rbp),%rax
movslq -0x2c(%rbp),%rcx
mulsd  (%rax,%rcx,8),%xmm0
movsd  %xmm0,-0x38(%rbp)
mov    -0x10(%rbp),%rax
mov    -0x24(%rbp),%ecx
imul   -0x30(%rbp),%ecx
add    -0x2c(%rbp),%ecx
movslq %ecx,%rcx
movsd  (%rax,%rcx,8),%xmm0
mov    -0x8(%rbp),%rax
mov    -0x24(%rbp),%ecx
imul   -0x30(%rbp),%ecx
add    -0x2c(%rbp),%ecx
movslq %ecx,%rcx
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x20(%rbp),%rax
movslq -0x2c(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm0
xorps  %xmm1,%xmm1
ucomisd %xmm1,%xmm0
jne    11c3 <unscale_beta+0xc3>
jp     11c3 <unscale_beta+0xc3>
jmp    120a <unscale_beta+0x10a>
mov    -0x20(%rbp),%rax
movslq -0x2c(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm1
movsd  -0x38(%rbp),%xmm0
divsd  %xmm1,%xmm0
movsd  %xmm0,-0x38(%rbp)
mov    -0x20(%rbp),%rax
movslq -0x2c(%rbp),%rcx
movsd  (%rax,%rcx,8),%xmm1
mov    -0x8(%rbp),%rax
mov    -0x24(%rbp),%ecx
imul   -0x30(%rbp),%ecx
add    -0x2c(%rbp),%ecx
movslq %ecx,%rcx
movsd  (%rax,%rcx,8),%xmm0
divsd  %xmm1,%xmm0
movsd  %xmm0,(%rax,%rcx,8)
movsd  -0x38(%rbp),%xmm0
addsd  -0x40(%rbp),%xmm0
movsd  %xmm0,-0x40(%rbp)
mov    -0x2c(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x2c(%rbp)
jmp    113e <unscale_beta+0x3e>
mov    -0x10(%rbp),%rax
mov    -0x24(%rbp),%ecx
imul   -0x30(%rbp),%ecx
movslq %ecx,%rcx
movsd  (%rax,%rcx,8),%xmm0
subsd  -0x40(%rbp),%xmm0
mov    -0x8(%rbp),%rax
mov    -0x24(%rbp),%ecx
imul   -0x30(%rbp),%ecx
movslq %ecx,%rcx
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x30(%rbp),%eax
add    $0x1,%eax
mov    %eax,-0x30(%rbp)
jmp    1123 <unscale_beta+0x23>
pop    %rbp
ret

x86 assembly (clang, -O2):
<unscale_beta>:
push   %r15
push   %r14
push   %r12
push   %rbx
test   %r9d,%r9d
jle    11ad <unscale_beta+0xad>
movslq %r8d,%r10
mov    %r9d,%r8d
mov    %r10d,%r14d
lea    0x0(,%r10,8),%r9
xor    %r11d,%r11d
xorpd  %xmm0,%xmm0
mov    %rdi,%rbx
mov    %rsi,%r12
jmp    114f <unscale_beta+0x4f>
movsd  (%rsi,%r15,8),%xmm2
subsd  %xmm1,%xmm2
movsd  %xmm2,(%rdi,%r15,8)
add    $0x1,%r11
add    %r9,%r12
add    %r9,%rbx
cmp    %r8,%r11
je     11ad <unscale_beta+0xad>
mov    %r11,%r15
imul   %r10,%r15
xorpd  %xmm1,%xmm1
cmp    $0x2,%r10d
jl     1130 <unscale_beta+0x30>
xorpd  %xmm1,%xmm1
mov    $0x1,%eax
jmp    117d <unscale_beta+0x7d>
nopl   0x0(%rax,%rax,1)
addsd  %xmm2,%xmm1
add    $0x1,%rax
cmp    %rax,%r14
je     1130 <unscale_beta+0x30>
movsd  (%r12,%rax,8),%xmm3
movsd  (%rdx,%rax,8),%xmm2
mulsd  %xmm3,%xmm2
movsd  %xmm3,(%rbx,%rax,8)
movsd  (%rcx,%rax,8),%xmm4
ucomisd %xmm0,%xmm4
jne    119e <unscale_beta+0x9e>
jnp    1170 <unscale_beta+0x70>
divsd  %xmm4,%xmm2
divsd  %xmm4,%xmm3
movsd  %xmm3,(%rbx,%rax,8)
jmp    1170 <unscale_beta+0x70>
pop    %rbx
pop    %r12
pop    %r14
pop    %r15
ret

LLVM IR (-O0):
define void @unscale_beta(double* %0, double* %1, double* %2, double* %3, i32 %4, i32 %5) {
%7 = alloca double*
%8 = alloca double*
%9 = alloca double*
%10 = alloca double*
%11 = alloca i32
%12 = alloca i32
%13 = alloca i32
%14 = alloca i32
%15 = alloca double
%16 = alloca double
store double* %0, double** %7
store double* %1, double** %8
store double* %2, double** %9
store double* %3, double** %10
store i32 %4, i32* %11
store i32 %5, i32* %12
store i32 0, i32* %14
br label %17
17: ; preds = %111, %6
%18 = load i32, i32* %14
%19 = load i32, i32* %12
%20 = icmp slt i32 %18, %19
br i1 %20, label %21, label %114
21: ; preds = %17
store double 0.000000e+00, double* %16
store i32 1, i32* %13
br label %22
22: ; preds = %92, %21
%23 = load i32, i32* %13
%24 = load i32, i32* %11
%25 = icmp slt i32 %23, %24
br i1 %25, label %26, label %95
26: ; preds = %22
%27 = load double*, double** %8
%28 = load i32, i32* %11
%29 = load i32, i32* %14
%30 = mul nsw i32 %28, %29
%31 = load i32, i32* %13
%32 = add nsw i32 %30, %31
%33 = sext i32 %32 to i64
%34 = getelementptr inbounds double, double* %27, i64 %33
%35 = load double, double* %34
%36 = load double*, double** %9
%37 = load i32, i32* %13
%38 = sext i32 %37 to i64
%39 = getelementptr inbounds double, double* %36, i64 %38
%40 = load double, double* %39
%41 = fmul double %35, %40
store double %41, double* %15
%42 = load double*, double** %8
%43 = load i32, i32* %11
%44 = load i32, i32* %14
%45 = mul nsw i32 %43, %44
%46 = load i32, i32* %13
%47 = add nsw i32 %45, %46
%48 = sext i32 %47 to i64
%49 = getelementptr inbounds double, double* %42, i64 %48
%50 = load double, double* %49
%51 = load double*, double** %7
%52 = load i32, i32* %11
%53 = load i32, i32* %14
%54 = mul nsw i32 %52, %53
%55 = load i32, i32* %13
%56 = add nsw i32 %54, %55
%57 = sext i32 %56 to i64
%58 = getelementptr inbounds double, double* %51, i64 %57
store double %50, double* %58
%59 = load double*, double** %10
%60 = load i32, i32* %13
%61 = sext i32 %60 to i64
%62 = getelementptr inbounds double, double* %59, i64 %61
%63 = load double, double* %62
%64 = fcmp une double %63, 0.000000e+00
br i1 %64, label %65, label %88
65: ; preds = %26
%66 = load double*, double** %10
%67 = load i32, i32* %13
%68 = sext i32 %67 to i64
%69 = getelementptr inbounds double, double* %66, i64 %68
%70 = load double, double* %69
%71 = load double, double* %15
%72 = fdiv double %71, %70
store double %72, double* %15
%73 = load double*, double** %10
%74 = load i32, i32* %13
%75 = sext i32 %74 to i64
%76 = getelementptr inbounds double, double* %73, i64 %75
%77 = load double, double* %76
%78 = load double*, double** %7
%79 = load i32, i32* %11
%80 = load i32, i32* %14
%81 = mul nsw i32 %79, %80
%82 = load i32, i32* %13
%83 = add nsw i32 %81, %82
%84 = sext i32 %83 to i64
%85 = getelementptr inbounds double, double* %78, i64 %84
%86 = load double, double* %85
%87 = fdiv double %86, %77
store double %87, double* %85
br label %88
88: ; preds = %65, %26
%89 = load double, double* %15
%90 = load double, double* %16
%91 = fadd double %90, %89
store double %91, double* %16
br label %92
92: ; preds = %88
%93 = load i32, i32* %13
%94 = add nsw i32 %93, 1
store i32 %94, i32* %13
br label %22
95: ; preds = %22
%96 = load double*, double** %8
%97 = load i32, i32* %11
%98 = load i32, i32* %14
%99 = mul nsw i32 %97, %98
%100 = sext i32 %99 to i64
%101 = getelementptr inbounds double, double* %96, i64 %100
%102 = load double, double* %101
%103 = load double, double* %16
%104 = fsub double %102, %103
%105 = load double*, double** %7
%106 = load i32, i32* %11
%107 = load i32, i32* %14
%108 = mul nsw i32 %106, %107
%109 = sext i32 %108 to i64
%110 = getelementptr inbounds double, double* %105, i64 %109
store double %104, double* %110
br label %111
111: ; preds = %95
%112 = load i32, i32* %14
%113 = add nsw i32 %112, 1
store i32 %113, i32* %14
br label %17
114: ; preds = %17
ret void
}

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