Skip to content

Commit 1b13159

Browse files
committed
small fix: Index validator enable int64
1 parent e38a7f3 commit 1b13159

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
@@ -397,7 +397,7 @@ def index_dtype_validator(node: Node) -> bool:
397397
for ind in index:
398398
if ind is not None:
399399
val = ind.meta.get("val")
400-
if val is not None and val.dtype != torch.int32:
400+
if val is not None and val.dtype not in (torch.int32, torch.int64):
401401
return False
402402
return True
403403

0 commit comments

Comments
 (0)