Fix compiler error with_trt (#47716)

6934ae2b · Wilber · GitHub · 84c9a0d6 · 6934ae2b · 6934ae2b
显示空白变更内容
内联并排

Showing with 5 addition and 2 deletion

cmake/flags.cmake cmake/flags.cmake +3 -1

paddle/fluid/inference/tensorrt/plugin/common/bertCommon.h paddle/fluid/inference/tensorrt/plugin/common/bertCommon.h +2 -1

未找到文件。
--- a/cmake/flags.cmake
+++ b/cmake/flags.cmake
@@ -257,7 +257,9 @@ if(WITH_ROCM)
  string(REPLACE "-Werror" "-Wno-error" CMAKE_C_FLAGS ${CMAKE_C_FLAGS})
 endif()

-if(WITH_PSCORE OR WITH_PSLIB)
+if(WITH_PSCORE
+   OR WITH_PSLIB
+   OR WITH_TENSORRT)
  string(REPLACE "-Wnon-virtual-dtor" "-Wno-non-virtual-dtor" CMAKE_CXX_FLAGS
                 ${CMAKE_CXX_FLAGS})
  string(REPLACE "-Wnon-virtual-dtor" "-Wno-non-virtual-dtor" CMAKE_C_FLAGS

--- a/paddle/fluid/inference/tensorrt/plugin/common/bertCommon.h
+++ b/paddle/fluid/inference/tensorrt/plugin/common/bertCommon.h
@@ -82,8 +82,9 @@ inline uint32_t getElementSize(nvinfer1::DataType t) noexcept {
    case nvinfer1::DataType::kBOOL:
    case nvinfer1::DataType::kINT8:
      return 1;
-  }
+    default:
      return 0;
+  }
 }

 inline int64_t getWeightsSize(const nvinfer1::Weights& w,