Skip to content

Commit 58b5ec9

Browse files
fmt
1 parent 7cb03fb commit 58b5ec9

1 file changed

Lines changed: 28 additions & 4 deletions

File tree

crates/arrow-ml-linalg/benches/matmul_bench.rs

Lines changed: 28 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -90,8 +90,20 @@ fn bench_matmul_f32_square(c: &mut Criterion) {
9090
let mut c = vec![0.0f32; s * s];
9191
unsafe {
9292
matrixmultiply::sgemm(
93-
s, s, s, 1.0, a_data.as_ptr(), s as isize, 1, b_data.as_ptr(),
94-
s as isize, 1, 0.0, c.as_mut_ptr(), s as isize, 1,
93+
s,
94+
s,
95+
s,
96+
1.0,
97+
a_data.as_ptr(),
98+
s as isize,
99+
1,
100+
b_data.as_ptr(),
101+
s as isize,
102+
1,
103+
0.0,
104+
c.as_mut_ptr(),
105+
s as isize,
106+
1,
95107
);
96108
}
97109
c
@@ -133,8 +145,20 @@ fn bench_matmul_f64_square(c: &mut Criterion) {
133145
let mut c = vec![0.0f64; s * s];
134146
unsafe {
135147
matrixmultiply::dgemm(
136-
s, s, s, 1.0, a_data.as_ptr(), s as isize, 1, b_data.as_ptr(),
137-
s as isize, 1, 0.0, c.as_mut_ptr(), s as isize, 1,
148+
s,
149+
s,
150+
s,
151+
1.0,
152+
a_data.as_ptr(),
153+
s as isize,
154+
1,
155+
b_data.as_ptr(),
156+
s as isize,
157+
1,
158+
0.0,
159+
c.as_mut_ptr(),
160+
s as isize,
161+
1,
138162
);
139163
}
140164
c

0 commit comments

Comments
 (0)