From 8393419f4a686a8161b9d2d00dbb68e5573cea53 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Luka=20Govedi=C4=8D?= Date: Thu, 21 Aug 2025 18:40:24 -0400 Subject: [PATCH] shorten comment MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Signed-off-by: Luka Govedič --- tests/compile/piecewise/test_toy_llama.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/compile/piecewise/test_toy_llama.py b/tests/compile/piecewise/test_toy_llama.py index f95932c498e9b..6a2947fb2b715 100644 --- a/tests/compile/piecewise/test_toy_llama.py +++ b/tests/compile/piecewise/test_toy_llama.py @@ -21,8 +21,7 @@ from vllm.config import (CompilationConfig, CompilationLevel, CUDAGraphMode, VllmConfig, set_current_vllm_config) from vllm.forward_context import BatchDescriptor, set_forward_context -# Import shared test operations -# The standard attention operation is automatically registered when imported +# This import automatically registers torch ops for testing (like silly.attention) import tests.compile.test_operations