Edgar Gabriel
63 라인
1.9 KiB
C++
63 라인
1.9 KiB
C++
/*************************************************************************
|
|
* Copyright (c) 2024 Advanced Micro Devices, Inc. All rights reserved.
|
|
*
|
|
* See LICENSE.txt for license information
|
|
************************************************************************/
|
|
|
|
#ifndef __ALT_RSMI_H__
|
|
#define __ALT_RSMI_H__
|
|
|
|
/*
|
|
** This is a light-weight implementation of the RSMI functionality used in RCCL
|
|
** The code is based on the actual rocm_smi_library code, but extracted to contain only
|
|
** the bits actually required by RCCL.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <dirent.h>
|
|
#include <sys/types.h>
|
|
#include <sys/stat.h>
|
|
#include <unistd.h>
|
|
|
|
#include <iostream>
|
|
#include <fstream>
|
|
#include <sstream>
|
|
#include <cstring>
|
|
#include <map>
|
|
#include <cassert>
|
|
#include <algorithm>
|
|
#include <iomanip>
|
|
|
|
/**
|
|
** This is an exact copy of the IO Link types from rocm_smi.h
|
|
** These definitions are required since we do not know whether the
|
|
** code will also be compiled such that it includes the rocm_smi.h
|
|
** file or not. The values have to be identical however
|
|
*/
|
|
typedef enum _ARSMI_IO_LINK_TYPE {
|
|
ARSMI_IOLINK_TYPE_UNDEFINED = 0, //!< unknown type.
|
|
ARSMI_IOLINK_TYPE_PCIEXPRESS, //!< PCI Express
|
|
ARSMI_IOLINK_TYPE_XGMI, //!< XGMI
|
|
ARSMI_IOLINK_TYPE_NUMIOLINKTYPES, //!< Number of IO Link types
|
|
ARSMI_IOLINK_TYPE_SIZE = 0xFFFFFFFF //!< Max of IO Link types
|
|
} ARSMI_IO_LINK_TYPE;
|
|
|
|
struct ARSMI_linkInfo {
|
|
uint32_t src_node;
|
|
uint32_t dst_node;
|
|
uint64_t hops;
|
|
ARSMI_IO_LINK_TYPE type;
|
|
uint64_t weight;
|
|
uint64_t min_bandwidth;
|
|
uint64_t max_bandwidth;
|
|
};
|
|
typedef struct ARSMI_linkInfo ARSMI_linkInfo;
|
|
|
|
int ARSMI_init (void);
|
|
int ARSMI_get_num_devices (uint32_t *num_devices);
|
|
int ARSMI_dev_pci_id_get(uint32_t dv_ind, uint64_t *bdfid);
|
|
int ARSMI_topo_get_link_info(uint32_t dv_ind_src, uint32_t dv_ind_dst,
|
|
ARSMI_linkInfo *info);
|
|
|
|
#endif
|