Skip to content

GitLab

Commit 79c3e8ad authored by Jing Zhang's avatar Jing Zhang
Browse files

fixed ci

parent 59589133
Show whitespace changes
Inline Side-by-side
Showing with 0 additions and 6 deletions
test/grouped_gemm/grouped_gemm_fp16.cpp
View file @ 79c3e8ad
......@@ -121,8 +121,6 @@ bool TestGroupedGemm(DeviceGroupedGemmPtr_& groupedGemmPtr)
b_tensors_device.reserve(group_count);
c_tensors_device.reserve(group_count);
std::size_t flop = 0, num_btype = 0;
for(int i = 0; i < gemm_shapes.size(); i++)
{
a_tensors.emplace_back(Tensor<ADataType>(f_host_tensor_descriptor(
......@@ -134,10 +132,6 @@ bool TestGroupedGemm(DeviceGroupedGemmPtr_& groupedGemmPtr)
c_device_tensors.emplace_back(Tensor<CDataType>(f_host_tensor_descriptor(
gemm_shapes[i].M, gemm_shapes[i].N, gemm_shapes[i].StrideC, CLayout{})));
// std::cout << "gemm[" << i << "] a_m_k: " << a_tensors[i].mDesc
//<< " b_k_n: " << b_tensors[i].mDesc << " c_m_n: " << c_device_tensors[i].mDesc
//<< std::endl;
flop += std::size_t(2) * gemm_shapes[i].M * gemm_shapes[i].K * gemm_shapes[i].N;
num_btype += sizeof(ADataType) * a_tensors[i].mDesc.GetElementSize() +
sizeof(BDataType) * b_tensors[i].mDesc.GetElementSize() +
......
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