Eliminate wrapper facades for the Triton hooks API (#4245)

Signed-off-by: Anatoly Myachev <anatoly.myachev@intel.com>

Author: anmyachev

python/triton/_utils.py

@@ -35,26 +35,3 @@ def _impl(path: tuple[int, ...], current: Any):
     _impl((), iterable)
 
     return list(ret.keys())
-
-
-class ClassPropertyDescriptor:
-
-    def __init__(self, fget, fset=None):
-        self.fget = fget
-        self.fset = fset
-
-    def __get__(self, obj, cls):
-        return self.fget(cls)
-
-    def __set__(self, obj, value):
-        if self.fset is None:
-            raise AttributeError("can't set attribute")
-        self.fset(obj.__class__, value)
-
-    def setter(self, fset):
-        self.fset = fset
-        return self
-
-
-def classproperty(func):
-    return ClassPropertyDescriptor(func)

python/triton/compiler/compiler.py

@@ -9,7 +9,6 @@
 from ..runtime.cache import get_cache_manager, get_dump_manager, get_override_manager
 from ..runtime.driver import driver
 from ..tools.disasm import get_sass, get_spvdis
-from .._utils import classproperty
 # TODO: this shouldn't be here
 from .code_generator import ast_to_ttir
 from pathlib import Path
@@ -432,24 +431,6 @@ def __missing__(self, key):
 
 class CompiledKernel:
 
-    # FIXME: remove launch_enter_hook/launch_exit_hook properties
-    # when pytorch has a compatible layer for the new API.
-    @classproperty
-    def launch_enter_hook(cls):
-        return knobs.runtime.launch_enter_hook
-
-    @launch_enter_hook.setter
-    def launch_enter_hook(cls, value):
-        knobs.runtime.launch_enter_hook = value
-
-    @classproperty
-    def launch_exit_hook(cls):
-        return knobs.runtime.launch_exit_hook
-
-    @launch_exit_hook.setter
-    def launch_exit_hook(cls, value):
-        knobs.runtime.launch_exit_hook = value
-
     def __init__(self, src, metadata_group, hash):
         from collections import namedtuple
         metadata_path = next((Path(p) for c, p in metadata_group.items() if c.endswith(".json")))

python/triton/runtime/jit.py

@@ -14,7 +14,7 @@
 from types import ModuleType
 from .. import knobs
 from ..runtime.driver import driver
-from .._utils import find_paths_if, get_iterable_path, classproperty
+from .._utils import find_paths_if, get_iterable_path
 
 TRITON_MODULE = __name__[:-len(".runtime.jit")]
 
@@ -494,24 +494,6 @@ class JitFunctionInfo:
 
 class JITFunction(KernelInterface[T]):
 
-    # FIXME: remove cache_hook/compiled_hook properties
-    # when pytorch has a compatible layer for the new API.
-    @classproperty
-    def cache_hook(cls):
-        return knobs.runtime.jit_cache_hook
-
-    @cache_hook.setter
-    def cache_hook(cls, value):
-        knobs.runtime.jit_cache_hook = value
-
-    @classproperty
-    def compiled_hook(cls):
-        return knobs.runtime.jit_post_compile_hook
-
-    @compiled_hook.setter
-    def compiled_hook(cls, value):
-        knobs.runtime.jit_post_compile_hook = value
-
     def _call_hook(
         self,
         hook,