Skip to content

GitLab

"git@developer.sourcefind.cn:gaoqiong/composable_kernel.git" did not exist on "546a764eb4890ad9c178397bf4880d499a257a0b"
Commit c82093c4 authored by myamlak's avatar myamlak
Browse files

Cleaning

parent d478a389
Hide whitespace changes
Inline Side-by-side
Showing with 8 additions and 8 deletions
example/21_cgemm/cgemm_xdl_bf16.cpp
View file @ c82093c4
...@@ -162,14 +162,14 @@ int main(int argc, char* argv[]) ...@@ -162,14 +162,14 @@ int main(int argc, char* argv[])
{ {
case 0: break; case 0: break;
case 1: case 1:
a_m_k_real.GenerateTensorValue(GeneratorTensor_2<ADataType>{-5, 5}); a_m_k_real.GenerateTensorValue(GeneratorTensor_2<ADataType>{-2, 2});
a_m_k_imag.GenerateTensorValue(GeneratorTensor_2<ADataType>{-5, 5}); a_m_k_imag.GenerateTensorValue(GeneratorTensor_2<ADataType>{-2, 2});
b_k_n_real.GenerateTensorValue(GeneratorTensor_2<BDataType>{-5, 5}); b_k_n_real.GenerateTensorValue(GeneratorTensor_2<BDataType>{-2, 2});
b_k_n_imag.GenerateTensorValue(GeneratorTensor_2<BDataType>{-5, 5}); b_k_n_imag.GenerateTensorValue(GeneratorTensor_2<BDataType>{-2, 2});
break; break;
default: default:
a_m_k_real.GenerateTensorValue(GeneratorTensor_3<ADataType>{0.0, 1.0}); a_m_k_real.GenerateTensorValue(GeneratorTensor_3<ADataType>{-0.5, 0.5});
a_m_k_imag.GenerateTensorValue(GeneratorTensor_3<ADataType>{0.0, 1.0}); a_m_k_imag.GenerateTensorValue(GeneratorTensor_3<ADataType>{-0.5, 0.5});
b_k_n_real.GenerateTensorValue(GeneratorTensor_3<BDataType>{-0.5, 0.5}); b_k_n_real.GenerateTensorValue(GeneratorTensor_3<BDataType>{-0.5, 0.5});
b_k_n_imag.GenerateTensorValue(GeneratorTensor_3<BDataType>{-0.5, 0.5}); b_k_n_imag.GenerateTensorValue(GeneratorTensor_3<BDataType>{-0.5, 0.5});
} }
...@@ -278,12 +278,12 @@ int main(int argc, char* argv[]) ...@@ -278,12 +278,12 @@ int main(int argc, char* argv[])
c_m_n_real_host_result.mData, c_m_n_real_host_result.mData,
"Verification error: incorrect results in real part!", "Verification error: incorrect results in real part!",
1e-2f, 1e-2f,
1e-3f); 1e-1f);
ck::utils::check_err(c_m_n_imag_device_f32_result.mData, ck::utils::check_err(c_m_n_imag_device_f32_result.mData,
c_m_n_imag_host_result.mData, c_m_n_imag_host_result.mData,
"Verification error: incorrect results in imaginary part!", "Verification error: incorrect results in imaginary part!",
1e-2f, 1e-2f,
1e-3f); 1e-1f);
} }
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