neuralmagic
diff --git a/‎vllm/model_executor/layers/fused_moe/pplx_garden_prepare_finalize.py‎
Lines changed: 4 additions & 4 deletions b/‎vllm/model_executor/layers/fused_moe/pplx_garden_prepare_finalize.py‎
Lines changed: 4 additions & 4 deletions
@@ -264,12 +264,12 @@ def _receiver(
  expert_num_tokens=expert_num_tokens, expert_num_tokens_cpu=None
  )
 
+ logger.debug("PPLX_GARDEN receive X %s", expert_x.shape)
  logger.debug(
- "PPLX_GARDEN receive %s, %s, %s",
- expert_x.shape,
- expert_x_scale.shape,
- expert_tokens_meta,
+ "PPLX_GARDEN receive X_SCALE %s",
+ expert_x_scale.shape if expert_x_scale is not None else None,
  )
+ logger.debug("PPLX_GARDEN receive META %s", expert_tokens_meta)
 
  return expert_x, expert_x_scale, expert_tokens_meta, None, None
Original file line number	Diff line number	Diff line change
`@@ -264,12 +264,12 @@ def _receiver(`
`264`	`264`	`expert_num_tokens=expert_num_tokens, expert_num_tokens_cpu=None`
`265`	`265`	`)`
`266`	`266`
	`267`	`+ logger.debug("PPLX_GARDEN receive X %s", expert_x.shape)`
`267`	`268`	`logger.debug(`
`268`		`- "PPLX_GARDEN receive %s, %s, %s",`
`269`		`- expert_x.shape,`
`270`		`- expert_x_scale.shape,`
`271`		`- expert_tokens_meta,`
	`269`	`+ "PPLX_GARDEN receive X_SCALE %s",`
	`270`	`+ expert_x_scale.shape if expert_x_scale is not None else None,`
`272`	`271`	`)`
	`272`	`+ logger.debug("PPLX_GARDEN receive META %s", expert_tokens_meta)`
`273`	`273`
`274`	`274`	`return expert_x, expert_x_scale, expert_tokens_meta, None, None`
`275`	`275`