Skip to content

GitLab

"examples/community/imagic_stable_diffusion.py" did not exist on "726aba089d12503249d824bbaf4070f47d0fe44d"
Commit 57b856b3 authored by ltqin's avatar ltqin
Browse files

fix get N bug

parent 8afad0f6
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
example/32_batched_gemm_scale_softmax_gemm/batched_multihead_attention_backward_fp16.cpp
View file @ 57b856b3
...@@ -210,7 +210,7 @@ void run_attention_fwd_host(const TensorQ& q_g_m_k, ...@@ -210,7 +210,7 @@ void run_attention_fwd_host(const TensorQ& q_g_m_k,
// masking // masking
#if USING_MASK #if USING_MASK
auto N = s_g_m_n.GetLengths()[1]; auto N = s_g_m_n.GetLengths()[2];
const auto mask = DeviceGemmInstance::C0MatrixMask(N); const auto mask = DeviceGemmInstance::C0MatrixMask(N);
s_g_m_n.ForEach([&](auto& self, auto idx) { s_g_m_n.ForEach([&](auto& self, auto idx) {
if(mask.IsMaskedElement(idx[1], idx[2])) if(mask.IsMaskedElement(idx[1], idx[2]))
......
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