class TableFields:
ACC_PMU = [
"accId", "readBwLevel", "writeBwLevel", "readOstLevel", "writeOstLevel", "timestampNs", "deviceId"
]
AICORE_FREQ = [
"deviceId", "timestampNs", "freq"
]
CANN_API = [
"startNs", "endNs", "type", "globalTid", "connectionId", "name"
]
COMMUNICATION_OP = [
"opName", "startNs", "endNs", "connectionId", "groupName", "opId", "relay", "retry", "dataType", "algType",
"count", "opType", "deviceId"
]
COMMUNICATION_TASK_INFO = [
"name", "globalTaskId", "taskType", "planeId", "groupName", "notifyId", "rdmaType", "srcRank", "dstRank", "transportType",
"size", "dataType", "linkType", "opId", "isMaster", "bandwidth"
]
COMPUTE_TASK_INFO = [
"name", "globalTaskId", "blockNum", "mixBlockNum", "taskType", "opType", "inputFormats", "inputDataTypes",
"inputShapes", "outputFormats", "outputDataTypes", "outputShapes", "attrInfo", "opState", "hf32Eligible"
]
CONNECTION_IDS = [
"id", "connectionId"
]
CPU_USAGE = [
"timestampNs", "cpuId", "usage"
]
ENUM_API_TYPE = [
"id", "name"
]
ENUM_HCCL_DATA_TYPE = [
"id", "name"
]
ENUM_HCCL_LINK_TYPE = [
"id", "name"
]
ENUM_HCCL_RDMA_TYPE = [
"id", "name"
]
ENUM_HCCL_TRANSPORT_TYPE = [
"id", "name"
]
ENUM_MEMCPY_OPERATION = [
"id", "name"
]
ENUM_MODULE = [
"id", "name"
]
ENUM_MSTX_EVENT_TYPE = [
"id", "name"
]
HBM = [
"deviceId", "timestampNs", "bandwidth", "hbmId", "type"
]
HCCS = [
"deviceId", "timestampNs", "txThroughput", "rxThroughput"
]
HOST_INFO = [
"hostUid", "hostName"
]
HOST_MEM_USAGE = [
"timestampNs", "usage"
]
LLC = [
"deviceId", "llcId", "timestampNs", "hitRate", "throughput", "mode"
]
MEMCPY_INFO = [
"globalTaskId", "size", "memcpyOperation"
]
MEMORY_RECORD = [
"component", "timestamp", "totalAllocated", "totalReserved", "totalActive", "streamPtr", "deviceId"
]
META_DATA = [
"name", "value"
]
NETDEV_STATS = [
"deviceId", "timestampNs", "macTxPfcPkt", "macRxPfcPkt", "macTxByte", "macTxBandwidth", "macRxByte", "macRxBandwidth",
"macTxBadByte", "macRxBadByte", "roceTxPkt", "roceRxPkt", "roceTxErrPkt", "roceRxErrPkt", "roceTxCnpPkt", "roceRxCnpPkt",
"roceNewPktRty", "nicTxByte", "nicTxBandwidth", "nicRxByte", "nicRxBandwidth"
]
NIC = [
"deviceId", "timestampNs", "bandwidth", "rxPacketRate", "rxByteRate", "rxPackets", "rxBytes", "rxErrors", "rxDropped",
"txPacketRate", "txByteRate", "txPackets", "txBytes", "txErrors", "txDropped", "funcId"
]
NPU_INFO = [
"id", "name"
]
NPU_MEM = [
"type", "ddr", "hbm", "timestampNs", "deviceId"
]
NPU_MODULE_MEM = [
"moduleId", "timestampNs", "totalReserved", "deviceId"
]
NPU_OP_MEM = [
"operatorName", "addr", "type", "size", "timestampNs", "globalTid", "totalAllocate", "totalReserve", "component",
"deviceId"
]
OP_MEMORY = [
"name", "size", "allocationTime", "releaseTime", "activeReleaseTime", "duration", "activeDuration", "allocationTotalAllocated",
"allocationTotalReserved", "allocationTotalActive", "releaseTotalAllocated", "releaseTotalReserved", "releaseTotalActive",
"streamPtr", "deviceId"
]
PCIE = [
"deviceId", "timestampNs", "txPostMin", "txPostMax", "txPostAvg", "txNonpostMin", "txNonpostMax", "txNonpostAvg", "txCplMin",
"txCplMax", "txCplAvg", "txNonpostLatencyMin", "txNonpostLatencyMax", "txNonpostLatencyAvg", "rxPostMin", "rxPostMax",
"rxPostAvg", "rxNonpostMin", "rxNonpostMax", "rxNonpostAvg", "rxCplMin", "rxCplMax", "rxCplAvg"
]
PYTORCH_API = [
"startNs", "endNs", "globalTid", "connectionId", "name", "sequenceNumber", "fwdThreadId", "inputDtypes", "inputShapes",
"callchainId", "type"
]
PYTORCH_CALLCHAINS = [
"id", "stack", "stackDepth"
]
QOS = [
"deviceId", "eventName", "bandwidth", "timestampNs"
]
ROCE = [
"deviceId", "timestampNs", "bandwidth", "rxPacketRate", "rxByteRate", "rxPackets", "rxBytes", "rxErrors", "rxDropped",
"txPacketRate", "txByteRate", "txPackets", "txBytes", "txErrors", "txDropped", "funcId"
]
SESSION_TIME_INFO = [
"startTimeNs", "endTimeNs"
]
SOC_BANDWIDTH_LEVEL = [
"l2BufferBwLevel", "mataBwLevel", "timestampNs", "deviceId"
]
STEP_TIME = [
"id", "startNs", "endNs"
]
STRING_IDS = [
"id", "value"
]
TASK = [
"startNs", "endNs", "deviceId", "connectionId", "globalTaskId", "globalPid", "taskType", "contextId", "streamId", "taskId", "modelId"
]
TASK_PMU_INFO = [
"globalTaskId", "name", "value"
]
RANK_DEVICE_MAP = [
"rankId", "deviceId"
]
ClusterCommunicationBandwidth = ["step", "rank_id", "hccl_op_name", "group_name", "band_type", "transit_size",
"transit_time", "bandwidth", "large_packet_ratio", "package_size", "count", "total_duration"]
ClusterCommunicationMatrix = ["step", "hccl_op_name", "group_name", "src_rank", "dst_rank", "transport_type", "op_name",
"transit_size", "transit_time", "bandwidth"]
ClusterCommunicationTime = ["step", "rank_id", "hccl_op_name", "group_name", "start_timestamp", "elapsed_time", "transit_time",
"wait_time", "synchronization_time", "idle_time", "synchronization_time_ratio", "wait_time_ratio"]
ClusterStepTraceTime = ["step", "type", "index", "computing", "communication_not_overlapped", "overlapped", "communication",
"free", "stage", "bubble", "communication_not_overlapped_and_exclude_receive", "preparing", "dp_index",
"pp_index", "tp_index"]
CommunicationGroupMapping = ["type", "rank_set", "group_name", "group_id", "pg_name"]
HostInfo = ["hostUid", "hostName"]
RankDeviceMap = ["rankId", "deviceId", "hostUid", "profilePath"]
@classmethod
def get_fields(cls, name):
"""
Retrieve the list of field names with the specified name.
:param name: The name of the field list.
:return: The corresponding list of field names, returning None if the name does not exist.
"""
if hasattr(cls, name):
return getattr(cls, name)
else:
return None