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

Fixed Windows build

parent 6299d190
No related branches found
No related tags found
2 merge requests!166Update 0.5.0 -> 0.6.0,!158Added OpenMP
......@@ -29,7 +29,7 @@ void MatMulImpl_cpu_forward_kernel(const std::size_t n, const std::size_t k, con
#ifdef _OPENMP
#pragma omp parallel for if (n > 32)
#endif
for (std::size_t i = 0; i < n; ++i) {
for (int i = 0; i < static_cast<int>(n); ++i) {
for (std::size_t l = 0; l < k; ++l) {
for (std::size_t j = 0; j < m; ++j) {
output[i*m + j] += static_cast<O>(input1[i*k + l] * input2[l*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