Skip to content

GitLab

"example/vscode:/vscode.git/clone" did not exist on "eac50708d9893645ae6e0a590bb4706debe0a6b7"
Commit ee8f18f3 authored by guangzlu's avatar guangzlu
Browse files

fixed a bug in device_batched_multihead_attention_forward_xdl_cshuffle.hpp

parent 0db65bfb
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
include/ck/tensor_operation/gpu/device/impl/device_batched_multihead_attention_forward_xdl_cshuffle.hpp
View file @ ee8f18f3
...@@ -106,8 +106,8 @@ __global__ void ...@@ -106,8 +106,8 @@ __global__ void
p_b_grid + b_batch_offset, p_b_grid + b_batch_offset,
p_b1_grid + b1_batch_offset, p_b1_grid + b1_batch_offset,
p_c_grid + c_batch_offset, p_c_grid + c_batch_offset,
nullptr ? nullptr : p_z_grid + z_batch_offset, p_z_grid == nullptr ? nullptr : p_z_grid + z_batch_offset,
nullptr ? nullptr : p_lse_grid + lse_batch_offset, p_lse_grid == nullptr ? nullptr : p_lse_grid + lse_batch_offset,
p_shared, p_shared,
a_element_op, a_element_op,
b_element_op, b_element_op,
......
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