[diffusion] feat: add performance mode server args (#24491)

2026-05-14 00:57:46 +08:00
parent e2290b155a
commit ff70aeac30
79 changed files with 1734 additions and 532 deletions
@@ -86,8 +86,9 @@ class TestLMHeadFP32(unittest.TestCase):
                state.update(called=True, ooperationp="linear", a=x.dtype, b=w.dtype)
            return original_linear(x, w, bias)

-        with patch("torch.matmul", new=probe_matmul), patch(
-            "torch.nn.functional.linear", new=probe_linear
+        with (
+            patch("torch.matmul", new=probe_matmul),
+            patch("torch.nn.functional.linear", new=probe_linear),
        ):
            logits = logprocessor._get_logits(hidden_state, head, meta)
        self.assertEqual(hidden_state.dtype, hidden_state_dtype)