[Log] include max_token_num and hidden_dim in FlashInfer workspace init log (#26605)

2026-06-01 13:26:52 -07:00
parent f2beb7bc76
commit da01f2974e
1 changed files with 2 additions and 1 deletions
@@ -436,7 +436,8 @@ class FlashInferWorkspaceManager:
        backend = getattr(self.workspace, "backend", "unknown")
        logger.info(
            f"FlashInfer workspace initialized for rank {rank}, "
-            f"world_size {world_size}, backend {backend}"
+            f"world_size {world_size}, backend {backend}, "
+            f"max_token_num {max_token_num}, hidden_dim {hidden_dim}"
        )

    def is_buffer_size_sufficient(