[inductor] Rename knobs > triton_knobs in static_cuda_launcher (#153189)

danzimm · pytorchmergebot · commit 658aea980ce7 · 2025-05-10T00:26:21.000Z
Summary: A follow up from #152457 since I didn't address the comment then Test Plan: CI Differential Revision: D74421432 Pull Request resolved: #153189 Approved by: https://github.com/jamesjwu
diff --git a/torch/_inductor/runtime/static_cuda_launcher.py b/torch/_inductor/runtime/static_cuda_launcher.py
@@ -3,7 +3,7 @@
 from typing import Any, Optional
 from typing_extensions import Unpack
 
-from .triton_compat import ASTSource, CompiledKernel, knobs
+from .triton_compat import ASTSource, CompiledKernel, knobs as triton_knobs
 
 
 class StaticallyLaunchedCudaKernel:
@@ -47,12 +47,12 @@ def __init__(self, kernel: CompiledKernel) -> None:
 
         self.hash = kernel.hash
 
-        if knobs is None:
+        if triton_knobs is None:
             launch_enter = kernel.__class__.launch_enter_hook
             launch_exit = kernel.__class__.launch_exit_hook
         else:
-            launch_enter = knobs.runtime.launch_enter_hook
-            launch_exit = knobs.runtime.launch_exit_hook
+            launch_enter = triton_knobs.runtime.launch_enter_hook
+            launch_exit = triton_knobs.runtime.launch_exit_hook
 
         if launch_enter is not None or launch_exit is not None:
             raise NotImplementedError(
