Skip to content

Commit 99d1f32

Browse files
committed
fix: Update the call
Signed-off-by: Anurag Dixit <[email protected]>
1 parent f36a8c6 commit 99d1f32

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

core/conversion/converters/impl/select.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -352,7 +352,7 @@ auto select_registrations TORCHTRT_UNUSED =
352352
}})
353353
.pattern({"aten::split.sizes(Tensor(a -> *) self, int[] split_size, int dim=0) -> (Tensor[])",
354354
[](ConversionCtx* ctx, const torch::jit::Node* n, args& args) -> bool {
355-
add_split(ctx, n, args, true);
355+
add_split(ctx, n, args, true, false);
356356
LOG_DEBUG("Converted split op into a list of IValues");
357357
return true;
358358
}})

0 commit comments

Comments
 (0)