diff --git a/thunder/core/jit_ext.py b/thunder/core/jit_ext.py index 6c38caf893..ce774c4936 100644 --- a/thunder/core/jit_ext.py +++ b/thunder/core/jit_ext.py @@ -20,7 +20,7 @@ import inspect import time -from thunder.core.compile_data import compile_data_and_stats, get_cache_option, get_compile_data, get_compile_option +from thunder.core.compile_data import compile_data_and_stats, get_cache_option, get_compile_data import thunder.clang as clang import thunder.core.transforms from thunder.core.baseutils import run_once