Skip to content

GitLab

Commit 5ee74015 authored by Hejing Li's avatar Hejing Li
Browse files

homanode: pass node cfg to app cfg prepare cmd

parent d63364dc
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
experiments/simbricks/orchestration/nodeconfig.py
View file @ 5ee74015
...@@ -484,8 +484,8 @@ class HomaClientNode(AppConfig): ...@@ -484,8 +484,8 @@ class HomaClientNode(AppConfig):
self.workload = 2 self.workload = 2
self.protocol = 'homa' self.protocol = 'homa'
def prepare_post_cp(self) -> tp.List[str]: def prepare_post_cp(self, node) -> tp.List[str]:
return super().prepare_post_cp() + [ return super().prepare_post_cp(node) + [
'insmod homa.ko' 'insmod homa.ko'
] ]
# pylint: disable=consider-using-with # pylint: disable=consider-using-with
...@@ -513,8 +513,8 @@ class HomaServerNode(AppConfig): ...@@ -513,8 +513,8 @@ class HomaServerNode(AppConfig):
super().__init__() super().__init__()
self.protocol = 'homa' self.protocol = 'homa'
def prepare_post_cp(self) -> tp.List[str]: def prepare_post_cp(self, node) -> tp.List[str]:
return super().prepare_post_cp() + [ return super().prepare_post_cp(node) + [
'insmod homa.ko' 'insmod homa.ko'
] ]
# pylint: disable=consider-using-with # pylint: disable=consider-using-with
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment