[ Mixed Tensor ] Enable FP32 unittest cases

[platform/core/ml/nntrainer.git] / nntrainer / tensor / blas_interface.h

diff --git a/nntrainer/tensor/blas_interface.h b/nntrainer/tensor/blas_interface.h
index 63274a9..1aff5b5 100644 (file)
--- a/nntrainer/tensor/blas_interface.h
+++ b/nntrainer/tensor/blas_interface.h
@@ -39,42 +39,48 @@ enum CBLAS_TRANSPOSE {
 
 namespace nntrainer {
 
+#ifdef ENABLE_FP16
+void sscal(const unsigned int N, const float alpha, __fp16 *X, const int incX);
+__fp16 snrm2(const int N, const __fp16 *X, const int incX);
+void scopy(const unsigned int N, const __fp16 *X, const int incX, __fp16 *Y,
+           const int intY);
+__fp16 sdot(const unsigned int N, const __fp16 *X, const unsigned int incX,
+            const __fp16 *Y, const unsigned int incY);
+void saxpy(const unsigned int N, const float alpha, const __fp16 *X,
+           const int incX, __fp16 *Y, const int incY);
+void sgemm(CBLAS_ORDER order, CBLAS_TRANSPOSE TransA, CBLAS_TRANSPOSE TransB,
+           const unsigned int M, const unsigned int N, const unsigned int K,
+           const float alpha, const __fp16 *A, const unsigned int lda,
+           const __fp16 *B, const unsigned int ldb, const float beta, __fp16 *C,
+           const unsigned int ldc);
+void sgemv(CBLAS_ORDER order, CBLAS_TRANSPOSE TransA, const unsigned int M,
+           const unsigned int N, const float alpha, const __fp16 *A,
+           const unsigned int lda, const __fp16 *X, const int incX,
+           const float beta, __fp16 *Y, const int incY);
+unsigned int isamax(const unsigned int N, const __fp16 *X, const int incX);
+#endif
+
 void sscal(const unsigned int N, const float alpha, void *X, const int incX,
            ml::train::TensorDim::DataType d_type);
 
 void sscal(const unsigned int N, const float alpha, float *X, const int incX);
 
-void sscal(const unsigned int N, const float alpha, __fp16 *X, const int incX);
-
 float snrm2(const int N, const float *X, const int incX);
 
-__fp16 snrm2(const int N, const __fp16 *X, const int incX);
-
 void scopy(const unsigned int N, const void *X, const int incX, void *Y,
            const int incY, ml::train::TensorDim::DataType d_type);
-
 void scopy(const unsigned int N, const float *X, const int incX, float *Y,
            const int intY);
 
-void scopy(const unsigned int N, const __fp16 *X, const int incX, __fp16 *Y,
-           const int intY);
-
 float sdot(const unsigned int N, const float *X, const unsigned int incX,
            const float *Y, const unsigned int incY);
 
-__fp16 sdot(const unsigned int N, const __fp16 *X, const unsigned int incX,
-            const __fp16 *Y, const unsigned int incY);
-
 void saxpy(const unsigned int N, const float alpha, const void *X,
            const int incX, void *Y, const int incY,
            ml::train::TensorDim::DataType d_type);
-
 void saxpy(const unsigned int N, const float alpha, const float *X,
            const int incX, float *Y, const int incY);
 
-void saxpy(const unsigned int N, const float alpha, const __fp16 *X,
-           const int incX, __fp16 *Y, const int incY);
-
 void sgemm(CBLAS_ORDER order, CBLAS_TRANSPOSE TransA, CBLAS_TRANSPOSE TransB,
            const unsigned int M, const unsigned int N, const unsigned int K,
            const float alpha, const void *A, const unsigned int lda,
@@ -87,12 +93,6 @@ void sgemm(CBLAS_ORDER order, CBLAS_TRANSPOSE TransA, CBLAS_TRANSPOSE TransB,
            const float *B, const unsigned int ldb, const float beta, float *C,
            const unsigned int ldc);
 
-void sgemm(CBLAS_ORDER order, CBLAS_TRANSPOSE TransA, CBLAS_TRANSPOSE TransB,
-           const unsigned int M, const unsigned int N, const unsigned int K,
-           const float alpha, const __fp16 *A, const unsigned int lda,
-           const __fp16 *B, const unsigned int ldb, const float beta, __fp16 *C,
-           const unsigned int ldc);
-
 void sgemv(CBLAS_ORDER order, CBLAS_TRANSPOSE TransA, const unsigned int M,
            const unsigned int N, const float alpha, const void *A,
            const unsigned int lda, const void *X, const int incX,
@@ -104,15 +104,8 @@ void sgemv(CBLAS_ORDER order, CBLAS_TRANSPOSE TransA, const unsigned int M,
            const unsigned int lda, const float *X, const int incX,
            const float beta, float *Y, const int incY);
 
-void sgemv(CBLAS_ORDER order, CBLAS_TRANSPOSE TransA, const unsigned int M,
-           const unsigned int N, const float alpha, const __fp16 *A,
-           const unsigned int lda, const __fp16 *X, const int incX,
-           const float beta, __fp16 *Y, const int incY);
-
 unsigned int isamax(const unsigned int N, const float *X, const int incX);
 
-unsigned int isamax(const unsigned int N, const __fp16 *X, const int incX);
-
 } /* namespace nntrainer */
 #endif /* __cplusplus */
 #endif /* __BLAS_INTERFACE_H__ */