Skip to content

Commit ee3a3b3

Browse files
authored
fix
1 parent 4ad14dd commit ee3a3b3

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

lightllm/common/triton_utils/autotuner.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -113,11 +113,11 @@ def __init__(
113113
]
114114
self._run_key_func_param_names = [name for name, _ in inspect.signature(self.run_key_func).parameters.items()]
115115
self.mutates_args = mutates_args
116-
116+
117117
assert self.autotune_level in [
118-
AutotuneLevel.NO_AUTOTUNE,
119-
AutotuneLevel.AUTOTUNE,
120-
AutotuneLevel.AUTOTUNE_OVERWRITE,
118+
AutotuneLevel.USE_AUTOTUNE_HIS_CONFIG,
119+
AutotuneLevel.ADAPTIVE_AUTOTUNE,
120+
AutotuneLevel.FORCE_AUTOTUNE,
121121
AutotuneLevel.CLOSE_AUTOTUNE,
122122
]
123123
return

0 commit comments

Comments
 (0)