Skip to content

Commit 2cd9fad

Browse files
committed
refactor: apply linting
Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
1 parent 88e9335 commit 2cd9fad

File tree

2 files changed

+5
-4
lines changed

2 files changed

+5
-4
lines changed

core/conversion/conversionctx/ConversionCtx.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -70,7 +70,8 @@ ConversionCtx::ConversionCtx(BuilderSettings build_settings)
7070
TRTORCH_CHECK(builder->platformHasFastInt8(), "Requested inference in INT8 but platform does not support INT8");
7171
cfg->setFlag(nvinfer1::BuilderFlag::kINT8);
7272
if (settings.calibrator == nullptr) {
73-
LOG_INFO("INT8 kernels are enabled but not calibrator was provided, assuming source model was trained quantization aware");
73+
LOG_INFO(
74+
"INT8 kernels are enabled but not calibrator was provided, assuming source model was trained quantization aware");
7475
}
7576
break;
7677
case nvinfer1::DataType::kFLOAT:

core/util/trt_util.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -88,11 +88,11 @@ inline std::ostream& operator<<(std::ostream& stream, const nvinfer1::DeviceType
8888
inline std::ostream& operator<<(std::ostream& stream, const nvinfer1::EngineCapability& cap) {
8989
switch (cap) {
9090
case nvinfer1::EngineCapability::kSTANDARD:
91-
return stream << "Default";
91+
return stream << "standard";
9292
case nvinfer1::EngineCapability::kSAFETY:
93-
return stream << "Safe GPU";
93+
return stream << "safety";
9494
case nvinfer1::EngineCapability::kDLA_STANDALONE:
95-
return stream << "Safe DLA";
95+
return stream << "DLA standalone";
9696
default:
9797
return stream << "Unknown Engine Capability Setting";
9898
}

0 commit comments

Comments
 (0)