Commit 7f0915f6 authored by Chao Liu's avatar Chao Liu
Browse files

format

parent d1e9d6cc
...@@ -168,5 +168,4 @@ int main(int, char*[]) ...@@ -168,5 +168,4 @@ int main(int, char*[])
std::cout << "test magic number division: Fail" << std::endl; std::cout << "test magic number division: Fail" << std::endl;
return -1; return -1;
} }
} }
...@@ -69,7 +69,6 @@ struct gemmArgs ...@@ -69,7 +69,6 @@ struct gemmArgs
int KBatch; int KBatch;
}; };
int test_gemm(const gemmArgs& args) int test_gemm(const gemmArgs& args)
{ {
bool a_row_major, b_row_major, c_row_major; bool a_row_major, b_row_major, c_row_major;
...@@ -115,8 +114,10 @@ int test_gemm(const gemmArgs& args) ...@@ -115,8 +114,10 @@ int test_gemm(const gemmArgs& args)
Tensor<float> a_m_k(f_host_tensor_descriptor(args.M, args.K, args.StrideA, a_row_major)); Tensor<float> a_m_k(f_host_tensor_descriptor(args.M, args.K, args.StrideA, a_row_major));
Tensor<float> b_k_n(f_host_tensor_descriptor(args.K, args.N, args.StrideB, b_row_major)); Tensor<float> b_k_n(f_host_tensor_descriptor(args.K, args.N, args.StrideB, b_row_major));
Tensor<float> c_m_n_host_result(f_host_tensor_descriptor(args.M, args.N, args.StrideC, c_row_major)); Tensor<float> c_m_n_host_result(
Tensor<float> c_m_n_device_result(f_host_tensor_descriptor(args.M, args.N, args.StrideC, c_row_major)); f_host_tensor_descriptor(args.M, args.N, args.StrideC, c_row_major));
Tensor<float> c_m_n_device_result(
f_host_tensor_descriptor(args.M, args.N, args.StrideC, c_row_major));
// init data // init data
std::size_t num_thread = std::thread::hardware_concurrency(); std::size_t num_thread = std::thread::hardware_concurrency();
...@@ -242,12 +243,11 @@ int main(int argc, char* argv[]) ...@@ -242,12 +243,11 @@ int main(int argc, char* argv[])
printf("arg2 to 7: M, N, K, StrideA, StrideB, StrideC KBatch\n"); printf("arg2 to 7: M, N, K, StrideA, StrideB, StrideC KBatch\n");
return -1; return -1;
} }
for(const auto& kinder: test_cases) for(const auto& kinder : test_cases)
{ {
const auto res = test_gemm(kinder); const auto res = test_gemm(kinder);
if(!res) if(!res)
return -1; return -1;
} }
return 0; return 0;
} }
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