Compilation Folder Reset (#11539)
Signed-off-by: Oasis-Git <ayw.sirius19@gmail.com>
This commit is contained in:
@@ -0,0 +1,19 @@
|
||||
# Adapted from https://github.com/vllm-project/vllm/blob/v0.10.0/vllm/compilation/compilation_config.py
|
||||
|
||||
from typing import List
|
||||
|
||||
|
||||
# TODO(Yuwei): support better compile config support
|
||||
class CompilationConfig:
|
||||
def __init__(self, capture_sizes: List[int]):
|
||||
self.traced_files = set()
|
||||
self.capture_sizes = capture_sizes
|
||||
|
||||
def add_traced_file(self, file_path: str):
|
||||
self.traced_files.add(file_path)
|
||||
|
||||
def get_traced_files(self):
|
||||
return self.traced_files
|
||||
|
||||
def get_capture_sizes(self):
|
||||
return self.capture_sizes
|
||||
Reference in New Issue
Block a user