Skip to content

Commit bee0f1d

Browse files
committed
correcting the validator error message
1 parent b7f2dee commit bee0f1d

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -218,7 +218,7 @@ def aten_ops_native_group_norm(
218218

219219

220220
def cat_validator(node: Node, settings: Optional[CompilationSettings] = None) -> bool:
221-
# Validate only one user, which is a getitem node that accesses the first element in the list
221+
# 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.
222222
for each_input in node.args[0]:
223223
if isinstance(each_input, TRTTensor) and any(s == 0 for s in each_input.shape):
224224
return False

0 commit comments

Comments
 (0)