chore: rebase and apply linting

Signed-off-by: Naren Dasan <[email protected]>
Signed-off-by: Naren Dasan <[email protected]>

Author: narendasan

.pre-commit-config.yaml

@@ -6,6 +6,7 @@ repos:
       - id: trailing-whitespace
       - id: check-added-large-files
       - id: check-vcs-permalinks
+      - id: check-merge-conflict
       - id: mixed-line-ending
         args:
           - --fix=lf

core/conversion/converters/impl/conv_deconv.cpp

@@ -194,7 +194,7 @@ bool add_conv_deconv(ConversionCtx* ctx, const torch::jit::Node* n, args& args)
 
       nvinfer1::Dims stride;
       stride.nbDims = in_nbDims;
-      for (size_t i = 0; i < in_nbDims; i++) {
+      for (int64_t i = 0; i < in_nbDims; i++) {
         stride.d[i] = 1;
       }
       const auto& dummy = stride;
@@ -206,7 +206,7 @@ bool add_conv_deconv(ConversionCtx* ctx, const torch::jit::Node* n, args& args)
 
       nvinfer1::Dims constantDims;
       constantDims.nbDims = in_nbDims;
-      for (size_t i = 0; i < in_nbDims; i++) {
+      for (int64_t i = 0; i < in_nbDims; i++) {
         constantDims.d[i] = 1;
       }
       constantDims.d[diff - 1] =

core/ir/BUILD

@@ -13,9 +13,13 @@ config_setting(
 cc_library(
     name = "ir",
     srcs = [
+        "GraphInputs.cpp",
         "Input.cpp",
         "StaticParams.cpp",
-        "GraphInputs.cpp"
+        "ir.cpp",
+    ],
+    hdrs = [
+        "ir.h",
     ],
     deps = [
         "@tensorrt//:nvinfer",

tests/cpp/BUILD

@@ -12,6 +12,7 @@ config_setting(
 test_suite(
     name = "api_tests",
     tests = [
+        ":test_collections",
         ":test_compiled_modules",
         ":test_default_input_types",
         ":test_example_tensors",
@@ -20,15 +21,13 @@ test_suite(
         ":test_multiple_registered_engines",
         ":test_runtime_thread_safety",
         ":test_serialization",
-        ":test_module_fallback",
-        ":test_example_tensors",
-        ":test_collections"
     ],
 )
 
 test_suite(
     name = "aarch64_api_tests",
     tests = [
+        ":test_collections",
         ":test_compiled_modules",
         ":test_default_input_types",
         ":test_example_tensors",
@@ -37,9 +36,6 @@ test_suite(
         ":test_multiple_registered_engines",
         ":test_runtime_thread_safety",
         ":test_serialization",
-        ":test_module_fallback",
-        ":test_example_tensors",
-        ":test_collections"
     ],
 )
 
@@ -141,8 +137,9 @@ cc_test(
     ] + select({
         ":use_pre_cxx11_abi": ["@libtorch_pre_cxx11_abi//:libtorch"],
         "//conditions:default": ["@libtorch//:libtorch"],
-    })
+    }),
 )
+
 cc_test(
     name = "test_compiled_modules",
     srcs = ["test_compiled_modules.cpp"],