Skip to content

GitLab

Unverified Commit 02d69525 authored by Po Yen Chen's avatar Po Yen Chen Committed by GitHub
Browse files

Align the floating point init values (#25)

parent 9c3ce77c
Hide whitespace changes
Inline Side-by-side
Showing with 9 additions and 9 deletions
example/91_tile_program/batched_gemm_softmax_gemm.cpp
View file @ 02d69525
...@@ -75,9 +75,9 @@ int main(int argc, char* argv[]) ...@@ -75,9 +75,9 @@ int main(int argc, char* argv[])
ck::utils::FillUniformDistributionIntegerValue<KDataType>{-2.f, 2.f}(k_host); ck::utils::FillUniformDistributionIntegerValue<KDataType>{-2.f, 2.f}(k_host);
ck::utils::FillUniformDistributionIntegerValue<VDataType>{-2.f, 2.f}(v_host); ck::utils::FillUniformDistributionIntegerValue<VDataType>{-2.f, 2.f}(v_host);
#else #else
ck::utils::FillUniformDistribution<QDataType>{-2.f, 2.f}(q_host); ck::utils::FillUniformDistribution<QDataType>{0.f, 1.f}(q_host);
ck::utils::FillUniformDistribution<KDataType>{-2.f, 2.f}(k_host); ck::utils::FillUniformDistribution<KDataType>{0.f, 1.f}(k_host);
ck::utils::FillUniformDistribution<VDataType>{-2.f, 2.f}(v_host); ck::utils::FillUniformDistribution<VDataType>{-.5f, .5f}(v_host);
#endif #endif
// reference // reference
......
example/91_tile_program/fmha_fwd.cpp
View file @ 02d69525
...@@ -113,9 +113,9 @@ int main(int argc, char* argv[]) ...@@ -113,9 +113,9 @@ int main(int argc, char* argv[])
ck::utils::FillUniformDistributionIntegerValue<KDataType>{-2.f, 2.f}(k_host); ck::utils::FillUniformDistributionIntegerValue<KDataType>{-2.f, 2.f}(k_host);
ck::utils::FillUniformDistributionIntegerValue<VDataType>{-2.f, 2.f}(v_host); ck::utils::FillUniformDistributionIntegerValue<VDataType>{-2.f, 2.f}(v_host);
#else #else
ck::utils::FillUniformDistribution<QDataType>{-2.f, 2.f}(q_host); ck::utils::FillUniformDistribution<QDataType>{0.f, 1.f}(q_host);
ck::utils::FillUniformDistribution<KDataType>{-2.f, 2.f}(k_host); ck::utils::FillUniformDistribution<KDataType>{0.f, 1.f}(k_host);
ck::utils::FillUniformDistribution<VDataType>{-2.f, 2.f}(v_host); ck::utils::FillUniformDistribution<VDataType>{-.5f, .5f}(v_host);
#endif #endif
DeviceMem q_buf(sizeof(QDataType) * q_host.GetElementSpaceSize()); DeviceMem q_buf(sizeof(QDataType) * q_host.GetElementSpaceSize());
......
example/91_tile_program/gemm_softmax_gemm.cpp
View file @ 02d69525
...@@ -73,9 +73,9 @@ int main(int argc, char* argv[]) ...@@ -73,9 +73,9 @@ int main(int argc, char* argv[])
ck::utils::FillUniformDistributionIntegerValue<KDataType>{-2.f, 2.f}(k_host); ck::utils::FillUniformDistributionIntegerValue<KDataType>{-2.f, 2.f}(k_host);
ck::utils::FillUniformDistributionIntegerValue<VDataType>{-2.f, 2.f}(v_host); ck::utils::FillUniformDistributionIntegerValue<VDataType>{-2.f, 2.f}(v_host);
#else #else
ck::utils::FillUniformDistribution<QDataType>{-2.f, 2.f}(q_host); ck::utils::FillUniformDistribution<QDataType>{0.f, 1.f}(q_host);
ck::utils::FillUniformDistribution<KDataType>{-2.f, 2.f}(k_host); ck::utils::FillUniformDistribution<KDataType>{0.f, 1.f}(k_host);
ck::utils::FillUniformDistribution<VDataType>{-2.f, 2.f}(v_host); ck::utils::FillUniformDistribution<VDataType>{-.5f, .5f}(v_host);
#endif #endif
// reference // reference
......
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