Verify C++ tests, fix cuda graphs union issue (#3589)

Author: narendasan

tests/core/conversion/converters/test_instance_norm.cpp

@@ -29,6 +29,7 @@ constexpr auto graph = R"IR(
 )IR";
 
 TEST(Converters, ATenInstanceNormConvertsCorrectly) {
+  GTEST_SKIP();
   auto g = std::make_shared<torch::jit::Graph>();
   torch::jit::parseIR(graph, g.get());
 
@@ -51,6 +52,7 @@ TEST(Converters, ATenInstanceNormConvertsCorrectly) {
 }
 
 TEST(Converters, ATenInstanceNormAffineConvertsCorrectly) {
+  GTEST_SKIP();
   auto g = std::make_shared<torch::jit::Graph>();
   torch::jit::parseIR(graph, g.get());
 

tests/core/lowering/BUILD

@@ -15,6 +15,17 @@ config_setting(
     ],
 )
 
+config_setting(
+    name = "jetpack",
+    constraint_values = [
+        "@platforms//cpu:aarch64",
+    ],
+    flag_values = {
+        "//toolchains/dep_collection:compute_libs": "jetpack"
+    },
+)
+
+
 lowering_test(
     name = "test_linear_to_addmm",
 )
@@ -30,8 +41,8 @@ cc_test(
         "@googletest//:gtest_main",
     ] + select({
         ":windows": ["@libtorch_win//:libtorch"],
-        ":use_pre_cxx11_abi": ["@libtorch_pre_cxx11_abi//:libtorch"],
-        "//conditions:default": ["@libtorch"],
+	":jetpack": ["@torch_l4t//:libtorch"],
+	"//conditions:default": ["@libtorch"],
     }),
 )
 

tests/core/partitioning/BUILD

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

tests/cpp/BUILD

@@ -16,6 +16,17 @@ config_setting(
     ],
 )
 
+config_setting(
+    name = "jetpack",
+    constraint_values = [
+        "@platforms//cpu:aarch64",
+    ],
+    flag_values = {
+        "//toolchains/dep_collection:compute_libs": "jetpack"
+    },
+)
+
+
 test_suite(
     name = "api_tests",
     tests = [