fix comment

PaddlePaddle · Oct 28, 2021 · 339ac8f · 339ac8f · paddle-bot-old · Oct 29, 2021
1 parent b312dcb
commit 339ac8f
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 9 deletions.
diff --git a/paddle/fluid/operators/math/matrix_solve.cc b/paddle/fluid/operators/math/matrix_solve.cc
@@ -51,8 +51,8 @@ class TriangularSolveFunctor<platform::CPUDeviceContext, T> {
     int a_dim_size = a->dims().size();
     int b_dim_size = b->dims().size();
 
-    int M = b->dims()[b_dim_size - 2];
-    int N = b->dims()[b_dim_size - 1];
+    int M = static_cast<int>(b->dims()[b_dim_size - 2]);
+    int N = static_cast<int>(b->dims()[b_dim_size - 1]);
     auto lda = left ? std::max(1, M) : std::max(1, N);
     auto ldb = std::max(1, N);
 

diff --git a/paddle/fluid/operators/math/matrix_solve.cu.cc b/paddle/fluid/operators/math/matrix_solve.cu.cc
@@ -180,8 +180,8 @@ class TriangularSolveFunctor<platform::CUDADeviceContext, T> {
     int a_dim_size = a->dims().size();
     int b_dim_size = b->dims().size();
 
-    int M = b->dims()[b_dim_size - 2];
-    int N = b->dims()[b_dim_size - 1];
+    int M = static_cast<int>(b->dims()[b_dim_size - 2]);
+    int N = static_cast<int>(b->dims()[b_dim_size - 1]);
     auto lda = left ? std::max(1, M) : std::max(1, N);
     auto ldb = std::max(1, N);
 

diff --git a/paddle/fluid/platform/dynload/mklml.h b/paddle/fluid/platform/dynload/mklml.h
@@ -20,10 +20,6 @@ limitations under the License. */
 #include "paddle/fluid/platform/dynload/dynamic_loader.h"
 #include "paddle/fluid/platform/port.h"
 
-extern "C" void strsm_(char *side, char *uplo, char *trans, char *diag, int *n,
-                       int *nrhs, float *alpha, float *a, int *lda, float *b,
-                       int *ldb);
-
 namespace paddle {
 namespace platform {
 namespace dynload {
@@ -53,7 +49,6 @@ extern void *mklml_dso_handle;
 #define DECLARE_DYNAMIC_LOAD_MKLML_WRAP(__name) DYNAMIC_LOAD_MKLML_WRAP(__name)
 
 #define MKLML_ROUTINE_EACH(__macro) \
-  __macro(strsm_);                  \
   __macro(cblas_sgemm);             \
   __macro(cblas_dgemm);             \
   __macro(cblas_cgemm);             \