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
This commit is contained in:
Amir Ghamarian
2026-03-30 10:24:09 +00:00
parent d93efe1b61
commit db98d0e5aa

View File

@@ -67,8 +67,7 @@ struct TopkSoftmaxDecodePipeline
w_(idx) = WeightType(1) / (WeightType(1) + exp(-w_(idx)));
}
};
tile_sweeper<decltype(w_), decltype(w_f)> ts{w_, w_f};
ts();
sweep_tile(w_, w_f);
return w_;
}();