Skip to content

Commit fabfc55

Browse files
authored
Merge pull request #2246 from zewenli98/unary_dynamo_converters
feat: support many unary dynamo converters
2 parents e58f319 + 36d221a commit fabfc55

26 files changed

+1924
-93
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

+398-24
Large diffs are not rendered by default.

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

+1-2
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,7 @@
11
from typing import Optional
22

33
import numpy as np
4+
import tensorrt as trt
45
from torch.fx.node import Target
56
from torch_tensorrt.dynamo._SourceIR import SourceIR
67
from torch_tensorrt.dynamo.conversion.impl.elementwise.base import (
@@ -16,8 +17,6 @@
1617
from torch_tensorrt.fx.types import TRTNetwork, TRTTensor
1718
from torch_tensorrt.fx.utils import Frameworks, unified_dtype_converter
1819

19-
import tensorrt as trt
20-
2120

2221
def trunc_div(
2322
network: TRTNetwork,

0 commit comments

Comments
 (0)