generate templated isl types

Author: Sven Verdoolaege

CMakeLists.txt

@@ -225,6 +225,7 @@ message(STATUS "Found ATen.so file: ${ATEN_LIBRARIES}")
 ################################################################################
 # use locally generated C++ bindings
 include_directories(AFTER ${PROJECT_SOURCE_DIR}/isl_interface/include)
+include_directories(AFTER ${CMAKE_CURRENT_BINARY_DIR}/isl_interface/include)
 include_directories(AFTER ${PROJECT_SOURCE_DIR}/third-party/islpp/include)
 include_directories(AFTER ${CMAKE_CURRENT_BINARY_DIR}/third-party/islpp/include)
 add_subdirectory(external/isl)

isl_interface/CMakeLists.txt

@@ -106,6 +106,23 @@ add_custom_command(
   COMMAND cat ${ISL_DIR}/cpp/cpp.h.bot >> ${ISL_CPP_H} || exit 1
   DEPENDS extract_isl_interface
 )
+
+add_executable(generate_template_isl generate_template_isl.cc)
+
+find_program(CLANG_FORMAT_BIN clang-format PATHS ${CLANG_PREFIX}
+  PATH_SUFFIXES bin
+  NO_DEFAULT_PATH)
+
+SET(ISL_TEMPLATE_CPP_H "${CMAKE_CURRENT_BINARY_DIR}/include/isl/template_cpp.h")
+add_custom_command(
+  OUTPUT ${ISL_TEMPLATE_CPP_H}
+  DEPENDS ${ISL_CPP_H}
+  DEPENDS generate_template_isl
+  COMMAND mkdir -p ${CMAKE_CURRENT_BINARY_DIR}/include/isl
+  COMMAND generate_template_isl < ${ISL_CPP_H} > ${ISL_TEMPLATE_CPP_H}
+  COMMAND ${CLANG_FORMAT_BIN} -i ${ISL_TEMPLATE_CPP_H}
+)
+
 # generate_isl_cpp_h is the dependency that should be used
 # by code that depends on the isl C++ bindings.
-add_custom_target(generate_isl_cpp_h DEPENDS ${ISL_CPP_H})
+add_custom_target(generate_isl_cpp_h DEPENDS ${ISL_CPP_H} ${ISL_TEMPLATE_CPP_H})