Fix tfrt.case: Prevent out-of-bounds access to a vector.

hanbinyoon · copybara-github · commit 0a00ce94f7a0 · 2022-04-26T08:59:13.000-07:00
PiperOrigin-RevId: 444573692
diff --git a/lib/basic_kernels/control_flow_kernels.cc b/lib/basic_kernels/control_flow_kernels.cc
@@ -74,6 +74,7 @@ static void TFRTCase(RemainingArguments args, RemainingResults results,
                        " # branches: ", branches.size()),
           tfrt::ErrorCode::kInvalidArgument);
       for (auto& result : results) result = error;
+      return;
     }
     branches[branch_index]->Execute(exec_ctx, args.drop_front(), results);
   };

Original file line number	Diff line number	Diff line change
`@@ -74,6 +74,7 @@ static void TFRTCase(RemainingArguments args, RemainingResults results,`
`74`	`74`	`" # branches: ", branches.size()),`
`75`	`75`	`tfrt::ErrorCode::kInvalidArgument);`
`76`	`76`	`for (auto& result : results) result = error;`
	`77`	`+ return;`
`77`	`78`	`}`
`78`	`79`	`branches[branch_index]->Execute(exec_ctx, args.drop_front(), results);`
`79`	`80`	`};`