refactor: apply linting

Signed-off-by: Naren Dasan <[email protected]> Signed-off-by: Naren Dasan <[email protected]>
pytorch · Aug 7, 2021 · 2cd9fad · 2cd9fad
1 parent 88e9335
commit 2cd9fad
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 4 deletions.
diff --git a/core/conversion/conversionctx/ConversionCtx.cpp b/core/conversion/conversionctx/ConversionCtx.cpp
@@ -70,7 +70,8 @@ ConversionCtx::ConversionCtx(BuilderSettings build_settings)
         TRTORCH_CHECK(builder->platformHasFastInt8(), "Requested inference in INT8 but platform does not support INT8");
         cfg->setFlag(nvinfer1::BuilderFlag::kINT8);
         if (settings.calibrator == nullptr) {
-          LOG_INFO("INT8 kernels are enabled but not calibrator was provided, assuming source model was trained quantization aware");
+          LOG_INFO(
+              "INT8 kernels are enabled but not calibrator was provided, assuming source model was trained quantization aware");
         }
         break;
       case nvinfer1::DataType::kFLOAT:

diff --git a/core/util/trt_util.h b/core/util/trt_util.h
@@ -88,11 +88,11 @@ inline std::ostream& operator<<(std::ostream& stream, const nvinfer1::DeviceType
 inline std::ostream& operator<<(std::ostream& stream, const nvinfer1::EngineCapability& cap) {
   switch (cap) {
     case nvinfer1::EngineCapability::kSTANDARD:
-      return stream << "Default";
+      return stream << "standard";
     case nvinfer1::EngineCapability::kSAFETY:
-      return stream << "Safe GPU";
+      return stream << "safety";
     case nvinfer1::EngineCapability::kDLA_STANDALONE:
-      return stream << "Safe DLA";
+      return stream << "DLA standalone";
     default:
       return stream << "Unknown Engine Capability Setting";
   }