void __attribute__((visibility("default")))
scalar_mul_matrix(double *matrix, unsigned long shape0, unsigned long shape1,
                  unsigned long stride, double scalar) {
  unsigned long i, j;
  for (i = 0; i < shape0; i++) {
    for (j = 0; j < shape1; j++) {
      matrix[i * stride + j] = scalar * matrix[i * stride + j];
    }
  }
}


x86 assembly (clang, -O0):
<scalar_mul_matrix>:
push   %rbp
mov    %rsp,%rbp
mov    %rdi,-0x8(%rbp)
mov    %rsi,-0x10(%rbp)
mov    %rdx,-0x18(%rbp)
mov    %rcx,-0x20(%rbp)
movsd  %xmm0,-0x28(%rbp)
movq   $0x0,-0x30(%rbp)
mov    -0x30(%rbp),%rax
cmp    -0x10(%rbp),%rax
jae    119d <scalar_mul_matrix+0x9d>
movq   $0x0,-0x38(%rbp)
mov    -0x38(%rbp),%rax
cmp    -0x18(%rbp),%rax
jae    1187 <scalar_mul_matrix+0x87>
movsd  -0x28(%rbp),%xmm0
mov    -0x8(%rbp),%rax
mov    -0x30(%rbp),%rcx
imul   -0x20(%rbp),%rcx
add    -0x38(%rbp),%rcx
mulsd  (%rax,%rcx,8),%xmm0
mov    -0x8(%rbp),%rax
mov    -0x30(%rbp),%rcx
imul   -0x20(%rbp),%rcx
add    -0x38(%rbp),%rcx
movsd  %xmm0,(%rax,%rcx,8)
mov    -0x38(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x38(%rbp)
jmp    1137 <scalar_mul_matrix+0x37>
jmp    118c <scalar_mul_matrix+0x8c>
mov    -0x30(%rbp),%rax
add    $0x1,%rax
mov    %rax,-0x30(%rbp)
jmp    1121 <scalar_mul_matrix+0x21>
pop    %rbp
ret

x86 assembly (clang, -O2):
<scalar_mul_matrix>:
push   %r15
push   %r14
push   %r13
push   %r12
push   %rbx
test   %rsi,%rsi
je     123e <scalar_mul_matrix+0x13e>
lea    -0x4(%rdx),%r8
mov    %r8,%r9
shr    $0x2,%r9
add    $0x1,%r9
mov    %rdx,%r11
and    $0xfffffffffffffffc,%r11
movapd %xmm0,%xmm1
unpcklpd %xmm0,%xmm1
mov    %r9,%r10
and    $0xfffffffffffffffe,%r10
lea    0x30(%rdi),%r13
lea    0x0(,%rcx,8),%r14
xor    %r15d,%r15d
mov    %rdi,%r12
jmp    1163 <scalar_mul_matrix+0x63>
nopl   0x0(%rax,%rax,1)
add    $0x1,%r15
add    %r14,%r13
add    %r14,%r12
cmp    %rsi,%r15
je     123e <scalar_mul_matrix+0x13e>
test   %rdx,%rdx
je     1150 <scalar_mul_matrix+0x50>
cmp    $0x4,%rdx
jae    1180 <scalar_mul_matrix+0x80>
xor    %eax,%eax
jmp    1220 <scalar_mul_matrix+0x120>
cs nopw 0x0(%rax,%rax,1)
nop
cmp    $0x4,%r8
jae    118a <scalar_mul_matrix+0x8a>
xor    %ebx,%ebx
jmp    11e2 <scalar_mul_matrix+0xe2>
mov    %r10,%rax
xor    %ebx,%ebx
nop
movupd -0x30(%r13,%rbx,8),%xmm2
movupd -0x20(%r13,%rbx,8),%xmm3
movupd -0x10(%r13,%rbx,8),%xmm4
movupd 0x0(%r13,%rbx,8),%xmm5
mulpd  %xmm1,%xmm2
mulpd  %xmm1,%xmm3
movupd %xmm2,-0x30(%r13,%rbx,8)
movupd %xmm3,-0x20(%r13,%rbx,8)
mulpd  %xmm1,%xmm4
mulpd  %xmm1,%xmm5
movupd %xmm4,-0x10(%r13,%rbx,8)
movupd %xmm5,0x0(%r13,%rbx,8)
add    $0x8,%rbx
add    $0xfffffffffffffffe,%rax
jne    1190 <scalar_mul_matrix+0x90>
test   $0x1,%r9b
je     1210 <scalar_mul_matrix+0x110>
mov    %r15,%rax
imul   %rcx,%rax
add    %rax,%rbx
movupd (%rdi,%rbx,8),%xmm2
movupd 0x10(%rdi,%rbx,8),%xmm3
mulpd  %xmm1,%xmm2
mulpd  %xmm1,%xmm3
movupd %xmm2,(%rdi,%rbx,8)
movupd %xmm3,0x10(%rdi,%rbx,8)
mov    %r11,%rax
cmp    %rdx,%r11
je     1150 <scalar_mul_matrix+0x50>
nopl   0x0(%rax)
movsd  (%r12,%rax,8),%xmm2
mulsd  %xmm0,%xmm2
movsd  %xmm2,(%r12,%rax,8)
add    $0x1,%rax
cmp    %rax,%rdx
jne    1220 <scalar_mul_matrix+0x120>
jmp    1150 <scalar_mul_matrix+0x50>
pop    %rbx
pop    %r12
pop    %r13
pop    %r14
pop    %r15
ret

LLVM IR (-O0):
define void @scalar_mul_matrix(double* %0, i64 %1, i64 %2, i64 %3, double %4) {
%6 = alloca double*
%7 = alloca i64
%8 = alloca i64
%9 = alloca i64
%10 = alloca double
%11 = alloca i64
%12 = alloca i64
store double* %0, double** %6
store i64 %1, i64* %7
store i64 %2, i64* %8
store i64 %3, i64* %9
store double %4, double* %10
store i64 0, i64* %11
br label %13
13: ; preds = %44, %5
%14 = load i64, i64* %11
%15 = load i64, i64* %7
%16 = icmp ult i64 %14, %15
br i1 %16, label %17, label %47
17: ; preds = %13
store i64 0, i64* %12
br label %18
18: ; preds = %40, %17
%19 = load i64, i64* %12
%20 = load i64, i64* %8
%21 = icmp ult i64 %19, %20
br i1 %21, label %22, label %43
22: ; preds = %18
%23 = load double, double* %10
%24 = load double*, double** %6
%25 = load i64, i64* %11
%26 = load i64, i64* %9
%27 = mul i64 %25, %26
%28 = load i64, i64* %12
%29 = add i64 %27, %28
%30 = getelementptr inbounds double, double* %24, i64 %29
%31 = load double, double* %30
%32 = fmul double %23, %31
%33 = load double*, double** %6
%34 = load i64, i64* %11
%35 = load i64, i64* %9
%36 = mul i64 %34, %35
%37 = load i64, i64* %12
%38 = add i64 %36, %37
%39 = getelementptr inbounds double, double* %33, i64 %38
store double %32, double* %39
br label %40
40: ; preds = %22
%41 = load i64, i64* %12
%42 = add i64 %41, 1
store i64 %42, i64* %12
br label %18
43: ; preds = %18
br label %44
44: ; preds = %43
%45 = load i64, i64* %11
%46 = add i64 %45, 1
store i64 %46, i64* %11
br label %13
47: ; preds = %13
ret void
}

LLVM IR (-O2):
define void @scalar_mul_matrix(double* nocapture %0, i64 %1, i64 %2, i64 %3, double %4) {
%6 = icmp eq i64 %1, 0
br i1 %6, label %85, label %7
7: ; preds = %5
%8 = icmp eq i64 %2, 0
%9 = add i64 %2, -4
%10 = lshr i64 %9, 2
%11 = add nuw nsw i64 %10, 1
%12 = icmp ult i64 %2, 4
%13 = and i64 %2, -4
%14 = insertelement <2 x double> poison, double %4, i64 0
%15 = shufflevector <2 x double> %14, <2 x double> poison, <2 x i32> zeroinitializer
%16 = insertelement <2 x double> poison, double %4, i64 0
%17 = shufflevector <2 x double> %16, <2 x double> poison, <2 x i32> zeroinitializer
%18 = and i64 %11, 1
%19 = icmp ult i64 %9, 4
%20 = and i64 %11, 9223372036854775806
%21 = icmp eq i64 %18, 0
%22 = icmp eq i64 %13, %2
br label %23
23: ; preds = %7, %82
%24 = phi i64 [ 0, %7 ], [ %83, %82 ]
br i1 %8, label %82, label %25
25: ; preds = %23
%26 = mul i64 %24, %3
br i1 %12, label %72, label %27
27: ; preds = %25
br i1 %19, label %57, label %28
28: ; preds = %27, %28
%29 = phi i64 [ %54, %28 ], [ 0, %27 ]
%30 = phi i64 [ %55, %28 ], [ 0, %27 ]
%31 = add i64 %29, %26
%32 = getelementptr inbounds double, double* %0, i64 %31
%33 = bitcast double* %32 to <2 x double>*
%34 = load <2 x double>, <2 x double>* %33
%35 = getelementptr inbounds double, double* %32, i64 2
%36 = bitcast double* %35 to <2 x double>*
%37 = load <2 x double>, <2 x double>* %36
%38 = fmul <2 x double> %34, %15
%39 = fmul <2 x double> %37, %17
%40 = bitcast double* %32 to <2 x double>*
store <2 x double> %38, <2 x double>* %40
%41 = bitcast double* %35 to <2 x double>*
store <2 x double> %39, <2 x double>* %41
%42 = or i64 %29, 4
%43 = add i64 %42, %26
%44 = getelementptr inbounds double, double* %0, i64 %43
%45 = bitcast double* %44 to <2 x double>*
%46 = load <2 x double>, <2 x double>* %45
%47 = getelementptr inbounds double, double* %44, i64 2
%48 = bitcast double* %47 to <2 x double>*
%49 = load <2 x double>, <2 x double>* %48
%50 = fmul <2 x double> %46, %15
%51 = fmul <2 x double> %49, %17
%52 = bitcast double* %44 to <2 x double>*
store <2 x double> %50, <2 x double>* %52
%53 = bitcast double* %47 to <2 x double>*
store <2 x double> %51, <2 x double>* %53
%54 = add nuw i64 %29, 8
%55 = add i64 %30, 2
%56 = icmp eq i64 %55, %20
br i1 %56, label %57, label %28
57: ; preds = %28, %27
%58 = phi i64 [ 0, %27 ], [ %54, %28 ]
br i1 %21, label %71, label %59
59: ; preds = %57
%60 = add i64 %58, %26
%61 = getelementptr inbounds double, double* %0, i64 %60
%62 = bitcast double* %61 to <2 x double>*
%63 = load <2 x double>, <2 x double>* %62
%64 = getelementptr inbounds double, double* %61, i64 2
%65 = bitcast double* %64 to <2 x double>*
%66 = load <2 x double>, <2 x double>* %65
%67 = fmul <2 x double> %63, %15
%68 = fmul <2 x double> %66, %17
%69 = bitcast double* %61 to <2 x double>*
store <2 x double> %67, <2 x double>* %69
%70 = bitcast double* %64 to <2 x double>*
store <2 x double> %68, <2 x double>* %70
br label %71
71: ; preds = %57, %59
br i1 %22, label %82, label %72
72: ; preds = %25, %71
%73 = phi i64 [ 0, %25 ], [ %13, %71 ]
br label %74
74: ; preds = %72, %74
%75 = phi i64 [ %80, %74 ], [ %73, %72 ]
%76 = add i64 %75, %26
%77 = getelementptr inbounds double, double* %0, i64 %76
%78 = load double, double* %77
%79 = fmul double %78, %4
store double %79, double* %77
%80 = add nuw i64 %75, 1
%81 = icmp eq i64 %80, %2
br i1 %81, label %82, label %74
82: ; preds = %74, %71, %23
%83 = add nuw i64 %24, 1
%84 = icmp eq i64 %83, %1
br i1 %84, label %85, label %23
85: ; preds = %82, %5
ret void
}