Commit f91153c3 authored by Anthony Chang's avatar Anthony Chang
Browse files

tidy

parent 8aa65c41
...@@ -11,19 +11,15 @@ ...@@ -11,19 +11,15 @@
#include "gemm_f16_tn_instance.hpp" #include "gemm_f16_tn_instance.hpp"
#include "gemm_f16_tt_instance.hpp" #include "gemm_f16_tt_instance.hpp"
template <ck::index_t... Is>
using S = ck::Sequence<Is...>;
using Row = ck::tensor_layout::gemm::RowMajor; using Row = ck::tensor_layout::gemm::RowMajor;
using Col = ck::tensor_layout::gemm::ColumnMajor; using Col = ck::tensor_layout::gemm::ColumnMajor;
using PassThrough = ck::tensor_operation::element_wise::PassThrough; using PassThrough = ck::tensor_operation::element_wise::PassThrough;
using F16 = ck::half_t; using F16 = ck::half_t;
using ADataType = F16; using ADataType = F16;
using BDataType = F16; using BDataType = F16;
using AccDataType = float; using AccDataType = float;
using CShuffleDataType = float; using CDataType = F16;
using CDataType = F16;
using ALayout = Row; using ALayout = Row;
using BLayout = Col; using BLayout = Col;
...@@ -33,9 +29,6 @@ using AElementOp = PassThrough; ...@@ -33,9 +29,6 @@ using AElementOp = PassThrough;
using BElementOp = PassThrough; using BElementOp = PassThrough;
using CElementOp = PassThrough; using CElementOp = PassThrough;
using ReferenceGemmInstance = ck::tensor_operation::host::
ReferenceGemm<ADataType, BDataType, CDataType, AccDataType, AElementOp, BElementOp, CElementOp>;
using ck::gemm_util::GemmParams; using ck::gemm_util::GemmParams;
using ck::tensor_operation::device::BaseOperator; using ck::tensor_operation::device::BaseOperator;
using ck::tensor_operation::device::DeviceGemm; using ck::tensor_operation::device::DeviceGemm;
......
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