Skip to content

Commit d58af32

Browse files
Stardep-lmcclaude
andcommitted
fix: add missing thread_barrier to mthreads spec _core.py
Upstream triton.experimental.gluon.language.__init__.py imports thread_barrier from ._core, but the mthreads spec_path copy only had the older barrier() function. This caused 10 test collection errors across test_frontend and tle tests. Co-Authored-By: Claude <noreply@anthropic.com>
1 parent dc5b998 commit d58af32

1 file changed

Lines changed: 8 additions & 0 deletions

File tree

  • third_party/mthreads/backend/spec/triton/experimental/gluon/language

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

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -573,6 +573,14 @@ def num_ctas(_semantic=None):
573573
return _semantic.num_ctas()
574574

575575

576+
@builtin
577+
def thread_barrier(_semantic=None):
578+
"""
579+
Insert a barrier to synchronize threads within a CTA.
580+
"""
581+
return _semantic.debug_barrier()
582+
583+
576584
@builtin
577585
def barrier(*, cluster: bool = False, _semantic=None):
578586
"""

0 commit comments

Comments
 (0)