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_; }();