Skip to content

Commit bb239b5

Browse files
fixup
Signed-off-by: CodeNine-CJ <[email protected]>
1 parent 87f3f61 commit bb239b5

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/ut/torchair/test_torchair_model_runner.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -142,6 +142,10 @@ def setup_npu_torchair_model_runner(self, mocker: MockerFixture):
142142
"vllm_ascend.torchair.utils.torchair_quant_method_register")
143143
mocker.patch("vllm_ascend.envs.VLLM_ASCEND_TRACE_RECOMPILES",
144144
return_value=False)
145+
mock_attn_builder = Mock()
146+
mock_attn_backend = Mock()
147+
mock_attn_backend.get_builder_cls.return_value = lambda *args, **kwargs: mock_attn_builder
148+
145149

146150
# 设置类属性(如果需要)
147151
NPUTorchairModelRunner.decode_token_per_req = 1

0 commit comments

Comments
 (0)