Skip to content

Commit 622404b

Browse files
committed
correcting the validator error message
1 parent 0f15e7d commit 622404b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -219,15 +219,17 @@ def aten_ops_native_group_norm(
219219

220220

221221
def 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
)
232234
def aten_ops_cat(
233235
ctx: ConversionContext,

0 commit comments

Comments
 (0)