Skip to content

Commit 521fed1

Browse files
committed
Temporarily removing rsub pending fix
1 parent 9ac3898 commit 521fed1

File tree

3 files changed

+0
-129
lines changed

3 files changed

+0
-129
lines changed

py/torch_tensorrt/dynamo/converters/aten_ops_converters.py

Lines changed: 0 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -196,22 +196,6 @@ def aten_ops_unsqueeze(
196196
)
197197

198198

199-
@dynamo_tensorrt_converter(torch.ops.aten.rsub.Tensor)
200-
def aten_ops_rsub(
201-
network: TRTNetwork,
202-
target: Target,
203-
args: Tuple[Argument, ...],
204-
kwargs: Dict[str, Argument],
205-
name: str,
206-
) -> Union[TRTTensor, Sequence[TRTTensor]]:
207-
alpha = None
208-
if "alpha" in kwargs:
209-
alpha = kwargs["alpha"]
210-
return impl.elementwise.rsub(
211-
network, target, SourceIR.ATEN, name, args[0], args[1], alpha
212-
)
213-
214-
215199
@dynamo_tensorrt_converter(torch.ops.aten._softmax.default)
216200
def aten_ops_softmax(
217201
network: TRTNetwork,

py/torch_tensorrt/dynamo/converters/impl/elementwise/ops.py

Lines changed: 0 additions & 34 deletions
Original file line numberDiff line numberDiff line change
@@ -139,40 +139,6 @@ def rsqrt(
139139
return output
140140

141141

142-
def rsub(
143-
network: TRTNetwork,
144-
target: Target,
145-
source_ir: Optional[SourceIR],
146-
name: str,
147-
input: TRTTensor,
148-
other: TRTTensor,
149-
alpha: Optional[Any] = None,
150-
) -> TRTTensor:
151-
scaled_tensor_output = other
152-
if alpha is not None:
153-
scaled_tensor_output = convert_binary_elementwise(
154-
network,
155-
target,
156-
source_ir,
157-
f"{name}_scaled",
158-
trt.ElementWiseOperation.PROD,
159-
other,
160-
alpha,
161-
)
162-
163-
output = convert_binary_elementwise(
164-
network,
165-
target,
166-
source_ir,
167-
f"{name}_output",
168-
trt.ElementWiseOperation.SUB,
169-
input,
170-
scaled_tensor_output,
171-
)
172-
173-
return output
174-
175-
176142
def fmod(
177143
network: TRTNetwork,
178144
target: Target,

tests/py/dynamo/converters/test_rsub_aten.py

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

0 commit comments

Comments
 (0)