topk_softmax_warp_per_row_pipeline.hpp File Reference

topk_softmax_warp_per_row_pipeline.hpp File Reference#

Composable Kernel: topk_softmax_warp_per_row_pipeline.hpp File Reference
topk_softmax_warp_per_row_pipeline.hpp File Reference
#include "ck_tile/core.hpp"
#include "ck_tile/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_policy.hpp"
#include <string>
#include <type_traits>

Go to the source code of this file.

Classes

struct  ck_tile::TopkSoftmaxWarpPerRowPipeline< Problem_, Policy_ >

Namespaces

namespace  ck_tile

Macros

#define TOPK_SOFTMAX_USE_RAW_TILE_WINDOW   0

Macro Definition Documentation

◆ TOPK_SOFTMAX_USE_RAW_TILE_WINDOW

#define TOPK_SOFTMAX_USE_RAW_TILE_WINDOW   0