softmax.hpp Source File# Composable Kernel: softmax.hpp Source File includeck_tileops 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/softmax/block/block_softmax_2d.hpp" 7#include "ck_tile/ops/softmax/block/block_softmax_2d_problem.hpp" 8#include "ck_tile/ops/common/generic_2d_block_shape.hpp" 9#include "ck_tile/ops/common/load_interleaved_pk_type.hpp" 10#include "ck_tile/ops/common/streamk_common.hpp" 11#include "ck_tile/ops/common/tensor_layout.hpp" 12#include "ck_tile/ops/common/utils.hpp" block_softmax_2d.hpp block_softmax_2d_problem.hpp generic_2d_block_shape.hpp load_interleaved_pk_type.hpp streamk_common.hpp tensor_layout.hpp utils.hpp