minor

srivatsankrishnan · Oct 26, 2023 · e64bd26 · e64bd26
1 parent 4214d0d
commit e64bd26
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/sims/Timeloop/Dockerfile b/sims/Timeloop/Dockerfile
@@ -38,5 +38,5 @@ RUN cd oss-arch-gym/acme && pip install .[jax,tf,testing,envs]
 ENV LD_LIBRARY_PATH=$LD_LIBRARY_PATH:/usr/local/bin/conda/envs/arch-gym/lib
 RUN echo $LD_LIBRARY_PATH
 
-ENTRYPOINT ["conda", "run", "-n", "arch-gym", "python", "/home/workspace/src/timeloop-examples/workspace/final-project/example_designs/eyeriss_like/oss-arch-gym/sims/Timeloop/script/run_timeloop.sh"]
+ENTRYPOINT ["conda", "run", "-n", "arch-gym", "python", "/home/workspace/src/timeloop-examples/workspace/final-project/example_designs/eyeriss_like/oss-arch-gym/sims/Timeloop/launch_gcp.py"]
 
diff --git a/sims/Timeloop/script/run_timeloop.sh b/sims/Timeloop/script/run_timeloop.sh
@@ -3,13 +3,13 @@ HOME="/home/workspace"
 
 cd /home/workspace/src/timeloop-examples/workspace/final-project/example_designs/eyeriss_like
 
-OUTPUT_DIR="./../output"
+OUTPUT_DIR="./oss-arch-gym/sims/Timeloop/output"
 LAYER_SHAPE="AlexNet/AlexNet_layer5.yaml"
 
 # Invoke timeloop
-echo " " | timeloop-mapper ./../arch/eyeriss_like.yaml \
-  ./../arch/components/*.yaml \
-  ./../mapper/mapper.yaml constraints/*.yaml \
+echo " " | timeloop-mapper ./oss-arch-gym/sims/Timeloop/arch/eyeriss_like.yaml \
+  ./oss-arch-gym/sims/Timeloop/arch/components/*.yaml \
+  ./oss-arch-gym/sims/Timeloop/mapper/mapper.yaml constraints/*.yaml \
   ../../layer_shapes/$LAYER_SHAPE >$OUTPUT_DIR/timeloop_simulation_output.txt
 
-mv timeloop-mapper.stats.txt ./../output
+mv timeloop-mapper.stats.txt ./oss-arch-gym/sims/Timeloop/output