chore(pooling|interpolate): More specific warnings when plugin is used

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

core/conversion/converters/impl/interpolate.cpp

@@ -21,7 +21,7 @@ void create_plugin(ConversionCtx* ctx, const torch::jit::Node* n, nvinfer1::ITen
                                                                                          std::vector<int64_t> out_shape,
                                                                                          std::vector<int64_t> out_size,
                                                                                          std::string mode) {
-    LOG_WARNING("Interpolation layer will be run through ATen, not TensorRT. Performance may differ.");
+    LOG_WARNING("Interpolation layer will be run through ATen, not TensorRT. Performance may be lower than expected");
 
     auto creator = new plugins::InterpolatePluginCreator();
     auto plugin = creator->createPlugin(name, in_shape, out_shape, out_size, mode, false);

core/conversion/converters/impl/pooling.cpp

@@ -279,7 +279,11 @@ auto pooling_registrations TRTORCH_UNUSED = RegisterNodeConversionPatterns()
             auto out_size = util::toVec(util::toDims(args[1].unwrapToIntList()));
 
             if (ctx->input_is_dynamic) {
+#if NV_TENSORRT_MAJOR < 7 || (NV_TENSORRT_MAJOR == 7 && NV_TENSORRT_MINOR < 1)
+                LOG_WARNING("Adaptive pooling layer will be run through ATen, via not TensorRT, performace will be lower than expected. Consider switching either to static input shape or moving to non adaptive pooling if this is an issue");
+#else
                 LOG_WARNING("Adaptive pooling layer will be run through ATen (on CPU), via not TensorRT, performace will suffer. Consider switching either to static input shape or moving to non adaptive pooling");
+#endif
 
                 auto out_shape = in_shape;
                 std::copy(out_size.begin(), out_size.end(), out_shape.begin() + (in_shape.size() - out_size.size()));