Skip to content

Commit b4afd43

Browse files
committed
fix
1 parent da01e98 commit b4afd43

File tree

1 file changed

+5
-8
lines changed

1 file changed

+5
-8
lines changed

lightllm/common/triton_utils/autotuner.py

Lines changed: 5 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -1,20 +1,17 @@
11
import triton
22
import orjson
33
import os
4-
from pathlib import Path
5-
from tqdm import tqdm
6-
from frozendict import frozendict
74
import inspect
85
import torch
6+
import fcntl
97
import torch.distributed as dist
10-
from functools import lru_cache
8+
from pathlib import Path
9+
from tqdm import tqdm
10+
from frozendict import frozendict
1111
from lightllm.utils.device_utils import get_current_device_name
12-
import math
13-
import fcntl
1412
from lightllm.utils.log_utils import init_logger
15-
import traceback
1613
from typing import Callable, Optional, Union, List
17-
from lightllm.utils.envs_utils import is_triton_autotune_enabled, disable_triton_autotune
14+
from lightllm.utils.envs_utils import is_triton_autotune_enabled
1815
from lightllm.common.kernel_config import KernelConfigs
1916

2017

0 commit comments

Comments
 (0)