fix: Small fix to the evaluator set + updating some test infra (#3904)

Author: narendasan

.bazelversion

@@ -1 +1 @@
-8.1.1
+8.4.2

core/conversion/evaluators/aten.cpp

@@ -948,7 +948,7 @@ auto aten_registrations TORCHTRT_UNUSED =
                  stack.push_back(*args.at(v).IValue());
                }
                stack.push_back(input_num);
-               auto& ops = torch::jit::getAllOperatorsFor(c10::Symbol::fromQualString("aten::format"));
+               auto ops = torch::jit::getAllOperatorsFor(c10::Symbol::fromQualString("aten::format"));
                auto& aten_format = ops.front();
                aten_format->getOperation()(stack);
                std::string output;

tests/core/conversion/converters/BUILD

@@ -14,6 +14,16 @@ config_setting(
     ],
 )
 
+config_setting(
+    name = "jetpack",
+    constraint_values = [
+        "@platforms//cpu:aarch64",
+    ],
+    flag_values = {
+        "//toolchains/dep_collection:compute_libs": "jetpack",
+    },
+)
+
 converter_test(
     name = "test_activation",
 )

tests/core/conversion/converters/converter_test.bzl

@@ -19,8 +19,10 @@ def converter_test(name, visibility = None):
             "//tests/util",
             "@googletest//:gtest_main",
         ] + select({
+            ":jetpack": ["@torch_l4t//:libtorch"],
+            ":use_torch_whl": ["@torch_whl//:libtorch"],
             ":windows": ["@libtorch_win//:libtorch"],
-            "//conditions:default": ["@libtorch//:libtorch"],
+            "//conditions:default": ["@libtorch"],
         }),
         timeout = "moderate",
     )

tests/core/conversion/evaluators/BUILD

@@ -14,6 +14,16 @@ config_setting(
     ],
 )
 
+config_setting(
+    name = "jetpack",
+    constraint_values = [
+        "@platforms//cpu:aarch64",
+    ],
+    flag_values = {
+        "//toolchains/dep_collection:compute_libs": "jetpack",
+    },
+)
+
 evaluator_test(
     name = "test_prim_evaluators",
 )

tests/core/conversion/evaluators/evaluator_test.bzl

@@ -19,8 +19,10 @@ def evaluator_test(name, visibility = None):
             "//tests/util",
             "@googletest//:gtest_main",
         ] + select({
+            ":jetpack": ["@torch_l4t//:libtorch"],
+            ":use_torch_whl": ["@torch_whl//:libtorch"],
             ":windows": ["@libtorch_win//:libtorch"],
-            "//conditions:default": ["@libtorch//:libtorch"],
+            "//conditions:default": ["@libtorch"],
         }),
         timeout = "long",
     )

tests/core/lowering/BUILD

@@ -40,9 +40,10 @@ cc_test(
         "//tests/util",
         "@googletest//:gtest_main",
     ] + select({
+        ":jetpack": ["@torch_l4t//:libtorch"],
+        ":use_torch_whl": ["@torch_whl//:libtorch"],
         ":windows": ["@libtorch_win//:libtorch"],
-	":jetpack": ["@torch_l4t//:libtorch"],
-	"//conditions:default": ["@libtorch"],
+        "//conditions:default": ["@libtorch"],
     }),
 )
 

tests/core/lowering/lowering_test.bzl

@@ -19,9 +19,9 @@ def lowering_test(name, visibility = None):
             "//tests/util",
             "@googletest//:gtest_main",
         ] + select({
-            ":windows": ["@libtorch_win//:libtorch"],
-            ":use_torch_whl": ["@torch_whl//:libtorch"],
             ":jetpack": ["@torch_l4t//:libtorch"],
+            ":use_torch_whl": ["@torch_whl//:libtorch"],
+            ":windows": ["@libtorch_win//:libtorch"],
             "//conditions:default": ["@libtorch"],
         }),
         timeout = "short",

tests/core/partitioning/partitioning_test.bzl

@@ -19,10 +19,10 @@ def partitioning_test(name, visibility = None):
             "//tests/util",
             "@googletest//:gtest_main",
         ] + select({
-            ":windows": ["@libtorch_win//:libtorch"],
-            ":use_torch_whl": ["@torch_whl//:libtorch"],
             ":jetpack": ["@torch_l4t//:libtorch"],
+            ":use_torch_whl": ["@torch_whl//:libtorch"],
+            ":windows": ["@libtorch_win//:libtorch"],
             "//conditions:default": ["@libtorch"],
-        }),
+        })
         #timeout = "short",
     )

tests/core/runtime/runtime_test.bzl

@@ -19,9 +19,9 @@ def runtime_test(name, visibility = None):
             "//tests/util",
             "@googletest//:gtest_main",
         ] + select({
-            ":windows": ["@libtorch_win//:libtorch"],
-            ":use_torch_whl": ["@torch_whl//:libtorch"],
             ":jetpack": ["@torch_l4t//:libtorch"],
+            ":use_torch_whl": ["@torch_whl//:libtorch"],
+            ":windows": ["@libtorch_win//:libtorch"],
             "//conditions:default": ["@libtorch"],
         }),
     )