EngineX-Hygon/sglang
5
0
Fork 0
Code Issues Pull Requests Actions 7 Projects Releases Wiki Activity

Add set_var to interpreter.py (#263)

Browse Source
This commit is contained in:
Lin Tianchuan
2024-03-07 23:20:11 +08:00
committed by GitHub
parent b0b722ee8e
commit 30d67b2bca
1 changed files with 9 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
python/sglang/lang/interpreter.py
View File

@@ -245,6 +245,9 @@ class StreamExecutor:
self.variable_event[name].wait() self.variable_event[name].wait()
return self.variables[name] return self.variables[name]
def set_var(self, name, value):
self.variables[name] = value
def get_meta_info(self, name): def get_meta_info(self, name):
if name in self.variable_event: if name in self.variable_event:
self.variable_event[name].wait() self.variable_event[name].wait()
@@ -726,6 +729,9 @@ class ProgramState:
def get_var(self, name): def get_var(self, name):
return self.stream_executor.get_var(name) return self.stream_executor.get_var(name)
def set_var(self, name, value):
return self.stream_executor.set_var(name, value)
def get_meta_info(self, name): def get_meta_info(self, name):
return self.stream_executor.get_meta_info(name) return self.stream_executor.get_meta_info(name)
@@ -736,6 +742,9 @@ class ProgramState:
def __getitem__(self, name): def __getitem__(self, name):
return self.get_var(name) return self.get_var(name)
def __setitem__(self, name, value):
self.set_var(name, value)
def __del__(self): def __del__(self):
self.stream_executor.end() self.stream_executor.end()