From a8584badbe7c3a3be92de4689c93544e09274921 Mon Sep 17 00:00:00 2001
From: Yifan Zhao <yifanz16@illinois.edu>
Date: Sun, 7 Feb 2021 04:07:52 -0600
Subject: [PATCH] Fixed a bug in model P1 when using CPU tensor
---
predtuner/modeledapp.py | 3 ++-
1 file changed, 2 insertions(+), 1 deletion(-)
diff --git a/predtuner/modeledapp.py b/predtuner/modeledapp.py
index c0eeb47..a22c06d 100644
--- a/predtuner/modeledapp.py
+++ b/predtuner/modeledapp.py
@@ -211,7 +211,8 @@ class QoSModelP1(IQoSModel):
assert self.baseline_tensor is not None
with_approxes = self.app.add_baseline_to_knobs(with_approxes)
delta_tensors = np.array(
- [self.delta_tensors[op][knob] for op, knob in with_approxes.items()]
+ [self.delta_tensors[op][knob] for op, knob in with_approxes.items()],
+ dtype=np.object
)
ret = delta_tensors.sum() + self.baseline_tensor
return float(self.qos_metric(ret))
--
GitLab