From 11b0cb30366654d43243d9833bf66a8bebd90859 Mon Sep 17 00:00:00 2001 From: Jingyue Wu Date: Fri, 30 Jan 2026 15:41:18 -0800 Subject: [PATCH] Print TensorView for debugging Otherwise I couldn't even locate the Expr in a printed fusion. --- csrc/host_ir/lower_to_communication.cpp | 12 ++---------- 1 file changed, 2 insertions(+), 10 deletions(-) diff --git a/csrc/host_ir/lower_to_communication.cpp b/csrc/host_ir/lower_to_communication.cpp index 79947a12768..8c73b1a356d 100644 --- a/csrc/host_ir/lower_to_communication.cpp +++ b/csrc/host_ir/lower_to_communication.cpp @@ -148,16 +148,8 @@ void lowerToBroadcast( const DeviceMesh& sender_mesh = input_tv->getDeviceMesh(); const DeviceMesh& receiver_mesh = output_tv->getDeviceMesh(); - NVF_ERROR_EQ( - sender_mesh.rank(), - 1, - "Broadcast only supports a 1D sender mesh. Given ", - sender_mesh); - NVF_ERROR_EQ( - receiver_mesh.rank(), - 1, - "Broadcast only supports a 1D receiver mesh. Given ", - receiver_mesh); + NVF_ERROR_EQ(sender_mesh.rank(), 1, "sender: ", input_tv->toString()); + NVF_ERROR_EQ(receiver_mesh.rank(), 1, "receiver: ", output_tv->toString()); DeviceIdxType root = sender_mesh.at(0); Team team = receiver_mesh.vector();