File tree Expand file tree Collapse file tree 1 file changed +4
-2
lines changed
py/torch_tensorrt/dynamo/conversion Expand file tree Collapse file tree 1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -219,15 +219,17 @@ def aten_ops_native_group_norm(
219219
220220
221221def cat_validator (node : Node , settings : Optional [CompilationSettings ] = None ) -> bool :
222- # Validate only one user, which is a getitem node that accesses the first element in the list
222+ # empty tensor in cat input as ITensor leads to [RemoveDeadLayers] Input Tensor y is unused or used only at compile-time, but is not being removed.
223223 for each_input in node .args [0 ]:
224224 if isinstance (each_input , TRTTensor ) and any (s == 0 for s in each_input .shape ):
225225 return False
226226 return True
227227
228228
229229@dynamo_tensorrt_converter (
230- torch .ops .aten .cat .default , supports_dynamic_shapes = True , validator = cat_validator
230+ torch .ops .aten .cat .default ,
231+ capability_validator = cat_validator ,
232+ supports_dynamic_shapes = True ,
231233)
232234def aten_ops_cat (
233235 ctx : ConversionContext ,
You can’t perform that action at this time.
0 commit comments