[Eagle Warning fix] replace the deprecated 'and' with & (#9215)
Signed-off-by: Xuchun Shang <xuchun.shang@linux.alibaba.com>
This commit is contained in:
@@ -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
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user