Skip to content

Commit 74007c4

Browse files
committed
add format
1 parent 34c0c46 commit 74007c4

File tree

2 files changed

+6
-5
lines changed

2 files changed

+6
-5
lines changed

tools/python_utils.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -8,12 +8,12 @@
88
DEFAULT_PYTHON_VERSION = "3.12"
99

1010
PYTHON_VERSION_MAP = {
11-
"3.11": {
11+
"3.11": {
1212
"pytorch_url": "cp311",
13-
},
14-
"3.12": {
13+
},
14+
"3.12": {
1515
"pytorch_url": "cp312",
16-
},
16+
},
1717
}
1818
REPO_DIR = Path(__file__).parent.parent
1919

tritonbench/operators/launch_latency/operator.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,6 +1,8 @@
11
import triton.language as tl
22
from torch import zeros
33
from torch._C import _cuda_getCurrentRawStream as get_raw_stream
4+
5+
from torch._inductor.utils import triton_version_uses_attrs_dict
46
from triton.compiler import CompiledKernel
57

68
from tritonbench.utils.triton_op import (
@@ -10,7 +12,6 @@
1012
register_metric,
1113
)
1214

13-
from torch._inductor.utils import triton_version_uses_attrs_dict
1415
from .kernels import get_trivial_add_kernel, nop_kernel, nop_with_args_kernel
1516

1617

0 commit comments

Comments
 (0)