Skip to content

Commit 48c338c

Browse files
committed
update signature
1 parent a0a9555 commit 48c338c

File tree

2 files changed

+7
-7
lines changed

2 files changed

+7
-7
lines changed

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2685,7 +2685,7 @@ def aten_ops_scalar_tensor(
26852685
)
26862686

26872687

2688-
@dynamo_tensorrt_converter(torch.log10)
2688+
@dynamo_tensorrt_converter(torch.ops.aten.log10.default)
26892689
def log10(
26902690
ctx: ConversionContext,
26912691
target: Target,

tests/py/dynamo/conversion/test_log10.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -16,13 +16,13 @@ class TestLogConverter(DispatchTestCase):
1616
]
1717
)
1818
def test_log10_float(self, input_shape, dtype):
19-
class log(nn.Module):
19+
class log10(nn.Module):
2020
def forward(self, input):
21-
return torch.log10(input)
21+
return torch.ops.aten.log10.default(input)
2222

2323
inputs = [torch.randn(input_shape, dtype=dtype)]
2424
self.run_test(
25-
log(),
25+
log10(),
2626
inputs,
2727
)
2828

@@ -34,13 +34,13 @@ def forward(self, input):
3434
]
3535
)
3636
def test_log10_int(self, input_shape, dtype, low, high):
37-
class log(nn.Module):
37+
class log10(nn.Module):
3838
def forward(self, input):
39-
return torch.log10(input)
39+
return torch.ops.aten.log10.default(input)
4040

4141
inputs = [torch.randint(low, high, input_shape, dtype=dtype)]
4242
self.run_test(
43-
log(),
43+
log10(),
4444
inputs,
4545
)
4646

0 commit comments

Comments
 (0)