fix: Update the call

Signed-off-by: Anurag Dixit <a.dixit91@gmail.com>
pytorch · May 10, 2022 · 99d1f32 · 99d1f32
1 parent f36a8c6
commit 99d1f32
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/conversion/converters/impl/select.cpp b/core/conversion/converters/impl/select.cpp
@@ -352,7 +352,7 @@ auto select_registrations TORCHTRT_UNUSED =
                   }})
         .pattern({"aten::split.sizes(Tensor(a -> *) self, int[] split_size, int dim=0) -> (Tensor[])",
                   [](ConversionCtx* ctx, const torch::jit::Node* n, args& args) -> bool {
-                    add_split(ctx, n, args, true);
+                    add_split(ctx, n, args, true, false);
                     LOG_DEBUG("Converted split op into a list of IValues");
                     return true;
                   }})