Skip to content

GitLab

"...consistency_distillation/train_lcm_distill_sdxl_wds.py" did not exist on "4ff7264d9bbb13648f8d4fc819ce38a5f2a168e8"
Switch branch/tag
Find file Normal viewHistoryPermalink
TargetFlags.cmake 1.66 KB
Newer Older
Qianfeng's avatar
Add online compilation for dynamic kernels (#37)  
Qianfeng committed
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 

function(get_target_property2 VAR TARGET PROPERTY)
    get_target_property(_pflags ${TARGET} ${PROPERTY})
    if(_pflags)
        set(${VAR} ${_pflags} PARENT_SCOPE)
    else()
        set(${VAR} "" PARENT_SCOPE)
    endif()
endfunction()


macro(append_flags FLAGS TARGET PROPERTY PREFIX)
    get_target_property2(_pflags ${TARGET} ${PROPERTY})
    foreach(FLAG ${_pflags})
        if(TARGET ${FLAG})
            target_flags(_pflags2 ${FLAG})
            string(APPEND ${FLAGS} " ${_pflags2}")
        else()
            string(APPEND ${FLAGS} " ${PREFIX}${FLAG}")
        endif()
    endforeach()
endmacro()

macro(append_link_flags FLAGS TARGET PROPERTY)
    get_target_property2(_pflags ${TARGET} ${PROPERTY})
    foreach(FLAG ${_pflags})
        if(TARGET ${FLAG})
            target_flags(_pflags2 ${FLAG})
            string(APPEND ${FLAGS} " ${_pflags2}")
        elseif(FLAG MATCHES "^-.*")
            string(APPEND ${FLAGS} " ${FLAG}")
        elseif(EXISTS ${FLAG})
            string(APPEND ${FLAGS} " ${FLAG}")
        else()
            string(APPEND ${FLAGS} " -l${FLAG}")
        endif()
    endforeach()
endmacro()

function(target_flags FLAGS TARGET)
    set(_flags)
    append_flags(_flags ${TARGET} "INTERFACE_COMPILE_OPTIONS" "")
    append_flags(_flags ${TARGET} "INTERFACE_COMPILE_DEFINITIONS" "-D")
    append_flags(_flags ${TARGET} "INTERFACE_INCLUDE_DIRECTORIES" "-isystem ")
    append_flags(_flags ${TARGET} "INTERFACE_LINK_DIRECTORIES" "-L ")
    append_flags(_flags ${TARGET} "INTERFACE_LINK_OPTIONS" "")
    append_link_flags(_flags ${TARGET} "INTERFACE_LINK_LIBRARIES" "")
    # message("_flags: ${_flags}")
    set(${FLAGS} ${_flags} PARENT_SCOPE)
endfunction()