[inductor] Add some typing to common.py (pytorch#145691)

Pull Request resolved: pytorch#145691 Approved by: https://github.com/malfet ghstack dependencies: pytorch#145690
nWEIdia · Jan 27, 2025 · 2775da4 · 2775da4
1 parent 1ace012
commit 2775da4
Show file tree

Hide file tree

Showing 16 changed files with 134 additions and 106 deletions.
diff --git a/test/inductor/extension_backends/cpp/extension_codegen_backend.py b/test/inductor/extension_backends/cpp/extension_codegen_backend.py
@@ -15,7 +15,7 @@ def __init__(self, *args, **kwargs):
 
 class ExtensionScheduling(BaseScheduling):
     def __init__(self, scheduler):
-        self.scheduler = scheduler
+        super().__init__(scheduler)
         self._scheduling = cpp.CppScheduling(scheduler)
 
     def can_fuse_vertical(self, node1, node2):

diff --git a/test/inductor/extension_backends/triton/extension_codegen_backend.py b/test/inductor/extension_backends/triton/extension_codegen_backend.py
@@ -10,7 +10,7 @@ def __init__(self, *args, **kwargs):
 
 class ExtensionScheduling(BaseScheduling):
     def __init__(self, scheduler):
-        self.scheduler = scheduler
+        super().__init__(scheduler)
         self._triton_scheduling = triton.TritonScheduling(scheduler)
 
     def can_fuse_vertical(self, node1, node2):