fix: Fix tests and corner cases for resolving non tensor inputs in fallback

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

Author: peri044

core/partitioning/partitioning.cpp

@@ -44,7 +44,7 @@ bool isModifyingNodes(torch::jit::Node* node, torch::jit::Value* val) {
   for (size_t i = 0; i < node->inputs().size(); ++i) {
     if (node->inputs()[i] == val) {
       const at::AliasInfo* formal = schema.arguments()[i].alias_info();
-      if (formal->isWrite()) {
+      if (formal && formal->isWrite()) {
         return true;
       }
     }

tests/core/partitioning/test_resolve_nontensor_inputs.cpp

@@ -257,7 +257,7 @@ TEST(Partitioning, ConvertForTensorListInputsInFallbackCorrectly) {
   torch::jit::script::Module new_mod = torch_tensorrt::core::CompileGraph(mod, cfg);
   auto fallback_g = new_mod.get_method("forward").graph();
   int count = count_trt_engines(fallback_g);
-  ASSERT_TRUE(count == 2);
+  ASSERT_TRUE(count == 1);
 }
 
 TEST(Partitioning, ResolveOnlyNeccessaryNonTensorInputs) {