Skip to content

Commit f628c0c

Browse files
committed
add decorators
1 parent 4f585d8 commit f628c0c

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -57,6 +57,7 @@ def one_user_validator(node: Node) -> bool:
5757

5858
@dynamo_tensorrt_converter(torch.ops.aten.native_batch_norm.default, capability_validator=one_user_validator) # type: ignore[misc]
5959
@dynamo_tensorrt_converter(torch.ops.aten.batch_norm.default) # type: ignore[misc]
60+
@dynamo_tensorrt_converter(torch.ops.aten.batch_norm) # type: ignore[misc]
6061
@enforce_tensor_types(
6162
{
6263
0: (TRTTensor,),
@@ -89,6 +90,7 @@ def aten_ops_batch_norm(
8990

9091
@dynamo_tensorrt_converter(torch.ops.aten.native_layer_norm.default, capability_validator=one_user_validator) # type: ignore[misc]
9192
@dynamo_tensorrt_converter(torch.ops.aten.layer_norm.default) # type: ignore[misc]
93+
@dynamo_tensorrt_converter(torch.ops.aten.layer_norm) # type: ignore[misc]
9294
@enforce_tensor_types(
9395
{
9496
0: (TRTTensor,),
@@ -146,6 +148,7 @@ def aten_ops_native_group_norm(
146148

147149

148150
@dynamo_tensorrt_converter(torch.ops.aten.group_norm.default) # type: ignore[misc]
151+
@dynamo_tensorrt_converter(torch.ops.aten.group_norm) # type: ignore[misc]
149152
@enforce_tensor_types(
150153
{
151154
0: (TRTTensor,),

0 commit comments

Comments
 (0)