linux compile

Tyill · Jul 29, 2019 · 342b1d5 · 342b1d5
1 parent 394ee92
commit 342b1d5
Show file tree

Hide file tree

Showing 9 changed files with 13 additions and 4 deletions.
diff --git a/src/snAux/src/auxFunc.cpp b/src/snAux/src/auxFunc.cpp
@@ -24,7 +24,7 @@
 //
 #include <thread>
 #include <chrono>
-#include <string>
+#include <cstring>
 #include <vector>
 #include <algorithm>
 #include <cctype>

diff --git a/src/snOperatorCPU/src/CPU/tensor.cpp b/src/snOperatorCPU/src/CPU/tensor.cpp
@@ -1,5 +1,6 @@
 
 #include "snBase/snBase.h"
+#include <cstring>
 
 using namespace SN_Base;
 

diff --git a/src/snOperatorCPU/src/stdafx.h b/src/snOperatorCPU/src/stdafx.h
@@ -25,6 +25,8 @@
 
 #pragma once
 
+
+#include <cstring>
 #include <algorithm>
 #include <omp.h>
 #include "snBase/snBase.h"

diff --git a/src/snOperatorCPU/src/weightInit.cpp b/src/snOperatorCPU/src/weightInit.cpp
@@ -22,6 +22,10 @@
 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 // THE SOFTWARE.
 //
+
+#include <cstring>
+#include <cmath>
+
 #include "snBase/snBase.h"
 #include "random.h"
 #include "structurs.h"

diff --git a/src/snOperatorCUDA/src/CUDA/common.cu b/src/snOperatorCUDA/src/CUDA/common.cu
@@ -22,7 +22,6 @@
 // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
 // THE SOFTWARE.
 //
-#pragma once
 
 #include <cuda_runtime.h>
 #include "../stdafx.h"

diff --git a/src/snOperatorCUDA/src/Operator/batchNorm.h b/src/snOperatorCUDA/src/Operator/batchNorm.h
@@ -39,7 +39,7 @@ class BatchNorm final : SN_Base::OperatorBase{
 
     bool setBatchNorm(const SN_Base::batchNorm& bn) override;
 
-    SN_Base::batchNorm BatchNorm::getBatchNorm()const override;
+    SN_Base::batchNorm getBatchNorm()const override;
 
 private: 
 

diff --git a/src/snOperatorCUDA/src/Operator/convolution.cpp b/src/snOperatorCUDA/src/Operator/convolution.cpp
@@ -332,7 +332,6 @@ void Convolution::backward(const SN_Base::Tensor& inTns, const operationParam& o
     }   
 }
 
-#include <cuda_runtime.h>
 void Convolution::updateConfig(bool isLern, const snSize& newsz){
 
     size_t& kernel = convPrms_.kernel,

diff --git a/src/snSIMD/src/convolutionBWD.cpp b/src/snSIMD/src/convolutionBWD.cpp
@@ -24,8 +24,10 @@
 //
 
 
+#include <immintrin.h>
 #include <omp.h>
 #include <thread>
+#include <cstring>
 #include "snBase/snBase.h"
 
 using namespace std;

diff --git a/src/snSIMD/src/convolutionFWD.cpp b/src/snSIMD/src/convolutionFWD.cpp
@@ -23,9 +23,11 @@
 // THE SOFTWARE.
 //
 
+#include <immintrin.h>
 #include <omp.h>
 #include <thread>
 #include <iostream>
+#include <cstring>
 #include "snBase/snBase.h"
 #include "Lib/OpenBLAS/cblas.h"