From db98d0e5aa339274fa39e613f7b33f3dd97da1c3 Mon Sep 17 00:00:00 2001 From: Amir Ghamarian Date: Mon, 30 Mar 2026 10:24:09 +0000 Subject: [PATCH] Use direct sweep_tile in decode pipeline. Replace the local tile_sweeper wrapper call with sweep_tile(...) to keep the decode path simpler while preserving behavior. Made-with: Cursor --- .../ops/topk_softmax/pipeline/topk_softmax_decode_pipeline.hpp | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/include/ck_tile/ops/topk_softmax/pipeline/topk_softmax_decode_pipeline.hpp b/include/ck_tile/ops/topk_softmax/pipeline/topk_softmax_decode_pipeline.hpp index a94ebf7dbe..04b69743b0 100644 --- a/include/ck_tile/ops/topk_softmax/pipeline/topk_softmax_decode_pipeline.hpp +++ b/include/ck_tile/ops/topk_softmax/pipeline/topk_softmax_decode_pipeline.hpp @@ -67,8 +67,7 @@ struct TopkSoftmaxDecodePipeline w_(idx) = WeightType(1) / (WeightType(1) + exp(-w_(idx))); } }; - tile_sweeper ts{w_, w_f}; - ts(); + sweep_tile(w_, w_f); return w_; }();