Database/GEMM: updated profile

2017-09-03 20:10:26 -04:00
parent 6902e19a7f
commit 5e99bbe09d
8 changed files with 1830 additions and 7073 deletions
--- a/tune/regression.py
+++ b/tune/regression.py
@@ -50,7 +50,8 @@ def maximize(OpType, device, model, shapes, V):
    X[:, OpType.nshape_params:] = V
    X = OpType.get_valid(device, X)
    #Model predictions
-    predictions = model.predict(np.log2(X), batch_size=8192, verbose=0)
+    with tf.device('/cpu:0'):
+        predictions = model.predict(np.log2(X), batch_size=8192, verbose=0)
    pred_perfs = np.sort(predictions, axis=0)[::-1]
    pred_idxs = np.argsort(predictions, axis=0)[::-1]
    #Evaluate best predicted models