Unverified Commit 189af908 authored by Xuchun Shang's avatar Xuchun Shang Committed by GitHub
Browse files

[Eagle Warning fix] replace the deprecated 'and' with & (#9215)


Signed-off-by: default avatarXuchun Shang <xuchun.shang@linux.alibaba.com>
parent f8644a56
......@@ -1070,7 +1070,7 @@ def copy_all_layer_kv_cache(
num_loop = tl.cdiv(stride, BLOCK_SIZE)
for i in range(num_loop):
copy_offset = tl.arange(0, BLOCK_SIZE) + i * BLOCK_SIZE
mask = (num_locs_offset < num_locs)[:, None] and (copy_offset < stride)[None, :]
mask = (num_locs_offset < num_locs)[:, None] & (copy_offset < stride)[None, :]
value = tl.load(
data_ptr + src_locs[:, None] * stride + copy_offset[None, :], mask=mask
)
......
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