Skip to content

Commit 7c8b436

Browse files
committed
fix bug
1 parent cd7808f commit 7c8b436

File tree

1 file changed

+3
-2
lines changed
  • py/torch_tensorrt/dynamo/conversion/impl/unary

1 file changed

+3
-2
lines changed

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

Lines changed: 3 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -9,7 +9,6 @@
99
cast_trt_tensor,
1010
get_trt_tensor,
1111
)
12-
from torch_tensorrt.dynamo.conversion.impl.elementwise import div
1312
from torch_tensorrt.dynamo.conversion.impl.unary.base import convert_unary
1413
from torch_tensorrt.fx.types import TRTTensor
1514

@@ -70,7 +69,9 @@ def log2(
7069

7170
ln2 = 0.693147180559945309
7271

73-
return div(ctx, target, source_ir, f"{name}_div", log_layer_output, ln2)
72+
return impl.elementwise.div(
73+
ctx, target, source_ir, f"{name}_div", log_layer_output, ln2
74+
)
7475

7576

7677
def sqrt(

0 commit comments

Comments
 (0)