Skip to content

Commit 68b02e5

Browse files
committed
Changing the logging while casting in convolution from debug print to warning
1 parent 2c85f89 commit 68b02e5

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

core/conversion/converters/impl/conv_deconv.cpp

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -14,9 +14,9 @@ bool add_conv_deconv(ConversionCtx* ctx, const torch::jit::Node* n, args& args)
1414
// Input to conv/deconv
1515
auto in = args[0].ITensor();
1616
if (in->getType() == nvinfer1::DataType::kINT32) {
17-
LOG_DEBUG(
18-
"Found type " << in->getType() << " in aten::convolution, casting to "
19-
<< nvinfer1::DataType::kFLOAT << " for compatibility.");
17+
LOG_WARNING(
18+
"Found type " << in->getType() << "in aten::convolution, casting to" << nvinfer1::DataType::kFLOAT
19+
<< " for compatibility.");
2020
in = castITensor(ctx, in, nvinfer1::DataType::kFLOAT);
2121
}
2222
// Conv /deconv parameters

0 commit comments

Comments
 (0)