EngineX/xc-llm-ascend
3
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

[Bugfix] fix TransposeKvCacheByBlock op error report in plog (#7235)

Browse Source 
### What this PR does / why we need it?

As issue #7201 reported, there are some TransposeKvCacheByBlock
operation related ERRORs in plog when vllm launching, though it doesn't
influence the running of vllm, but ERRORs will be very confused in
debug, this PR fixed the problem as suggested.

### Does this PR introduce _any_ user-facing change?
no.

### How was this patch tested?

- vLLM version: v0.17.0
- vLLM main:
4034c3d32e

Signed-off-by: lidenghui <lidenghui1110@gmail.com>
This commit is contained in:
lidenghui1110
2026-03-17 10:08:32 +08:00
committed by GitHub
parent 3f39ac9c8d
commit 4e62a2ae15
1 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
csrc/transpose_kv_cache_by_block/op_host/transpose_kv_cache_by_block_def.cpp
View File

@@ -17,9 +17,9 @@ public:
.UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND}); .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND});
this->Input("blockIDs") this->Input("blockIDs")
.ParamType(REQUIRED) .ParamType(REQUIRED)
.DataTypeList({ge::DT_INT64}) .DataType({ge::DT_INT64, ge::DT_INT64})
.FormatList({ge::FORMAT_ND}) .Format({ge::FORMAT_ND, ge::FORMAT_ND})
.UnknownShapeFormat({ge::FORMAT_ND}); .UnknownShapeFormat({ge::FORMAT_ND, ge::FORMAT_ND});
this->Attr("blockSize").Int(); this->Attr("blockSize").Int();
this->Attr("headNum").Int(); this->Attr("headNum").Int();
this->Attr("headDim").Int(); this->Attr("headDim").Int();