Review comments- adding cases for stride, correcting validator and changing call to torch.ops.aten.empty.memory_format

Author: apbose

tests/py/dynamo/conversion/test_empty_aten.py

@@ -2,10 +2,11 @@
 import torch
 import torch.nn as nn
 import torch_tensorrt
-from .harness import DispatchTestCase
 from parameterized import parameterized
 from torch.testing._internal.common_utils import run_tests
 
+from .harness import DispatchTestCase
+
 empty_ops = [
     (
         "empty_one_dimension",
@@ -73,9 +74,12 @@
 ]
 
 
-class TestRandConverter(DispatchTestCase):
+class TestEmptyConverter(DispatchTestCase):
     @parameterized.expand(
-        [(empty_op[0], empty_op[1], empty_op[2], empty_op[3], empty_op[4]) for empty_op in empty_ops]
+        [
+            (empty_op[0], empty_op[1], empty_op[2], empty_op[3], empty_op[4])
+            for empty_op in empty_ops
+        ]
     )
     def test_empty(self, name, shape_or_input, data_type, device, memory_format):
         class TestModule(nn.Module):
@@ -84,7 +88,12 @@ def __init__(self):
 
             def forward(self, x):
                 shape_or_input[0] = x.shape[0]
-                return torch.ops.aten.empty.memory_format(shape_or_input, dtype = data_type, memory_format = layout, device = device)
+                return torch.ops.aten.empty.memory_format(
+                    shape_or_input,
+                    dtype=data_type,
+                    memory_format=memory_format,
+                    device=device,
+                )
 
         empty_model = TestModule()