This reverts commit 71e9b379c8. It breaks vllm-ascend/Qwen3-30B-A3B-W8A8 test
This commit is contained in:
@@ -173,9 +173,7 @@ def _maybe_prefetch_mlp_down_proj_impl(x_dependency: torch.Tensor) -> None:
|
||||
except AssertionError:
|
||||
return
|
||||
|
||||
prefetch_mlp_enabled = getattr(forward_context, 'prefetch_mlp_enabled',
|
||||
False)
|
||||
if not prefetch_mlp_enabled:
|
||||
if not forward_context.prefetch_mlp_enabled:
|
||||
return
|
||||
forward_context.prefetch_mlp_down_proj = True
|
||||
model_instance = forward_context.model_instance
|
||||
@@ -204,9 +202,7 @@ def _maybe_wait_prefetch_done_impl(x: torch.Tensor) -> None:
|
||||
except AssertionError:
|
||||
return
|
||||
|
||||
prefetch_mlp_enabled = getattr(forward_context, 'prefetch_mlp_enabled',
|
||||
False)
|
||||
if not prefetch_mlp_enabled:
|
||||
if not forward_context.prefetch_mlp_enabled:
|
||||
return
|
||||
if forward_context.prefetch_mlp_gate_up_proj or \
|
||||
forward_context.prefetch_mlp_down_proj:
|
||||
|
||||
Reference in New Issue
Block a user