add todo

NVIDIA · Dec 22, 2024 · 202e1bf · 202e1bf
1 parent 9288bac
commit 202e1bf
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/csrc/device_lower/utils.cpp b/csrc/device_lower/utils.cpp
@@ -1995,6 +1995,7 @@ std::vector<Expr*> getSyncExprs(AsyncOpType async_type, int64_t keep_stages) {
   sync_exprs.push_back(commit);
   auto wait = IrBuilder::create<kir::AsyncWait>(async_type, keep_stages);
   sync_exprs.push_back(wait);
+  // TODO Do not apply for warp specialization
   if (async_type == AsyncOpType::WgMma) {
     auto sync = IrBuilder::create<kir::BlockSync>(true);
     sync_exprs.push_back(sync);