Skip to content

Commit 4c61c1e

Browse files
committed
chore: remove comments and linting
1 parent 82a6514 commit 4c61c1e

File tree

3 files changed

+4
-102
lines changed

3 files changed

+4
-102
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,6 +7,7 @@
77
import numpy as np
88
import torch
99
from torch.fx.node import Argument, Node, Target
10+
1011
from torch_tensorrt.dynamo._settings import CompilationSettings
1112
from torch_tensorrt.dynamo._SourceIR import SourceIR
1213
from torch_tensorrt.dynamo.conversion import impl
@@ -2454,8 +2455,7 @@ def aten_ops_le(
24542455
def conv_param_validator(
24552456
conv_node: Node, settings: Optional[CompilationSettings] = None
24562457
) -> bool:
2457-
2458-
# return True
2458+
24592459
return conv_node.args[7] in ([0], [0, 0], [0, 0, 0])
24602460

24612461

py/torch_tensorrt/fx/converters/aten_ops_converters.py

Lines changed: 2 additions & 57 deletions
Original file line numberDiff line numberDiff line change
@@ -10,9 +10,10 @@
1010
# @manual=//deeplearning/trt/python:py_tensorrt
1111
import tensorrt as trt
1212
import torch
13-
import torch_tensorrt.fx.tracer.acc_tracer.acc_utils as acc_utils
1413
from torch.fx.immutable_collections import immutable_list
1514
from torch.fx.node import Argument, Target
15+
16+
import torch_tensorrt.fx.tracer.acc_tracer.acc_utils as acc_utils
1617
from torch_tensorrt.fx.converters import acc_ops_converters
1718
from torch_tensorrt.fx.converters.impl import activation, convolution
1819

@@ -103,62 +104,6 @@ def aten_ops_batch_norm(
103104
)
104105

105106

106-
# @tensorrt_converter(torch.ops.aten.convolution.default)
107-
# def aten_ops_convolution(
108-
# network: TRTNetwork,
109-
# target: Target,
110-
# args: Tuple[Argument, ...],
111-
# kwargs: Dict[str, Argument],
112-
# name: str,
113-
# ) -> Union[TRTTensor, Sequence[TRTTensor]]:
114-
# kwargs_new = {
115-
# "input": args[0],
116-
# "weight": args[1],
117-
# "bias": args[2],
118-
# "stride": args[3],
119-
# "padding": args[4],
120-
# "dilation": args[5],
121-
# "groups": args[8],
122-
# }
123-
# # we do not handle transposed.
124-
# if args[6] is True:
125-
# raise RuntimeError(f"Target {target} does not support `transposed=True` ")
126-
# # we do not handle output_padding.
127-
# if args[7] not in ([0], [0, 0], [0, 0, 0]):
128-
# raise RuntimeError(f"Target {target} has non-0 output_padding")
129-
130-
# if len(kwargs_new["stride"]) == 1:
131-
# return convolution.convNd(
132-
# network,
133-
# target,
134-
# source_ir=SourceIR.ATEN,
135-
# name=name,
136-
# is_conv1d=True,
137-
# input_val=kwargs_new["input"],
138-
# weight=kwargs_new["weight"],
139-
# bias=kwargs_new["bias"],
140-
# stride=kwargs_new["stride"],
141-
# padding=kwargs_new["padding"],
142-
# dilation=kwargs_new["dilation"],
143-
# groups=kwargs_new["groups"],
144-
# )
145-
# else:
146-
# return convolution.convNd(
147-
# network,
148-
# target,
149-
# source_ir=SourceIR.ATEN,
150-
# name=name,
151-
# is_conv1d=False,
152-
# input_val=kwargs_new["input"],
153-
# weight=kwargs_new["weight"],
154-
# bias=kwargs_new["bias"],
155-
# stride=kwargs_new["stride"],
156-
# padding=kwargs_new["padding"],
157-
# dilation=kwargs_new["dilation"],
158-
# groups=kwargs_new["groups"],
159-
# )
160-
161-
162107
@tensorrt_converter(torch.ops.aten.div.default)
163108
@tensorrt_converter(torch.ops.aten.div.Tensor_mode)
164109
@tensorrt_converter(torch.ops.aten.div.Tensor)

run_aot_moria_2dunet.py

Lines changed: 0 additions & 43 deletions
This file was deleted.

0 commit comments

Comments
 (0)