BREAKING CHANGE: Change EngineCapability enum to use the new values introduced in TRT 8.0

Signed-off-by: Dheeraj Peri <[email protected]>

Author: peri044

cpp/api/include/trtorch/trtorch.h

@@ -262,9 +262,9 @@ struct TRTORCH_API CompileSpec {
    * Emum for selecting engine capability
    */
   enum class EngineCapability : int8_t {
-    kDEFAULT,
-    kSAFE_GPU,
-    kSAFE_DLA,
+    kSTANDARD,
+    kSAFETY,
+    kDLA_STANDALONE,
   };
 
   class TRTORCH_API TensorFormat {
@@ -721,7 +721,7 @@ struct TRTORCH_API CompileSpec {
   /**
    * Sets the restrictions for the engine (CUDA Safety)
    */
-  EngineCapability capability = EngineCapability::kDEFAULT;
+  EngineCapability capability = EngineCapability::kSTANDARD;
 
   /**
    * Number of minimization timing iterations used to select kernels

cpp/api/src/compile_spec.cpp

@@ -385,13 +385,13 @@ core::CompileSpec to_internal_compile_spec(CompileSpec external) {
   }
 
   switch (external.capability) {
-    case CompileSpec::EngineCapability::kSAFE_GPU:
+    case CompileSpec::EngineCapability::kSAFETY:
       internal.convert_info.engine_settings.capability = nvinfer1::EngineCapability::kSAFETY;
       break;
-    case CompileSpec::EngineCapability::kSAFE_DLA:
+    case CompileSpec::EngineCapability::kDLA_STANDALONE:
       internal.convert_info.engine_settings.capability = nvinfer1::EngineCapability::kDLA_STANDALONE;
       break;
-    case CompileSpec::EngineCapability::kDEFAULT:
+    case CompileSpec::EngineCapability::kSTANDARD:
     default:
       internal.convert_info.engine_settings.capability = nvinfer1::EngineCapability::kSTANDARD;
   }