From 5b83b34afde1dba2b99207a0c15aacad5edbff17 Mon Sep 17 00:00:00 2001
From: Tri Dao <tridpq@gmail.com>
Date: Wed, 11 Feb 2026 14:19:29 -0500
Subject: [PATCH] [Cute-DSL] Add option for issue_clc_query without multicast

---
 python/CuTeDSL/cutlass/cute/arch/clc.py | 21 +++++++++++++++------
 1 file changed, 15 insertions(+), 6 deletions(-)

diff --git a/python/CuTeDSL/cutlass/cute/arch/clc.py b/python/CuTeDSL/cutlass/cute/arch/clc.py
index af344b5a..b2f7015f 100644
--- a/python/CuTeDSL/cutlass/cute/arch/clc.py
+++ b/python/CuTeDSL/cutlass/cute/arch/clc.py
@@ -23,6 +23,7 @@ from ..typing import Int32, Pointer, Int128
 def issue_clc_query(
     mbar_ptr: Pointer,
     clc_response_ptr: Pointer,
+    multicast: bool = True,
     loc=None,
     ip=None,
 ) -> None:
@@ -39,12 +40,20 @@ def issue_clc_query(
     """
     mbar_llvm_ptr = mbar_ptr.llvm_ptr
     clc_response_llvm_ptr = clc_response_ptr.llvm_ptr
-    nvvm.clusterlaunchcontrol_try_cancel_multicast(
-        clc_response_llvm_ptr,
-        mbar_llvm_ptr,
-        loc=loc,
-        ip=ip,
-    )
+    if multicast:
+        nvvm.clusterlaunchcontrol_try_cancel_multicast(
+            clc_response_llvm_ptr,
+            mbar_llvm_ptr,
+            loc=loc,
+            ip=ip,
+        )
+    else:
+        nvvm.clusterlaunchcontrol_try_cancel(
+            clc_response_llvm_ptr,
+            mbar_llvm_ptr,
+            loc=loc,
+            ip=ip,
+        )
 
 
 @dsl_user_op