Eager Compiler for Torch Compile (#11803)

Signed-off-by: Oasis-Git <ayw.sirius19@gmail.com>
2025-10-18 00:18:52 -07:00
parent f4488e9dd9
commit 1d726528f7
6 changed files with 68 additions and 18 deletions
--- a/python/sglang/srt/compilation/compilation_config.py
+++ b/python/sglang/srt/compilation/compilation_config.py
@@ -5,9 +5,10 @@ from typing import List

 # TODO(Yuwei): support better compile config support
 class CompilationConfig:
-    def __init__(self, capture_sizes: List[int]):
+    def __init__(self, capture_sizes: List[int], compiler: str = "eager"):
        self.traced_files = set()
        self.capture_sizes = capture_sizes
+        self.compiler = compiler

    def add_traced_file(self, file_path: str):
        self.traced_files.add(file_path)