File tree 1 file changed +3
-3
lines changed
1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -98,13 +98,13 @@ nvinfer1::ITensor* Var::ITensorOrFreeze(ConversionCtx* ctx) {
98
98
if (isIValue ()) {
99
99
auto tensor = ptr_.ivalue ->toTensor ();
100
100
if ((tensor.scalar_type () == at::kLong || tensor.scalar_type () == at::kDouble ) && !ctx->settings .truncate_long_and_double ) {
101
- TRTORCH_CHECK ( 0 , " Unable to freeze tensor of type kLong/kDouble into constant layer, try to compile model with truncate_long_and_double ON" );
101
+ TRTORCH_THROW_ERROR ( " Unable to freeze tensor of type Int64/Float64 into constant layer, try to compile model with truncate_long_and_double ON" );
102
102
} else if (tensor.scalar_type () == at::kLong && ctx->settings .truncate_long_and_double ) {
103
103
weights = converters::Weights (ctx, tensor.toType (at::kInt ));
104
- LOG_WARNING (" Warning: Truncating weight (constant in the graph) from kLong to kInt to indicate that only constants are affected ." );
104
+ LOG_WARNING (" Warning: Truncating weight (constant in the graph) from Int64 to Int32 ." );
105
105
} else if (tensor.scalar_type () == at::kDouble && ctx->settings .truncate_long_and_double ) {
106
106
weights = converters::Weights (ctx, tensor.toType (at::kFloat ));
107
- LOG_WARNING (" Warning: Truncating weight (constant in the graph) from kDouble to kFloat to indicate that only constants are affected ." );
107
+ LOG_WARNING (" Warning: Truncating weight (constant in the graph) from Float64 to Float32 ." );
108
108
} else {
109
109
weights = converters::Weights (ctx, tensor);
110
110
}
You can’t perform that action at this time.
0 commit comments