We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 2cf8e28 commit 72a2bf3Copy full SHA for 72a2bf3
vllm_ascend/worker/mtp_proposer_v1.py
@@ -48,9 +48,6 @@ def __init__(
48
device=self.runner.device)
49
self.torchair_compiled_model = None # type: ignore
50
self.torchair_compiled_models = {} # type: ignore
51
- self.reserved_mc2_mask = torch.zeros(512,
52
- dtype=torch.bool,
53
- device=self.runner.device)
54
self.torchair_graph_enabled = get_ascend_config(
55
).torchair_graph_config.enabled
56
0 commit comments