device_gemm_instance.hpp 794 Bytes
Newer Older
1
2
3
4
#ifndef DEVICE_GEMM_INSTANTCE_HPP
#define DEVICE_GEMM_INSTANTCE_HPP

#include "device_gemm.hpp"
Chao Liu's avatar
Chao Liu committed
5
#include "element_wise_operation.hpp"
6
7
8
9
10
11
12
13
14
15
16
17

namespace ck {
namespace tensor_operation {
namespace device {
namespace device_gemm_instance {

template <typename ADataType,
          typename BDataType,
          typename CDataType,
          typename ALayout,
          typename BLayout,
          typename CLayout>
Chao Liu's avatar
Chao Liu committed
18
19
20
21
void add_device_gemm_instance(
    std::vector<DeviceGemmPtr<ck::tensor_operation::element_wise::PassThrough,
                              ck::tensor_operation::element_wise::PassThrough,
                              ck::tensor_operation::element_wise::PassThrough>>&);
22
23
24
25
26
27

} // namespace device_gemm_instance
} // namespace device
} // namespace tensor_operation
} // namespace ck
#endif