Skip to content

GitLab

Commit da1afd7f authored by ltqin's avatar ltqin
Browse files

format

parent 5af78ac2
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 2 deletions
profiler/include/profiler/profile_batched_gemm_bias_softmax_gemm_permute_impl.hpp
View file @ da1afd7f
...@@ -241,7 +241,8 @@ bool profile_batched_gemm_bias_softmax_gemm_permute_impl(bool do_verification, ...@@ -241,7 +241,8 @@ bool profile_batched_gemm_bias_softmax_gemm_permute_impl(bool do_verification,
}); });
// mask out upper triangle // mask out upper triangle
acc0_g_m_n.ForEach([&](auto& self, auto idx) { acc0_g_m_n.ForEach([&](auto& self, auto idx) {
if(MaskingSpec == MaskingSpecialization::MaskUpperTriangleFromTopLeft && idx[1] < idx[2]) if(MaskingSpec == MaskingSpecialization::MaskUpperTriangleFromTopLeft &&
idx[1] < idx[2])
self(idx) = -ck::NumericLimits<float>::Infinity(); self(idx) = -ck::NumericLimits<float>::Infinity();
}); });
......
profiler/include/profiler/profile_batched_gemm_softmax_gemm_permute_impl.hpp
View file @ da1afd7f
...@@ -219,7 +219,8 @@ bool profile_batched_gemm_softmax_gemm_permute_impl(bool do_verification, ...@@ -219,7 +219,8 @@ bool profile_batched_gemm_softmax_gemm_permute_impl(bool do_verification,
// mask out upper triangle // mask out upper triangle
acc0_g_m_n.ForEach([&](auto& self, auto idx) { acc0_g_m_n.ForEach([&](auto& self, auto idx) {
if(MaskingSpec == MaskingSpecialization::MaskUpperTriangleFromTopLeft && idx[1] < idx[2]) if(MaskingSpec == MaskingSpecialization::MaskUpperTriangleFromTopLeft &&
idx[1] < idx[2])
self(idx) = -ck::NumericLimits<float>::Infinity(); self(idx) = -ck::NumericLimits<float>::Infinity();
}); });
......
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