From 6299d19022b79154ed34ca7b9090be10f2c8de8a Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Mon, 7 Apr 2025 15:13:08 +0200
Subject: [PATCH] Added OpenMP to MatMul
---
include/aidge/backend/cpu/operator/MatMulImpl_kernels.hpp | 3 +++
1 file changed, 3 insertions(+)
diff --git a/include/aidge/backend/cpu/operator/MatMulImpl_kernels.hpp b/include/aidge/backend/cpu/operator/MatMulImpl_kernels.hpp
index 5fc13baf..ec55d31a 100644
--- a/include/aidge/backend/cpu/operator/MatMulImpl_kernels.hpp
+++ b/include/aidge/backend/cpu/operator/MatMulImpl_kernels.hpp
@@ -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) {
--
GitLab