Update vllm/model_executor/layers/fused_moe/modular_kernel.py

caused by version difference

Co-authored-by: gemini-code-assist[bot] <176961590+gemini-code-assist[bot]@users.noreply.github.com>
Signed-off-by: HakJu Kim <[email protected]>
Signed-off-by: HakJu Kim <[email protected]>

Author: whitememory

vllm/model_executor/layers/fused_moe/modular_kernel.py

@@ -695,7 +695,7 @@ def _do_fused_experts(
         if isinstance(self.fused_experts, AiterExperts):
             workspace2 = None
         else:
-            workspace2 = self.workspace2_buffer.get(workspace2_shape,
+            workspace2 = buffers.workspace2.get(workspace2_shape,
                                                     device=a1.device,
                                                     dtype=workspace_dtype)