moe_sorting_policy.hpp Source File

• include
• ck_tile
• ops
• fused_moe
• pipeline

moe_sorting_policy.hpp

Go to the documentation of this file.

1// SPDX-License-Identifier: MIT

2// Copyright (c) 2018-2024, Advanced Micro Devices, Inc. All rights reserved.

3

4#pragma once

5

6#include "ck_tile/core.hpp"

7#include "ck_tile/ops/softmax.hpp"

8#include "ck_tile/ops/topk.hpp"

9

10namespace ck_tile {

11

12struct MoeSortingPolicy

13{

14};

15} // namespace ck_tile

core.hpp

ck_tile

Definition tile/core/algorithm/cluster_descriptor.hpp:13

softmax.hpp

ck_tile::MoeSortingPolicy

Definition moe_sorting_policy.hpp:13

topk.hpp