[Runtime] Fix triton_key import in autotuner disk caching (#7345)

Fixes
triton-lang/triton#7306 (comment)

Author: peterbell10

python/triton/runtime/autotuner.py

@@ -9,9 +9,11 @@
 from typing import Dict, Tuple, List, Optional
 
 from .. import knobs
-from .jit import KernelInterface
+from .jit import KernelInterface, JITFunction
 from .errors import OutOfResources, PTXASError
 from .driver import driver
+from .cache import get_cache_manager, triton_key
+from triton._C.libtriton import get_cache_invalidating_env_vars
 
 
 class Autotuner(KernelInterface):
@@ -169,10 +171,7 @@ def check_disk_cache(self, tuning_key, configs, bench_fn):
             bench_fn()
             return False
 
-        from triton._C.libtriton import get_cache_invalidating_env_vars
-        from triton.compiler.compiler import make_backend, triton_key
-        from triton.runtime.cache import get_cache_manager
-        from triton.runtime.jit import JITFunction
+        from triton.compiler.compiler import make_backend
 
         fn = self.fn
         while not isinstance(fn, JITFunction):