# Distributed under the OSI-approved BSD 3-Clause License.  See accompanying file
# Copyright.txt or https://cmake.org/licensing for details.

include(FindPackageHandleStandardArgs)

# ----------------------------------------------------------------------------------------#

if(NOT ROCM_PATH AND NOT "$ENV{ROCM_PATH}" STREQUAL "")
    set(ROCM_PATH "$ENV{ROCM_PATH}")
endif()

foreach(_DIR ${ROCmVersion_DIR} ${ROCM_PATH} /opt/rocm /opt/rocm/rocm_smi)
    if(EXISTS ${_DIR})
        get_filename_component(_ABS_DIR "${_DIR}" REALPATH)
        list(APPEND _ROCM_SMI_PATHS ${_ABS_DIR})
    endif()
endforeach()

# ----------------------------------------------------------------------------------------#

find_path(
    rocm-smi_ROOT_DIR
    NAMES include/rocm_smi/rocm_smi.h
    HINTS ${_ROCM_SMI_PATHS}
    PATHS ${_ROCM_SMI_PATHS}
    PATH_SUFFIXES rocm_smi)

mark_as_advanced(rocm-smi_ROOT_DIR)

# ----------------------------------------------------------------------------------------#

find_path(
    rocm-smi_INCLUDE_DIR
    NAMES rocm_smi/rocm_smi.h
    HINTS ${rocm-smi_ROOT_DIR} ${_ROCM_SMI_PATHS}
    PATHS ${rocm-smi_ROOT_DIR} ${_ROCM_SMI_PATHS}
    PATH_SUFFIXES include rocm_smi/include)

mark_as_advanced(rocm-smi_INCLUDE_DIR)

# ----------------------------------------------------------------------------------------#

find_library(
    rocm-smi_LIBRARY
    NAMES rocm_smi64 rocm_smi
    HINTS ${rocm-smi_ROOT_DIR} ${_ROCM_SMI_PATHS}
    PATHS ${rocm-smi_ROOT_DIR} ${_ROCM_SMI_PATHS}
    PATH_SUFFIXES rocm_smi/lib rocm_smi/lib64 lib lib64)

if(rocm-smi_LIBRARY)
    get_filename_component(rocm-smi_LIBRARY_DIR "${rocm-smi_LIBRARY}" PATH CACHE)
endif()

mark_as_advanced(rocm-smi_LIBRARY)

# ----------------------------------------------------------------------------------------#

find_package_handle_standard_args(rocm-smi DEFAULT_MSG rocm-smi_ROOT_DIR
                                  rocm-smi_INCLUDE_DIR rocm-smi_LIBRARY)

# ------------------------------------------------------------------------------#

if(rocm-smi_FOUND)
    add_library(rocm-smi::rocm-smi INTERFACE IMPORTED)
    add_library(rocm-smi::roctx INTERFACE IMPORTED)
    set(rocm-smi_INCLUDE_DIRS ${rocm-smi_INCLUDE_DIR})
    set(rocm-smi_LIBRARIES ${rocm-smi_LIBRARY})
    set(rocm-smi_LIBRARY_DIRS ${rocm-smi_LIBRARY_DIR})

    target_include_directories(rocm-smi::rocm-smi INTERFACE ${rocm-smi_INCLUDE_DIR})
    target_link_libraries(rocm-smi::rocm-smi INTERFACE ${rocm-smi_LIBRARY})

endif()

# ------------------------------------------------------------------------------#

unset(_ROCM_SMI_PATHS)

# ------------------------------------------------------------------------------#