[functorch] Revert "Exclude List[Optional[Tensor]] from the batched fallback"
This reverts commit 3f5f75208aa547a2f4ed23fd281c917d327b9819.
diff --git a/functorch/functorch/csrc/BatchedFallback.cpp b/functorch/functorch/csrc/BatchedFallback.cpp
index 8d3d8c6..73b3675 100644
--- a/functorch/functorch/csrc/BatchedFallback.cpp
+++ b/functorch/functorch/csrc/BatchedFallback.cpp
@@ -57,13 +57,7 @@
return std::any_of(
schema.arguments().begin(),
schema.arguments().end(),
- [] (const Argument& arg) {
- static auto ListOfOptionalTensors = ListType::create(OptionalType::ofTensor());
- if (arg.type()->isSubtypeOf(ListType::ofTensors())) {
- return true;
- }
- return arg.type()->isSubtypeOf(ListOfOptionalTensors);
- });
+ [] (const Argument& arg) { return arg.type()->isSubtypeOf(ListType::ofTensors()); });
}
// Returns if an operator is in-place. An operator is inplace if:
