Skip to content
Snippets Groups Projects
Commit 6299d190 authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Added OpenMP to MatMul

parent b9b5fa3a
No related branches found
No related tags found
2 merge requests!166Update 0.5.0 -> 0.6.0,!158Added OpenMP
......@@ -26,6 +26,9 @@ void MatMulImpl_cpu_forward_kernel(const std::size_t n, const std::size_t k, con
std::memset(output, O(0), n * m * sizeof(O));
#ifdef _OPENMP
#pragma omp parallel for if (n > 32)
#endif
for (std::size_t i = 0; i < n; ++i) {
for (std::size_t l = 0; l < k; ++l) {
for (std::size_t j = 0; j < m; ++j) {
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment