/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-composable-kernel/checkouts/docs-6.4.3/include/ck/tensor_operation/gpu/device/impl/device_gemm_dpp.hpp Source File#
device_gemm_dpp.hpp
Go to the documentation of this file.
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:13
GemmSpecialization
Definition: gemm_specialization.hpp:11
Definition: ck.hpp:264
Definition: stream_config.hpp:10
Definition: gridwise_gemm_dpp.hpp:184
Definition: gridwise_gemm_dpp.hpp:96
static constexpr __host__ bool CalculateHasMainKBlockLoop(index_t K)
Definition: gridwise_gemm_dpp.hpp:356
static __host__ auto CalculateGridSize(index_t M, index_t N)
Definition: gridwise_gemm_dpp.hpp:115
static constexpr __host__ bool CheckValidity(const Problem &problem)
Definition: gridwise_gemm_dpp.hpp:263
Definition: sequence.hpp:43
Definition: device_base.hpp:50
Definition: device_base.hpp:61
Definition: device_gemm_dpp.hpp:119
float Run(const BaseArgument *p_arg, const StreamConfig &stream_config=StreamConfig{}) override
Definition: device_gemm_dpp.hpp:156
float Run(const Argument &karg, const StreamConfig &stream_config=StreamConfig{})
Definition: device_gemm_dpp.hpp:120
Definition: device_gemm_dpp.hpp:70
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_dpp.hpp:203
typename GridwiseGemm::Argument Argument
Definition: device_gemm_dpp.hpp:115
std::unique_ptr< BaseInvoker > MakeInvokerPointer() override
Definition: device_gemm_dpp.hpp:228
bool IsSupportedArgument(const BaseArgument *p_arg) override
Definition: device_gemm_dpp.hpp:179
static constexpr bool IsValidCompilationParameter()
Definition: device_gemm_dpp.hpp:163
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_dpp.hpp:184
static bool IsSupportedArgument(const Argument &karg)
Definition: device_gemm_dpp.hpp:169
static auto MakeInvoker()
Definition: device_gemm_dpp.hpp:200
std::string GetTypeString() const override
Definition: device_gemm_dpp.hpp:234
Definition: device_gemm.hpp:22