include/ck_tile/ops/fused_moe/kernel/moe_sorting_kernel.hpp File Reference

• include
• ck_tile
• ops
• fused_moe
• kernel

Classes | Namespaces | Macros

moe_sorting_kernel.hpp File Reference

#include "ck_tile/core.hpp"
#include "ck_tile/ops/common.hpp"
#include "ck_tile/ops/elementwise.hpp"
#include "ck_tile/host/hip_check_error.hpp"
#include <string>
#include <type_traits>

Go to the source code of this file.

Classes
struct	ck_tile::MoeSortingHostArgs
struct	ck_tile::MoeSortingKernel< Problem_ >
struct	ck_tile::MoeSortingKernel< Problem_ >::Kargs

Namespaces
	ck_tile

Macros
#define	MOE_SORTING_MOCK_ID(token_id_, topk_id_)    static_cast<uint32_t>(((token_id_)&0x00ffffff) | (((topk_id_)&0xff) << 24))

Macro Definition Documentation

◆ MOE_SORTING_MOCK_ID

#define MOE_SORTING_MOCK_ID	(		token_id_,
			topk_id_
	)		static_cast<uint32_t>(((token_id_)&0x00ffffff) | (((topk_id_)&0xff) << 24))