Merge pull request #457 from bwasti/sweep_code

Check in code used to generate plots for loop_tool in paper

examples/run_loop_tool_sweep.py

@@ -0,0 +1,50 @@
+# Copyright (c) Facebook, Inc. and its affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+import sys
+
+import compiler_gym
+
+names = ["toggle_mode", "up", "down", "toggle_thread"]
+actions = [3, 0, 1, 3, 0]
+base = 1024 * 512
+K = int(sys.argv[1]) * base
+
+vectorize = int(sys.argv[2])
+run_log = int(sys.argv[3])
+with compiler_gym.make("loop_tool-v0") as env:
+    env.reset(
+        benchmark=env.datasets.benchmark(uri="benchmark://loop_tool-v0/{}".format(K)),
+        action_space="simple",
+    )
+    if vectorize - 1:
+        vs = [1] * (vectorize - 1)
+        actions += vs + [0, 1, 0] + vs + [0, 2, 0]
+    for a in actions:
+        o = env.step(a)
+
+    if run_log:
+        env.observation_space = "action_state"
+        inner = 1
+        step = 512
+        for i in range(1, step):
+            o = env.step(1)
+            inner += 1
+        while inner * vectorize < K:
+            env.observation_space = "loop_tree"
+            for i in range(step):
+                if i == step - 1:
+                    env.observation_space = "flops"
+                o = env.step(1)
+                inner += 1
+            print(f"{K}, {inner}, {vectorize}: {o[0]}", flush=True)
+            step *= 2
+    else:
+        for i in range(K // (vectorize * 1024)):
+            env.observation_space = "action_state"
+            for j in range(1022 if i == 0 else 1023):
+                o = env.step(1)
+            env.observation_space = "flops"
+            o = env.step(1)
+            print(f"{K}, {(i + 1) * 1024}, {vectorize}: {o[0]}", flush=True)

examples/run_loop_tool_sweep.sh

@@ -0,0 +1,34 @@
+#!/bin/bash
+# Copyright (c) Facebook, Inc. and its affiliates.
+#
+# This source code is licensed under the MIT license found in the
+# LICENSE file in the root directory of this source tree.
+set -ve
+mkdir -p results
+
+python run_loop_tool_sweep.py  1 4 1 | tee results/s1_v4_log.txt
+python run_loop_tool_sweep.py  1 1 1 | tee results/s1_v1_log.txt
+
+python run_loop_tool_sweep.py  2 4 1 | tee results/s2_v4_log.txt
+python run_loop_tool_sweep.py  2 1 1 | tee results/s2_v1_log.txt
+
+python run_loop_tool_sweep.py  4 4 1 | tee results/s4_v4_log.txt
+python run_loop_tool_sweep.py  4 1 1 | tee results/s4_v1_log.txt
+
+python run_loop_tool_sweep.py  8 4 1 | tee results/s8_v4_log.txt
+python run_loop_tool_sweep.py  8 1 1 | tee results/s8_v1_log.txt
+
+python run_loop_tool_sweep.py  16 4 1 | tee results/s16_v4_log.txt
+python run_loop_tool_sweep.py  16 1 1 | tee results/s16_v1_log.txt
+
+python run_loop_tool_sweep.py  1 4 0 | tee results/s1_v4_linear.txt
+python run_loop_tool_sweep.py  1 1 0 | tee results/s1_v1_linear.txt
+
+python run_loop_tool_sweep.py  2 4 0 | tee results/s2_v4_linear.txt
+python run_loop_tool_sweep.py  2 1 0 | tee results/s2_v1_linear.txt
+
+python run_loop_tool_sweep.py  4 4 0 | tee results/s4_v4_linear.txt
+python run_loop_tool_sweep.py  4 1 0 | tee results/s4_v1_linear.txt
+
+python run_loop_tool_sweep.py  8 4 0 | tee results/s8_v4_linear.txt
+python run_loop_tool_sweep.py  8 1 0 | tee results/s8_v1_linear.txt