update signature

Author: bowang007

py/torch_tensorrt/dynamo/conversion/aten_ops_converters.py

@@ -2672,7 +2672,7 @@ def aten_ops_flip(
     )
 
 
-@dynamo_tensorrt_converter(torch.log2)
+@dynamo_tensorrt_converter(torch.ops.aten.log2.default)
 def log2(
     ctx: ConversionContext,
     target: Target,

tests/py/dynamo/conversion/test_log2.py

@@ -16,13 +16,13 @@ class TestLogConverter(DispatchTestCase):
         ]
     )
     def test_log_float(self, input_shape, dtype):
-        class log(nn.Module):
+        class log2(nn.Module):
             def forward(self, input):
-                return torch.log2(input)
+                return torch.ops.aten.log2.default(input)
 
         inputs = [torch.randn(input_shape, dtype=dtype)]
         self.run_test(
-            log(),
+            log2(),
             inputs,
         )
 
@@ -34,13 +34,13 @@ def forward(self, input):
         ]
     )
     def test_log_int(self, input_shape, dtype, low, high):
-        class log(nn.Module):
+        class log2(nn.Module):
             def forward(self, input):
-                return torch.log2(input)
+                return torch.ops.aten.log2.default(input)
 
         inputs = [torch.randint(low, high, input_shape, dtype=dtype)]
         self.run_test(
-            log(),
+            log2(),
             inputs,
         )