Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
softmax.hpp 259 Bytes
Newer Older
Aleksander Dudek's avatar
[CK_TILE] Update the file structure  
Aleksander Dudek committed
1
2
3
4
5
6
7 
// SPDX-License-Identifier: MIT
// Copyright (c) 2018-2025, Advanced Micro Devices, Inc. All rights reserved.

#pragma once

#include "ck_tile/host/ops/softmax/block/block_softmax_2d.hpp"
#include "ck_tile/host/ops/softmax/block/block_softmax_2d_problem.hpp"