addressed the comments

Author: sharavak

lib/Dialect/Torch/Transforms/DecomposeComplexOps.cpp

@@ -11069,7 +11069,7 @@ class DecomposeAtenSgnOp : public OpRewritePattern<AtenSgnOp> {
 } // namespace
 
 namespace {
-// Decomposes  aten.heaviside op into
+// Decomposed aten.heaviside op into
 // using aten.eq, aten.lt, aten.logical_or, aten.where
 // Heaviside(x, y) returns:
 //  0 if x < 0

projects/pt1/python/torch_mlir_e2e_test/test_suite/elementwise_comparison.py

@@ -320,16 +320,18 @@ def __init__(self):
         super().__init__()
 
     @export
-    @annotate_args([None, ([-1, -1], torch.int32, True), ([-1], torch.int32, True)])
+    @annotate_args(
+        [None, ([-1, -1, -1], torch.int64, True), ([-1, -1, -1, -1], torch.int64, True)]
+    )
     def forward(self, x, values):
         return torch.heaviside(x, values)
 
 
 @register_test_case(module_factory=lambda: ElementwiseHeavisideIntModule())
 def ElementwiseHeavisideIntModule_basic(module, tu: TestUtils):
     module.forward(
-        tu.randint(5, 1, low=-100, high=1000).to(torch.int32),
-        tu.randint(1, low=-100, high=1000).to(torch.int32),
+        tu.randint(1, 2, 3, low=-100, high=1000),
+        tu.randint(1, 1, 1, 1, low=-100, high=1000),
     )