Fix lint related to welford and also local_cache (#646)

Author: yf225

examples/welford.py

@@ -80,7 +80,7 @@ def eager_layer_norm(
     weight: torch.Tensor, bias: torch.Tensor, x: torch.Tensor, eps: float = 1e-05
 ) -> torch.Tensor:
     return torch.nn.functional.layer_norm(
-        x, normalized_shape=(x.shape[-1],), weight=weight, bias=bias, eps=eps
+        x, normalized_shape=[x.shape[-1]], weight=weight, bias=bias, eps=eps
     )
 
 

helion/autotuner/local_cache.py

@@ -57,7 +57,7 @@ def _generate_key(self) -> LooseAutotuneCacheKey:
                 device_properties = torch.cuda.get_device_properties(arg.device)
                 if torch.version.cuda is not None:  # pyright: ignore[reportAttributeAccessIssue]
                     hardware = device_properties.name
-                    runtime_name = torch.version.cuda  # pyright: ignore[reportAttributeAccessIssue]
+                    runtime_name = str(torch.version.cuda)
                 else:
                     hardware = device_properties.gcnArchName
                     runtime_name = torch.version.hip  # pyright: ignore[reportAttributeAccessIssue]