ROCm/composable_kernel
1
0
Fork 0
mirror of https://github.com/ROCm/composable_kernel.git synced 2026-05-18 03:49:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

use single threaded tensor generator (#161)

Browse Source 
[ROCm/composable_kernel commit: f015c77687]
This commit is contained in:
Anthony Chang
2022-03-31 11:28:30 +08:00
committed by GitHub
parent 297ef9795d
commit 8bb6c6e120
20 changed files with 26 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
test/gemm_split_k/gemm_split_k.cpp
View File

@@ -120,7 +120,7 @@ int test_gemm(const gemmArgs& args)
f_host_tensor_descriptor(args.M, args.N, args.StrideC, c_row_major));
// init data
std::size_t num_thread = std::thread::hardware_concurrency();
std::size_t num_thread = 1;
a_m_k.GenerateTensorValue(GeneratorTensor_2<float>{-5, 5}, num_thread);
b_k_n.GenerateTensorValue(GeneratorTensor_2<float>{-5, 5}, num_thread);
// set zero to c_device_buf