Skip to content

Commit e34e937

Browse files
authored
fix
1 parent b390464 commit e34e937

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

lightllm/common/triton_utils/autotuner.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff 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,

0 commit comments

Comments
 (0)