File tree Expand file tree Collapse file tree 1 file changed +1
-2
lines changed
lightllm/common/triton_utils Expand file tree Collapse file tree 1 file changed +1
-2
lines changed Original file line number Diff line number Diff line change @@ -86,7 +86,6 @@ def __init__(
8686 run_key_distance_func : Callable = lambda run_key , config_key : abs (int (run_key ) - int (config_key )),
8787 mutates_args : List [str ] = [],
8888 ):
89- self .autotune_level = get_triton_autotune_level ()
9089
9190 self .configs_gen_func = configs_gen_func
9291 self .kernel_name = kernel_name
@@ -114,7 +113,7 @@ def __init__(
114113 self ._run_key_func_param_names = [name for name , _ in inspect .signature (self .run_key_func ).parameters .items ()]
115114 self .mutates_args = mutates_args
116115
117- assert self . autotune_level in [
116+ assert get_triton_autotune_level () in [
118117 AutotuneLevel .USE_AUTOTUNE_HIS_CONFIG ,
119118 AutotuneLevel .ADAPTIVE_AUTOTUNE ,
120119 AutotuneLevel .FORCE_AUTOTUNE ,
You can’t perform that action at this time.
0 commit comments