vbatts/llama.cpp
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

fix CANN compile fail bug: the assert in ascend kernel function doesn't supportted on some CANN version

Browse source
This commit is contained in:
leo-pony 2024-11-28 09:28:52 +08:00
parent dc96a0b06a
commit e509f79635
1 changed files with 0 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
ggml/src/ggml-cann/kernels/dup.cpp
View file

@ -20,7 +20,6 @@ class DupByRows {
// Input has four dims. // Input has four dims.
int64_t op_block_num = GetBlockNum(); int64_t op_block_num = GetBlockNum();
int64_t op_block_idx = GetBlockIdx(); int64_t op_block_idx = GetBlockIdx();
assert(op_block_idx < SUPPORTED_MAX_DIM && op_block_idx >= 0, "Invalid block index:%d, max is:%d\n", op_block_idx, SUPPORTED_MAX_DIM);
// param // param
num_rows = input_ne_ub[1] * input_ne_ub[2] * input_ne_ub[3]; num_rows = input_ne_ub[1] * input_ne_ub[2] * input_ne_ub[3];