topk_softmax.hpp Source File# Composable Kernel: topk_softmax.hpp Source File includeck_tileops topk_softmax.hpp Go to the documentation of this file. 1// SPDX-License-Identifier: MIT 2// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved. 3 4#pragma once 5 6#include "ck_tile/ops/topk_softmax/kernel/topk_softmax_kernel.hpp" 7#include "ck_tile/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_pipeline.hpp" 8#include "ck_tile/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_policy.hpp" 9#include "ck_tile/ops/topk_softmax/pipeline/topk_softmax_warp_per_row_problem.hpp" 10#include "ck_tile/ops/common/generic_2d_block_shape.hpp" 11#include "ck_tile/ops/common/load_interleaved_pk_type.hpp" 12#include "ck_tile/ops/common/streamk_common.hpp" 13#include "ck_tile/ops/common/tensor_layout.hpp" 14#include "ck_tile/ops/common/utils.hpp" generic_2d_block_shape.hpp load_interleaved_pk_type.hpp streamk_common.hpp tensor_layout.hpp topk_softmax_kernel.hpp topk_softmax_warp_per_row_pipeline.hpp topk_softmax_warp_per_row_policy.hpp topk_softmax_warp_per_row_problem.hpp utils.hpp