Commit accb4ca5 authored by ltqin's avatar ltqin
Browse files

add losting file device_gemm_xdl_splitk_instance.hpp

parent d862fdf0
#ifndef DEVICE_GEMM_XDL_SPLITK_INSTANCE
#define DEVICE_GEMM_XDL_SPLITK_INSTANCE
namespace ck {
namespace tensor_operation {
namespace device {
namespace device_gemm_instance {
template <>
void add_device_splitk_gemm_instance<float,
float,
float,
ck::tensor_layout::gemm::RowMajor,
ck::tensor_layout::gemm::RowMajor,
ck::tensor_layout::gemm::RowMajor>(
std::vector<DeviceGemmNoOpPtr>&);
template <>
void add_device_splitk_gemm_instance<float,
float,
float,
ck::tensor_layout::gemm::RowMajor,
ck::tensor_layout::gemm::ColumnMajor,
ck::tensor_layout::gemm::RowMajor>(
std::vector<DeviceGemmNoOpPtr>&);
template <>
void add_device_splitk_gemm_instance<float,
float,
float,
ck::tensor_layout::gemm::ColumnMajor,
ck::tensor_layout::gemm::RowMajor,
ck::tensor_layout::gemm::RowMajor>(
std::vector<DeviceGemmNoOpPtr>&);
template <>
void add_device_splitk_gemm_instance<float,
float,
float,
ck::tensor_layout::gemm::ColumnMajor,
ck::tensor_layout::gemm::ColumnMajor,
ck::tensor_layout::gemm::RowMajor>(
std::vector<DeviceGemmNoOpPtr>&);
} // namespace device_gemm_instance
} // namespace device
} // namespace tensor_operation
} // namespace ck
#endif
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment