diff --git a/python/triton/tools/compile.py b/python/triton/tools/compile.py index b43ac8e7d3..872332b03d 100644 --- a/python/triton/tools/compile.py +++ b/python/triton/tools/compile.py @@ -8,7 +8,7 @@ import triton from triton.compiler.code_generator import kernel_suffix -from triton.backends.intel.driver import ty_to_cpp +from triton.backends.nvidia.driver import ty_to_cpp desc = """ Triton ahead-of-time compiler: