Address comments

Signed-off-by: Yong Hoon Shin <[email protected]>

Author: sarckk

.buildkite/test-pipeline.yaml

@@ -354,6 +354,7 @@ steps:
     - pytest -v -s compile/test_sequence_parallelism.py
     - pytest -v -s compile/test_async_tp.py
     - pytest -v -s compile/test_fusion_all_reduce.py
+    - pytest -v -s compile/test_decorator.py
 
 - label: PyTorch Fullgraph Smoke Test # 9min
   mirror_hardwares: [amdexperimental]
@@ -367,6 +368,7 @@ steps:
   - pytest -v -s compile/piecewise/test_simple.py
   - pytest -v -s compile/piecewise/test_toy_llama.py
   - pytest -v -s compile/piecewise/test_full_cudagraph.py
+  - pytest -v -s compile/piecewise/test_multiple_graphs.py
 
 - label: PyTorch Fullgraph Test # 18min
   mirror_hardwares: [amdexperimental]

vllm/compilation/decorators.py

@@ -130,7 +130,7 @@ def forward(self, x: torch.Tensor, y: Optional[torch.Tensor]):
 
     `compile_cond` is a function that takes a `VllmConfig` object as input and
     returns a boolean value indicating whether to compile the model or not.
-    This is useful if you want to compile the model only when certaijAn
+    This is useful if you want to compile the model only when certain
     conditions are met.
     """