# Setup hook for ascend-toolkit (sourced when package is in buildInputs)
addAscendEnv() {
    local toolkit="${1:?ascend-toolkit path required}"

    export ASCEND_TOOLKIT_HOME="${toolkit}"
    export ASCEND_HOME_PATH="${toolkit}"
    export ASCEND_OPP_PATH="${toolkit}/opp"
    export ASCEND_AICPU_PATH="${toolkit}"
    export TOOLCHAIN_HOME="${toolkit}/toolkit"

    addToSearchPath PATH "${toolkit}/bin"
    addToSearchPath PATH "${toolkit}/tools/ccec_compiler/bin"
    addToSearchPath PATH "${toolkit}/tools/profiler/bin"
    addToSearchPath PATH "${toolkit}/tools/ascend_system_advisor/asys"
    addToSearchPath PATH "${toolkit}/tools/show_kernel_debug_data"
    addToSearchPath PATH "${toolkit}/tools/msobjdump"

    addToSearchPath LD_LIBRARY_PATH "${toolkit}/lib64"
    addToSearchPath LD_LIBRARY_PATH "${toolkit}/lib64/plugin/opskernel"
    addToSearchPath LD_LIBRARY_PATH "${toolkit}/lib64/plugin/nnengine"
    addToSearchPath LD_LIBRARY_PATH "${toolkit}/opp/built-in/op_impl/ai_core/tbe/op_tiling/lib/linux/x86_64"
    addToSearchPath LD_LIBRARY_PATH "${toolkit}/tools/aml/lib64"
    addToSearchPath LD_LIBRARY_PATH "${toolkit}/tools/aml/lib64/plugin"
    addToSearchPath LD_LIBRARY_PATH "${toolkit}/devlib"

    addToSearchPath PYTHONPATH "${toolkit}/python/site-packages"
    addToSearchPath PYTHONPATH "${toolkit}/opp/built-in/op_impl/ai_core/tbe"

    addToSearchPath CMAKE_PREFIX_PATH "${toolkit}/lib64/cmake"
    addToSearchPath CMAKE_PREFIX_PATH "${toolkit}/toolkit/tools/tikicpulib/lib/cmake"
}

addAscendEnv "@ASCEND_TOOLKIT@"