I want to introduce a new flag in DGL source code

i was trying to introduce some hardware related optimization in dgl source code for some specific kernel.
I don’t want to disturb the existing flow for other hardware’s so i was trying with introducing
#ifdef (FLAG_NAME)
//chages
#endif
i add that flag in cmake by add_definitions(-DFLAG_NAME) and set it to off for other hardware
if(NOT CMAKE_SYSTEM_PROCESSOR MATCHES “(FLAG_NAME)”)

message(STATUS “setting flag for HW specific optimizations = ${CMAKE_SYSTEM_PROCESSOR} to OFF.”)

set(FLAG_NAMEOFF)

endif()

something like this.

can someone help me out in this. because it is effecting other hardware’s also. which i don’t want.