cmake - add_custom_command for unit test kernel precompile

Author: RichardGe

CMakeLists.txt

@@ -378,10 +378,17 @@ endif()
 
 
 
-# files generated by compile.py
-set(KERNEL_HIPRT_COMP "${BASE_OUTPUT_DIR}/${CMAKE_BUILD_TYPE}/hiprt${version_str_}_${HIP_VERSION_STR}_amd.hipfb")
+# files generated by compile.py and precompile_bitcode.py
+if(WIN32)
+	set(KERNEL_OS_POSTFIX "win")
+else()
+	set(KERNEL_OS_POSTFIX "linux")
+endif()
+set(KERNEL_HIPRT_COMP    "${BASE_OUTPUT_DIR}/${CMAKE_BUILD_TYPE}/hiprt${version_str_}_${HIP_VERSION_STR}_amd.hipfb")                                        # example:  hiprt02005_6.2_amd.hipfb
+set(KERNEL_UNITTEST_COMP "${BASE_OUTPUT_DIR}/${CMAKE_BUILD_TYPE}/hiprt${version_str_}_${HIP_VERSION_STR}_precompiled_bitcode_${KERNEL_OS_POSTFIX}.hipfb")   # example:  hiprt02005_6.2_precompiled_bitcode_win.hipfb
 set(KERNEL_OROCHI_COMP "${BASE_OUTPUT_DIR}/${CMAKE_BUILD_TYPE}/oro_compiled_kernels.hipfb")
 
+
 # precompile kernels:
 if(PRECOMPILE)
 
@@ -434,11 +441,29 @@ if(PRECOMPILE)
 		)
 
 	if(NOT NO_UNITTEST)
-		message(">> Execute: ${PYTHON_EXECUTABLE} precompile_bitcode.py ${CUDA_OPTION} --hipSdkPath \"${HIP_FINAL_PATH}\"")
-		execute_process(
+
+		set(unittest_kernel_source
+		${CMAKE_SOURCE_DIR}/test/bitcodes/custom_func_table.cpp
+		${CMAKE_SOURCE_DIR}/test/bitcodes/unit_test.cpp
+		)
+		
+		message(">> add_custom_command: ${PYTHON_EXECUTABLE} precompile_bitcode.py ${CUDA_OPTION} --hipSdkPath \"${HIP_FINAL_PATH}\"")
+		add_custom_command(
+			OUTPUT ${KERNEL_UNITTEST_COMP}
 			COMMAND ${PYTHON_EXECUTABLE} precompile_bitcode.py ${CUDA_OPTION} --hipSdkPath ${HIP_FINAL_PATH}
+			DEPENDS ${KERNEL_HIPRT_COMP}  # Ensure compile.py has already run.
 			WORKING_DIRECTORY ${CMAKE_CURRENT_SOURCE_DIR}/scripts/bitcodes
+			COMMENT "Precompiling unit tests kernels via precompile_bitcode.py"
+			VERBATIM
+			DEPENDS ${unittest_kernel_source}
 			)
+			
+		# create the 'precompile_unittest_kernels' project
+		add_custom_target(precompile_unittest_kernels ALL
+			DEPENDS ${KERNEL_UNITTEST_COMP}
+			)
+			
+		add_dependencies(${HIPRT_NAME} precompile_unittest_kernels)
 	endif()