Skip to content

Commit f13158e

Browse files
fix the typo (#3773)
1 parent 40f1a12 commit f13158e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -73,10 +73,10 @@ def generate_signature(
7373
elif arg.type.isSubtypeOf(torch._C.IntType.get()):
7474
register_func_annotation[arg.name] = int
7575
impl_func_annotation[arg.name] = int
76-
elif arg.type.isSubtypeOf(torch._C.Booltype.get()):
76+
elif arg.type.isSubtypeOf(torch._C.BoolType.get()):
7777
register_func_annotation[arg.name] = bool
7878
impl_func_annotation[arg.name] = bool
79-
elif arg.type.isSubtypeOf(torch._C.Stringtype.get()):
79+
elif arg.type.isSubtypeOf(torch._C.StringType.get()):
8080
register_func_annotation[arg.name] = str
8181
impl_func_annotation[arg.name] = str
8282
else:

0 commit comments

Comments
 (0)