File tree Expand file tree Collapse file tree 2 files changed +6
-5
lines changed
tritonbench/operators/launch_latency Expand file tree Collapse file tree 2 files changed +6
-5
lines changed Original file line number Diff line number Diff line change 8
8
DEFAULT_PYTHON_VERSION = "3.12"
9
9
10
10
PYTHON_VERSION_MAP = {
11
- "3.11" : {
11
+ "3.11" : {
12
12
"pytorch_url" : "cp311" ,
13
- },
14
- "3.12" : {
13
+ },
14
+ "3.12" : {
15
15
"pytorch_url" : "cp312" ,
16
- },
16
+ },
17
17
}
18
18
REPO_DIR = Path (__file__ ).parent .parent
19
19
Original file line number Diff line number Diff line change 1
1
import triton .language as tl
2
2
from torch import zeros
3
3
from torch ._C import _cuda_getCurrentRawStream as get_raw_stream
4
+
5
+ from torch ._inductor .utils import triton_version_uses_attrs_dict
4
6
from triton .compiler import CompiledKernel
5
7
6
8
from tritonbench .utils .triton_op import (
10
12
register_metric ,
11
13
)
12
14
13
- from torch ._inductor .utils import triton_version_uses_attrs_dict
14
15
from .kernels import get_trivial_add_kernel , nop_kernel , nop_with_args_kernel
15
16
16
17
You can’t perform that action at this time.
0 commit comments