mirror of
https://github.com/ml-explore/mlx.git
synced 2025-07-19 15:41:13 +08:00
parent
50f5d14b11
commit
c3272d4917
@ -357,7 +357,7 @@ void explicit_gemm_conv_1D_cpu(
|
|||||||
gemm_out.set_data(allocator::malloc_or_wait(gemm_out.nbytes()));
|
gemm_out.set_data(allocator::malloc_or_wait(gemm_out.nbytes()));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Peform gemm
|
// Perform gemm
|
||||||
cblas_sgemm(
|
cblas_sgemm(
|
||||||
CblasRowMajor,
|
CblasRowMajor,
|
||||||
CblasNoTrans, // no trans A
|
CblasNoTrans, // no trans A
|
||||||
@ -459,7 +459,7 @@ void explicit_gemm_conv_2D_cpu(
|
|||||||
gemm_out.set_data(allocator::malloc_or_wait(gemm_out.nbytes()));
|
gemm_out.set_data(allocator::malloc_or_wait(gemm_out.nbytes()));
|
||||||
}
|
}
|
||||||
|
|
||||||
// Peform gemm
|
// Perform gemm
|
||||||
cblas_sgemm(
|
cblas_sgemm(
|
||||||
CblasRowMajor,
|
CblasRowMajor,
|
||||||
CblasNoTrans, // no trans A
|
CblasNoTrans, // no trans A
|
||||||
|
Loading…
Reference in New Issue
Block a user