Skip to content

GitLab

"driver/conv_driver.cpp" did not exist on "f58bf38445c4b28c88a4bbe5bcb3f12e474650c8"
Commit 5d42067e authored by Jing Zhang's avatar Jing Zhang
Browse files

format

parent 9fed0ade
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 3 deletions
example/01_gemm/gemm_xdl_fp16_pk_i4_v3.cpp
View file @ 5d42067e
...@@ -58,7 +58,7 @@ using DeviceGemmV2Instance = ...@@ -58,7 +58,7 @@ using DeviceGemmV2Instance =
static int NPerBlock = 32; static int NPerBlock = 32;
static int KPerBlock = 128; static int KPerBlock = 128;
#endif #endif
// clang-format on // clang-format on
using ReferenceGemmInstance = ck::tensor_operation::host::ReferenceGemm<ADataType, using ReferenceGemmInstance = ck::tensor_operation::host::ReferenceGemm<ADataType,
BDataType, BDataType,
...@@ -152,7 +152,7 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config) ...@@ -152,7 +152,7 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config)
DeviceMem b_k_n_device_buf(sizeof(BDataType) * b_k_n.mDesc.GetElementSpaceSize()); DeviceMem b_k_n_device_buf(sizeof(BDataType) * b_k_n.mDesc.GetElementSpaceSize());
DeviceMem c_m_n_device_buf(sizeof(CDataType) * c_m_n_device_result.mDesc.GetElementSpaceSize()); DeviceMem c_m_n_device_buf(sizeof(CDataType) * c_m_n_device_result.mDesc.GetElementSpaceSize());
//weight permute // weight permute
#if 0 #if 0
int N1 = NPerBlock; int N1 = NPerBlock;
int K1 = KPerBlock; int K1 = KPerBlock;
...@@ -182,7 +182,6 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config) ...@@ -182,7 +182,6 @@ bool run_gemm(const ProblemType& problem_size, const ExecutionConfig& config)
} }
} }
#else #else
for(int i = 0; i < N; i++) for(int i = 0; i < N; i++)
{ {
......
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