Minor bug fixes post c++ conversion.

UoB-HPC · Jan 24, 2024 · cabe7d1 · cabe7d1
1 parent 8d05a4c
commit cabe7d1
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 6 deletions.
diff --git a/ArmPL/gemm.hh b/ArmPL/gemm.hh
@@ -51,13 +51,13 @@ class gemm_cpu : public gemm<T> {
   /** Make a class to the BLAS Library Kernel. */
   virtual void callKernel() override {
     if constexpr (std::is_same_v<T, float>) {
-      cblas_sgemm(CblasColMajor, CblasNoTrans, CblasNoTrans, m, n, k, ALPHA,
-                  A_.data(), MAX(1, m), B.data(), MAX(1, k), BETA, C.data(),
-                  MAX(1, m));
+      cblas_sgemm(CblasColMajor, CblasNoTrans, CblasNoTrans, m_, n_, k_, ALPHA,
+                  A_.data(), MAX(1, m_), B_.data(), MAX(1, k_), BETA, C_.data(),
+                  MAX(1, m_));
     } else if constexpr (std::is_same_v<T, double>) {
-      cblas_dgemm(CblasColMajor, CblasNoTrans, CblasNoTrans, m, n, k, ALPHA,
-                  A_.data(), MAX(1, m), B.data(), MAX(1, k), BETA, C.data(),
-                  MAX(1, m));
+      cblas_dgemm(CblasColMajor, CblasNoTrans, CblasNoTrans, m_, n_, k_, ALPHA,
+                  A_.data(), MAX(1, m_), B_.data(), MAX(1, k_), BETA, C_.data(),
+                  MAX(1, m_));
     } else {
       // Un-specialised class will not do any work - print error and exit.
       std::cout << "ERROR - Datatype for ArmPL CPU GEMM kernel not supported."

diff --git a/DefaultGPU/gemm.hh b/DefaultGPU/gemm.hh
@@ -2,6 +2,7 @@
 
 #include <time.h>
 
+#include <cmath>
 #include <vector>
 
 #include "../include/GPU/gemm.hh"

diff --git a/include/helpers.hh b/include/helpers.hh
@@ -1,7 +1,9 @@
 #pragma once
 
+#include <cmath>
 #include <cstdint>
 #include <fstream>
+#include <iomanip>
 #include <iostream>
 #include <string>