Merge pull request #1609 from pytorch/convolution_cast

Convolution cast

Author: apbose

core/conversion/converters/impl/conv_deconv.cpp

@@ -13,7 +13,12 @@ namespace {
 bool add_conv_deconv(ConversionCtx* ctx, const torch::jit::Node* n, args& args) {
   // Input to conv/deconv
   auto in = args[0].ITensor();
-
+  if (in->getType() == nvinfer1::DataType::kINT32) {
+    LOG_WARNING(
+        "Found type  " << in->getType() << "in aten::convolution, casting to" << nvinfer1::DataType::kFLOAT
+                       << " for compatibility.");
+    in = castITensor(ctx, in, nvinfer1::DataType::kFLOAT);
+  }
   // Conv /deconv parameters
   auto stride = util::toDims(args[3].unwrapToIntList());
   auto padding = util::toDims(args[4].unwrapToIntList());