We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 40f1a12 commit f13158eCopy full SHA for f13158e
py/torch_tensorrt/dynamo/conversion/plugins/_generate_plugin.py
@@ -73,10 +73,10 @@ def generate_signature(
73
elif arg.type.isSubtypeOf(torch._C.IntType.get()):
74
register_func_annotation[arg.name] = int
75
impl_func_annotation[arg.name] = int
76
- elif arg.type.isSubtypeOf(torch._C.Booltype.get()):
+ elif arg.type.isSubtypeOf(torch._C.BoolType.get()):
77
register_func_annotation[arg.name] = bool
78
impl_func_annotation[arg.name] = bool
79
- elif arg.type.isSubtypeOf(torch._C.Stringtype.get()):
+ elif arg.type.isSubtypeOf(torch._C.StringType.get()):
80
register_func_annotation[arg.name] = str
81
impl_func_annotation[arg.name] = str
82
else:
0 commit comments