update CMakeLists.txt to create symbolic links for compute_programl a…

…nd compute_autophase
facebookresearch · Mar 6, 2022 · 0509377 · 0509377
1 parent 433d209
commit 0509377
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/compiler_gym/third_party/autophase/CMakeLists.txt b/compiler_gym/third_party/autophase/CMakeLists.txt
@@ -53,3 +53,6 @@ cg_cc_binary(
     ${LLVM_DEFINITIONS}
   PUBLIC
 )
+
+ADD_CUSTOM_TARGET(link_compute_autophase_target ALL
+                  COMMAND ${CMAKE_COMMAND} -E create_symlink ${CMAKE_BINARY_DIR}/compiler_gym/third_party/autophase/compute_autophase ${CMAKE_SOURCE_DIR}/compiler_gym/third_party/autophase/compute_autophase-prelinked)
diff --git a/compiler_gym/third_party/programl/CMakeLists.txt b/compiler_gym/third_party/programl/CMakeLists.txt
@@ -28,3 +28,6 @@ cg_cc_binary(
     ${LLVM_DEFINITIONS}
   PUBLIC
 )
+
+ADD_CUSTOM_TARGET(link_compute_programl_target ALL
+                  COMMAND ${CMAKE_COMMAND} -E create_symlink ${CMAKE_BINARY_DIR}/compiler_gym/third_party/programl/compute_programl ${CMAKE_SOURCE_DIR}/compiler_gym/third_party/programl/compute_programl)
diff --git a/examples/loop_optimizations_service/service_py/loops_opt_service.py b/examples/loop_optimizations_service/service_py/loops_opt_service.py
@@ -272,7 +272,7 @@ def get_observation(self, observation_space: ObservationSpace) -> Event:
                 [
                     os.path.join(
                         os.path.dirname(__file__),
-                        "../../../compiler_gym/third_party/autophase/compute_autophase",
+                        "../../../compiler_gym/third_party/autophase/compute_autophase-prelinked",
                     ),
                     self._llvm_path,
                 ],
@@ -289,7 +289,7 @@ def get_observation(self, observation_space: ObservationSpace) -> Event:
                 [
                     os.path.join(
                         os.path.dirname(__file__),
-                        "../../../compiler_gym/third_party/autophase/compute_autophase",
+                        "../../../compiler_gym/third_party/autophase/compute_autophase-prelinked",
                     ),
                     self._llvm_path,
                 ],