#include "hip/hip_runtime.h"
# try_compile_clang_cuda(result, cuda_arch)
# Parameters:
#    - result: result variable is set to TRUE if Clang-CUDA compilation worked.
#    - cuda_arch: CUDA architecture used for compilation test
#                 (doesn't have to match the architecture of an available GPU, no code is executed)
function(try_compile_clang_cuda result cuda_arch)
    set(CLANG_CUDA_TEST_SOURCE
"
__global__ void helloworld(int* in, int* out) {
    *out = *in;
}
int main(int argc, char* argv[]) {
    int* in;
    int* out;
    hipMalloc((void**)&in, sizeof(int));
    hipMalloc((void**)&out, sizeof(int));
    hipLaunchKernelGGL(helloworld, dim3(1), dim3(1), 0, 0, in, out);
    hipFree(in);
    hipFree(out);
}
")

    set(SRC_FILE ${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/TryClangCuda.cpp)
    file(WRITE "${SRC_FILE}" "${CLANG_CUDA_TEST_SOURCE}")

    set(CLANG_CUDA_FLAGS "-xcuda --cuda-path=${CUDAToolkit_BIN_DIR}/.. --cuda-gpu-arch=${cuda_arch}")
    try_compile(clang_cuda_works ${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY} ${SRC_FILE}
        COMPILE_DEFINITIONS "${CLANG_CUDA_FLAGS}"
        LINK_LIBRARIES CUDA::cudart
        OUTPUT_VARIABLE CLANG_CUDA_TRY_COMPILE_OUTPUT
        )
    if(NOT clang_cuda_works)
        file(APPEND ${CMAKE_BINARY_DIR}${CMAKE_FILES_DIRECTORY}/CMakeOutput.log
            "Testing CUDA compilation with clang gave the following output:\n${CLANG_CUDA_TRY_COMPILE_OUTPUT}\n\n")
    endif()
    set(${result} ${clang_cuda_works} PARENT_SCOPE)
endfunction()
