Skip to content

Commit 2c9bfb4

Browse files
committed
[SPEC] Add Triton 3.7 comments
1 parent 121d529 commit 2c9bfb4

3 files changed

Lines changed: 8 additions & 8 deletions

File tree

third_party/mthreads/backend/spec/triton/__init__.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -6,5 +6,5 @@
66

77
def triton_extend_globals(globals_dict):
88
# NOTE: Must use absolute path import.
9-
from triton.compiler.compiler import max_shared_mem
10-
globals_dict["max_shared_mem"] = max_shared_mem
9+
from triton.compiler.compiler import max_shared_mem # Triton 3.7
10+
globals_dict["max_shared_mem"] = max_shared_mem # Triton 3.7

third_party/mthreads/backend/spec/triton/experimental/gluon/language/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@
8181
static_range,
8282
store,
8383
sub,
84-
barrier,
84+
barrier, # Triton 3.7: thread_barrier() => barrier(*, cluster=False)
8585
to_linear_layout,
8686
to_tensor,
8787
warp_specialize,
Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,10 +1,10 @@
11
def language_extend_globals(globals_dict):
22
# NOTE: Must use absolute path import.
3-
from triton.language.standard import squeeze, unsqueeze
3+
from triton.language.standard import squeeze, unsqueeze # Triton 3.7
4+
from triton.language.core import to_tensor # Triton 3.7
45
from triton.language.core import _experimental_descriptor_load, _experimental_descriptor_store
5-
from triton.language.core import to_tensor
6-
globals_dict["squeeze"] = squeeze
7-
globals_dict["unsqueeze"] = unsqueeze
6+
globals_dict["squeeze"] = squeeze # Triton 3.7
7+
globals_dict["unsqueeze"] = unsqueeze # Triton 3.7
8+
globals_dict["to_tensor"] = to_tensor # Triton 3.7
89
globals_dict["_experimental_descriptor_load"] = _experimental_descriptor_load
910
globals_dict["_experimental_descriptor_store"] = _experimental_descriptor_store
10-
globals_dict["to_tensor"] = to_tensor

0 commit comments

Comments
 (0)