From 226ef7b36ce83bf98c2559523080611ab6af673a Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Thu, 28 Jan 2021 20:35:31 -0600
Subject: [PATCH] Delayed model P1 initialization
---
predtuner/modeledapp.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/predtuner/modeledapp.py b/predtuner/modeledapp.py
index 06efc8c..680c5f1 100644
--- a/predtuner/modeledapp.py
+++ b/predtuner/modeledapp.py
@@ -200,7 +200,7 @@ class QoSModelP1(IQoSModel):
self.delta_tensors = {
op: {k.name: None for k in self.app.knobs} for op in self.app.ops
}
- self.baseline_tensor = self.output_f({})
+ self.baseline_tensor = None
@property
def name(self) -> str:
@@ -218,7 +218,7 @@ class QoSModelP1(IQoSModel):
def _init(self):
dt = self.delta_tensors
- btensor = self.baseline_tensor
+ btensor = self.baseline_tensor = self.output_f({})
if self.storage and self.storage.is_file():
for op, knob, delta_tensor in self._load(self.storage):
dt[op][knob] = delta_tensor
--
GitLab