remove prints

feifeibear · Nov 5, 2024 · 3aa28e5 · 3aa28e5
1 parent 7682555
commit 3aa28e5
Showing 1 changed file with 0 additions and 5 deletions.
diff --git a/tests/core/test_xfuser_attn.py b/tests/core/test_xfuser_attn.py
@@ -137,14 +137,9 @@ def test_xfuser_attn_layer_joint_strategy_rear(self):
             joint_tensor_value=joint_v,
             joint_strategy=joint_strategy,
         )
-        print(f"local_q: {local_q.shape}, joint_q: {joint_q.shape}")
-        print(f"output: {output.shape}")
-        print(f"ref_output: {ref_output.shape}")
-        print(f"output - ref: {output - ref_output}")
         # assert torch.max(torch.abs(output - ref_output)) < 1e-3
         torch.testing.assert_close(ref_output, output, rtol=1e-3, atol=1e-3)
 
-
     def test_xfuser_attn_layer(self):
         """Test xFuserLongContextAttention layer in distributed mode"""
         # Create test tensors