Skip to content

Commit 507a8e8

Browse files
committed
Removing empty_like from evaluator and adding it to lowering pass
1 parent 002b35d commit 507a8e8

File tree

3 files changed

+1
-51
lines changed

3 files changed

+1
-51
lines changed

py/torch_tensorrt/dynamo/conversion/ops_evaluators.py

Lines changed: 0 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -47,14 +47,3 @@ def aten_ops_arange_start_step(
4747
name: str,
4848
) -> Union[TRTTensor, Sequence[TRTTensor]]:
4949
return np.arange(*args)
50-
51-
52-
@dynamo_tensorrt_converter(torch.ops.aten.empty_like.default)
53-
def aten_ops_empty_like(
54-
ctx: ConversionContext,
55-
target: Target,
56-
args: Tuple[Argument, ...],
57-
kwargs: Dict[str, Argument],
58-
name: str,
59-
) -> TRTTensor:
60-
return np.empty_like(*args)

py/torch_tensorrt/dynamo/lowering/_decomposition_groups.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -37,6 +37,7 @@
3737
aten.elu_backward,
3838
aten._embedding_bag,
3939
aten.embedding_dense_backward,
40+
aten.empty_like,
4041
aten._euclidean_dist.default,
4142
aten.expand_as,
4243
aten.eye,

tests/py/dynamo/conversion/test_empty_like.py

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

0 commit comments

Comments
 (0)