/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/develop/include/ck/tensor_operation/gpu/device/impl/device_gemm_xdl_cshuffle.hpp Source File#
device_gemm_xdl_cshuffle.hpp
Go to the documentation of this file.
23 // Note: inter-wave loop scheduler is rolled out to c-shuffle version first. Becuase non c-shuffle
float launch_and_time_kernel(const StreamConfig &stream_config, F kernel, dim3 grid_dim, dim3 block_dim, std::size_t lds_byte, Args... args)
Definition: kernel_launch.hpp:14
std::string getGemmSpecializationString(const GemmSpecialization &s)
Definition: gemm_specialization.hpp:32
GemmSpecialization
Definition: gemm_specialization.hpp:11
Definition: ck.hpp:268
@ Default
@ Interwave
constexpr LoopScheduler make_default_loop_scheduler()
Definition: loop_scheduler.hpp:20
Definition: stream_config.hpp:10
Definition: gridwise_gemm_xdl_cshuffle_v1.hpp:470
Definition: gridwise_gemm_xdl_cshuffle_v1.hpp:121
static constexpr __host__ bool CheckValidity(const Problem &problem)
Definition: gridwise_gemm_xdl_cshuffle_v1.hpp:571
Definition: integral_constant.hpp:20
Definition: device_base.hpp:197
Definition: device_base.hpp:208
Definition: device_gemm_xdl_cshuffle.hpp:149
float RunImp(const typename GridwiseGemm::Argument &arg, const StreamConfig &stream_config=StreamConfig{})
Definition: device_gemm_xdl_cshuffle.hpp:151
INVOKER_RUN3_IMPL float Run(const BaseArgument *p_arg, const StreamConfig &stream_config=StreamConfig{}) override
Definition: device_gemm_xdl_cshuffle.hpp:192
Definition: device_gemm_xdl_cshuffle.hpp:80
bool IsSupportedArgument(const BaseArgument *p_arg) override
Definition: device_gemm_xdl_cshuffle.hpp:238
static auto MakeArgument(const ADataType *p_a, const BDataType *p_b, CDataType *p_c, index_t M, index_t N, index_t K, index_t StrideA, index_t StrideB, index_t StrideC, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation)
Definition: device_gemm_xdl_cshuffle.hpp:243
std::string GetTypeString() const override
Definition: device_gemm_xdl_cshuffle.hpp:293
std::unique_ptr< BaseInvoker > MakeInvokerPointer() override
Definition: device_gemm_xdl_cshuffle.hpp:287
static constexpr auto I1
Definition: device_gemm_xdl_cshuffle.hpp:88
static bool IsSupportedArgument(const Argument &arg)
Definition: device_gemm_xdl_cshuffle.hpp:205
static constexpr bool IsValidCompilationParameter()
Definition: device_gemm_xdl_cshuffle.hpp:199
static constexpr auto I0
Definition: device_gemm_xdl_cshuffle.hpp:87
std::unique_ptr< BaseArgument > MakeArgumentPointer(const void *p_a, const void *p_b, void *p_c, index_t M, index_t N, index_t K, index_t StrideA, index_t StrideB, index_t StrideC, AElementwiseOperation, BElementwiseOperation, CElementwiseOperation) override
Definition: device_gemm_xdl_cshuffle.hpp:262
static auto MakeInvoker()
Definition: device_gemm_xdl_cshuffle.hpp:259
static constexpr auto NXdlPerWave32
Definition: device_gemm_xdl_cshuffle.hpp:85
typename GridwiseGemm64::Argument Argument
Definition: device_gemm_xdl_cshuffle.hpp:145
static constexpr auto I2
Definition: device_gemm_xdl_cshuffle.hpp:89
static constexpr GET_NXDL_PER_WAVE_IMPL auto NXdlPerWave64
Definition: device_gemm_xdl_cshuffle.hpp:84
Definition: device_gemm.hpp:22