pennylane-lightning/cmake/support_catalyst.cmake at fcc5229cf7d6ee6db2caf128cd6f86a7aef6493d · PennyLaneAI/pennylane-lightning · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
###############################################################################################
# This file provides macros to process Catalyst.
###############################################################################################

# Include this only once
include_guard()

macro(FindCatalyst target_name)
    if(LIGHTNING_CATALYST_SRC_PATH)
        if(NOT IS_ABSOLUTE ${LIGHTNING_CATALYST_SRC_PATH})
            message(FATAL_ERROR " LIGHTNING_CATALYST_SRC_PATH=${LIGHTNING_CATALYST_SRC_PATH} must be set to an absolute path")
        endif()
        if(CATALYST_GIT_TAG)
            message(WARN " Setting `LIGHTNING_CATALYST_SRC_PATH=${LIGHTNING_CATALYST_SRC_PATH}` overrides `CATALYST_GIT_TAG=${CATALYST_GIT_TAG}`")
        endif()

        # Acquire local git hash and use for CATALYST_GIT_TAG
        execute_process(COMMAND git rev-parse --short HEAD
            WORKING_DIRECTORY ${LIGHTNING_CATALYST_SRC_PATH}
            OUTPUT_VARIABLE CATALYST_GIT_TAG
        )
        message(INFO " Building against local Catalyst - path: ${LIGHTNING_CATALYST_SRC_PATH} - GIT TAG: ${CATALYST_GIT_TAG}")

        target_include_directories(${target_name} SYSTEM PUBLIC ${LIGHTNING_CATALYST_SRC_PATH}/runtime/lib/backend/common)
        target_include_directories(${target_name} SYSTEM PUBLIC ${LIGHTNING_CATALYST_SRC_PATH}/runtime/include)

    else()
        if(NOT CATALYST_GIT_TAG)
            set(CATALYST_GIT_TAG "b42c86e8b51ef67c3370dc1bd5d6b4c55af58937" CACHE STRING "GIT_TAG value to build Catalyst")
        endif()
        message(INFO " Building against Catalyst GIT TAG ${CATALYST_GIT_TAG}")

        # Fetching /lib/backend/common hpp headers
        set(LIB_BACKEND_COMMON_HEADERS  CacheManager.hpp
                                        Utils.hpp
        )

        foreach(HEADER ${LIB_BACKEND_COMMON_HEADERS})
            string(REGEX REPLACE "\\.[^.]*$" "" HEADER_NAME ${HEADER})
            FetchContent_Declare(
                ${HEADER_NAME}
                URL                 https://raw.githubusercontent.com/PennyLaneAI/catalyst/${CATALYST_GIT_TAG}/runtime/lib/backend/common/${HEADER}
                DOWNLOAD_NO_EXTRACT True
                SOURCE_DIR          ../../include
            )

            FetchContent_MakeAvailable(${HEADER_NAME})
        endforeach()

        # Fetching include hpp headers
        set(INCLUDE_HEADERS DataView.hpp
                        Exception.hpp
                        QuantumDevice.hpp
                        RuntimeCAPI.h
                        Types.h
        )

        foreach(HEADER ${INCLUDE_HEADERS})
            string(REGEX REPLACE "\\.[^.]*$" "" HEADER_NAME ${HEADER})
            FetchContent_Declare(
                ${HEADER_NAME}
                URL                 https://raw.githubusercontent.com/PennyLaneAI/catalyst/${CATALYST_GIT_TAG}/runtime/include/${HEADER}
                DOWNLOAD_NO_EXTRACT True
                SOURCE_DIR          ../../include
            )

            FetchContent_MakeAvailable(${HEADER_NAME})
        endforeach()

        target_include_directories(${target_name} SYSTEM PUBLIC ${CMAKE_CURRENT_BINARY_DIR}/../../include)

    endif()
endmacro()