/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/grid/gridwise_moe_mx_gemm.hpp File Reference#
gridwise_moe_mx_gemm.hpp File Reference
#include "ck/utility/common_header.hpp"
#include "ck/utility/env.hpp"
#include "ck/tensor_description/multi_index_transform_helper.hpp"
#include "ck/tensor_description/tensor_descriptor.hpp"
#include "ck/tensor_description/tensor_descriptor_helper.hpp"
#include "ck/tensor_operation/gpu/grid/block_to_ctile_map.hpp"
#include "ck/tensor_operation/gpu/block/blockwise_gemm_pipeline_xdlops_mx_moe_selector.hpp"
#include "ck/tensor_operation/gpu/block/thread_group_tensor_slice_transfer_v6r1.hpp"
#include "ck/tensor_operation/gpu/thread/threadwise_tensor_slice_transfer.hpp"
#include "ck/tensor_operation/gpu/element/unary_element_wise_operation.hpp"
#include "ck/tensor_operation/gpu/block/thread_group_tensor_slice_transfer_v7r3_scatter.hpp"
#include "ck/tensor_operation/gpu/block/thread_group_tensor_slice_transfer_direct_load.hpp"
#include "ck/tensor_operation/gpu/block/thread_group_tensor_slice_transfer_gather_direct_load.hpp"
Go to the source code of this file.
Namespaces | |
ck | |
Macros | |
#define | DEBUG_LOG 0 |
Enumerations | |
enum | ck::Activation { ck::gelu_and_mul = 0 , ck::silu_and_mul = 1 , ck::gelu_and_mul = 0 , ck::silu_and_mul = 1 , ck::gelu_and_mul = 0 , ck::silu_and_mul = 1 , ck::gelu_and_mul = 0 , ck::silu_and_mul = 1 , ck::gelu_and_mul = 0 , ck::silu_and_mul = 1 } |
Functions | |
template<typename GridwiseGemm , bool HasMainKBlockLoop, InMemoryDataOperationEnum CGlobalMemoryDataOperation, index_t MinimumOccupancy = 1, TailNumber TailNum = TailNumber::Even> | |
__global__ void | ck::kernel_moe_mxgemm_2lds (typename GridwiseGemm::Argument karg) |
Macro Definition Documentation
◆ DEBUG_LOG
#define DEBUG_LOG 0 |