Using Dump in the Graph Mode

View Source On Gitee

The input and output of the operator can be saved for debugging through the data dump when the training result deviates from the expectation.

  • For dynamic graph mode, the forward process can utilize Python's native execution capabilities, allowing users to view and record the corresponding inputs and outputs during the execution of the network script. The JIT and backward processes, which are part of graph compilation, can use synchronous dump functionality to save the input and output data of operators to disk files.

  • For the static graph mode, MindSpore provides the Dump function to save the graph and the input and output data of the operator during model training to a disk file.

MindSpore provides two Dump modes:

  • Synchronous Dump: After the operator is dispatched, the Host side performs stream synchronization, initiates data copying from the Device side, and saves it to a file.

  • Asynchronous Dump: Specifically developed for Ascend. After the operator execution is completed, the Device side actively initiates data dumping to disk.

Different modes require different configuration files, and the generated data formats also differ:

  • For GPU/CPU backends and Ascend backend with compilation levels O0/O1, it is recommended to use synchronous dump. For details, refer to synchronous dump step. For Ascend backend with compilation level O2, it is recommended to use asynchronous dump. For details, refer to asynchronous dump step.

  • Currently, Dump does not support heterogeneous training. If Dump is enabled in a heterogeneous training scenario, the generated Dump data object directory may not match the expected directory structure.

The support for Synchronous Dump on Ascend backend is shown in the table below (GPU/CPU backend refers to the O0/O1)

Synchronous Dump Feature O0/O1 O2
Full Dump Full network data dump Supported Not Supported
Partial Data Dump Statistics Dump Supports both host and device modes1 Not Supported
Data Sampling Dump Supported2 Not Supported
Overflow Dump Dump overflow operators Supported2 Not Supported
Conditional Dump Specify Operator Name Supported Not Supported
Specify Iteration Supported Not Supported
Specify Device Supported Not Supported
Specify File Format Not Applicable Not Supported
set_dump Supported2 Not Supported
Auxiliary Information Dump Graph IR Dump Supported Not Supported
Execution Sequence Dump Supported Not Supported

  1. In terms of statistics, the computing speed of the device is faster than that of the host(currently only supported on Ascend backend), but the host has more statistical indicators than the device. Refer to the statistic_category option for details.

  2. Only supported on the Ascend backend.

The support for Asynchronous Dump on Ascend backend is shown in the table below (not supported on GPU/CPU backend).

Asynchronous Dump Feature O0/O1 O2
Full Dump Full network data dump Supported, but without full_name information Supported
Partial Data Dump Statistics Dump Host mode only Host mode only
Data Sampling Dump Not Supported Not Supported
Overflow Dump Dump overflow operators Not Supported Supported
Conditional Dump Specify Operator Name Not Supported Supported
Specify Iteration Supported Supported
Specify Device Supported Supported
Specify File Format Supported Supported
set_dump Not Supported Not Supported
Auxiliary Information Dump Graph IR Dump Not Supported Not Supported
Execution Sequence Dump Not Supported Not Supported

Synchronous Dump

Dump Step

  1. Create a configuration file in json format , and the name and location of the JSON file can be customized.

    {
    "common_dump_settings": {
        "op_debug_mode": 0,
        "dump_mode": 0,
        "path": "/absolute_path",
        "net_name": "ResNet50",
        "iteration": "0|5-8|100-120",
        "saved_data": "tensor",
        "input_output": 0,
        "kernels": ["Default/Conv-op12"],
        "support_device": [0,1,2,3,4,5,6,7],
        "statistic_category": ["max", "min", "l2norm"]
    },
    "e2e_dump_settings": {
        "enable": true,
        "trans_flag": true,
        "stat_calc_mode": "host"
    }
}

    • op_debug_mode: This attribute is used for operator overflow or operator exception debugging. 0: save all operators or specified operators; 3: only save overflow operators; 4: only save input of the exception operator. Set it to 0 when the data is dumped. If it is not set to 0, only the data of the overflow operator or exception operator will be dumped. Default: 0.

    • dump_mode: 0: all operator data in the network dumped out; 1: the operator data specified in Dump "kernels"; 2: dump target and its contents using mindspore.set_dump. Specified data dump is supported only when "dump_mode' is set to 0.

    • path: The absolute path to Dump saved data.

    • net_name: The customized net name: "ResNet50".

    • iteration: Specify the iterations of data required to be dumped, type is string. Use "|" to separate the step data of different intervals to be saved. For example, "0 | 5-8 | 100-120" represents dump the data of the 1st, 6th to 9th, and 101st to 121st steps. If iteration set to "all", data of every iteration will be dumped. Specified iteration dump is supported only when "op_debug_mode" is set to 0 or 3, not supported when when "op_debug_mode" is set to 4.

    • saved_data: Specify what data is to be dumped, type is string. Use "tensor" to indicate complete tensor data Dumped, use "statistic" to dump tensor statistics, use "full" to dump both tensor data and statistics. Synchronous statistics dump is only supported on GPU and Ascend. Using "statistic" or "full" on CPU will result in exception. Default setting is "tensor". Statistic dump is only supported when "op_debug_mode" is set to 0.

    • input_output: 0: dump input and output of kernel, 1:dump input of kernel, 2:dump output of kernel. Only input of kernel can be saved when "op_debug_mode" is set to 4.

    • kernels: This item can be configured in three formats:

      1. List of operator names. Turn on the IR save switch set_context(save_graphs=2) and execute the network to obtain the operator name from the generated trace_code_graph_{graph_id}IR file. For details, please refer to Saving IR. Note that whether setting set_context(save_graphs=2) may cause the different IDs of the same operator, so when dump specified operators, keep this setting unchanged after obtaining the operator name. Or you can obtain the operator names from the file ms_output_trace_code_graph_{graph_id}.ir saved by Dump. Refer to Synchronous Dump Data Object Directory.

      2. You can also specify an operator type. When there is no operator scope information or operator id information in the string, the background considers it as an operator type, such as "conv". The matching rule of operator type is: when the operator name contains an operator type string, the matching is considered successful (case insensitive). For example, "conv" can match operators "Conv2D-op1234" and "Conv3D-op1221".

      3. Regular expressions are supported. When the string conforms to the format of "name-regex(xxx)", it would be considered a regular expression. For example, "name-regex(Default/.+)" can match all operators with names starting with "Default/".

    • support_device: Supported devices, default setting is [0,1,2,3,4,5,6,7]. You can specify specific device ids to dump specific device data. This configuration parameter is invalid on the CPU, because there is no concept of device on the CPU, but it is still need to reserve this parameter in the json file.

    • statistic_category: This attribute is used by users to configure the category of statistical information to be saved, and only takes effect when saving statistical information is enabled(i.e.saved_data is set to statistic or full). The type is a string list, where the optional values of the strings are as follows:

      • "max": represents the maximum value of the elements in tensor, supporting both device and host statistics;

      • "min": represents the minimum value of the elements in tensor, supporting both device and host statistics;

      • "avg": represents the average value of elements in tensor, supporting device and host statistics;

      • "count": represents the number of the elements in tensor;

      • "negative zero count": represents the number of the elements which is less then zero in tensor;

      • "positive zero count": represents the number of the elements which is greater then zero in tensor;

      • "nan count": represents the number of Nan elements in the tensor;

      • "negative inf count": represents the number of -Inf elements in the tensor;

      • "positive inf count": represents the number of +Inf elements in the tensor;

      • "zero count": represents the number of zero elements in the tensor;

      • "md5": represents the MD5 value of the tensor;

      • "l2norm": represents L2Norm value of the tensor, supporting both device and host statistics.

      Except for those marked as supporting device statistics, other statistics can be collected only on the host. This field is optional, with default values of ["max", "min", "l2norm"].

    • enable: When set to true, enable Synchronous Dump. When set to false, asynchronous dump will be used on Ascend and synchronous dump will still be used on GPU.

    • trans_flag: Enable trans flag. Transform the device data format into NCHW. If it is True, the data will be saved in the 4D format (NCHW) format on the Host side; if it is False, the data format on the Device side will be retained. Default: True.

    • stat_calc_mode: Select the backend for statistical calculations. Options are "host" and "device". Choosing "device" enables device computation of statistics, currently only effective on Ascend, and supports only min/max/avg/l2norm statistics.

    • sample_mode: Setting it to 0 means the sample dump function is not enabled. Enable the sample dump function in graph compilation with optimization level O0. This field is effective only when "op_debug_mode" is set to 0, sample dump cannot be enabled in other scene.

    • sample_num: Used to control the size of sample in sample dump. The default value is 100.

  2. Set Dump environment variable.

    Specify the json configuration file of Dump.

    export MINDSPORE_DUMP_CONFIG=${xxx}

    "xxx" represents the absolute path to the configuration file.

    export MINDSPORE_DUMP_CONFIG=/path/to/data_dump.json

    If the path field is not set or set to an empty string in the Dump configuration file, you also need to configure the environment variable MS_DIAGNOSTIC_DATA_PATH.

    export MS_DIAGNOSTIC_DATA_PATH=${yyy}

    Then "$MS_DIAGNOSTIC_DATA_PATH/debug_dump" is regarded as path. If the path field is set in Dump configuration file, the actual value of the field is still the same.

    Note:

    • Set the environment variables before executing the training script. Setting environment variables during training will not take effect.

    • Dump environment variables need to be configured before calling mindspore.communication.init.

  3. Execute the training script to dump data.

    After the training is started, if the MINDSPORE_DUMP_CONFIG environment variable is correctly configured, the content of the configuration file will be read and the operator data will be saved according to the data storage path specified in the Dump configuration. In synchronous mode, if you want to dump data in GPU environment, you must use the non-data sink mode (set the dataset_sink_mode parameter in model.train or DatasetHelper to False) to ensure that you can get the dump data of each step. If model.train or DatasetHelper is not called in the script, the default is non-data sinking mode. Using the Dump function will automatically generate the IR file of the final execution graph.

    You can set set_context(reserve_class_name_in_scope=False) in your training script to avoid dump failure because of file name is too long.

  4. Read and parse synchronous dump data through numpy.load, refer to Introduction to Synchronous Dump Data File.

Introduction to Data Object Directory and Data File

After starting the training, the data objects saved by the synchronous Dump include the final execution graph (ms_output_trace_code_graph_{graph_id}.ir file) and the input and output data of the operators in the graph. The data directory structure is as follows:

{path}/
    - rank_{rank_id}/
        - .dump_metadata/
        - {net_name}/
            - {graph_id}/
                - {iteration_id}/
                    {op_type}.{op_name}.json
                    statistic.csv
                    {op_type}.{op_name}.{task_id}.{stream_id}.{timestamp}.{input_output_index}.{slot}.{format}.npy
                - constants/
                    Parameter.data-{data_id}.0.0.{timestamp}.output.0.DefaultFormat.npy
            ...
        - graphs/
            ms_output_trace_code_graph_{graph_id}.pb
            ms_output_trace_code_graph_{graph_id}.ir
        - execution_order/
            ms_execution_order_graph_{graph_id}.csv
            ms_global_execution_order_graph_{graph_id}.csv

  • path: the absolute path set in the data_dump.json configuration file.

  • rank_id: the id of the logic device.

  • net_name: the network name set in the data_dump.json configuration file.

  • graph_id: the id of the training graph.

  • iteration_id: the iteration of the training.

  • op_type: the type of the operator.

  • op_name: the name of the operator.

  • task_id: the id of the task.

  • stream_id: the id of the stream.

  • timestamp: the time stamp.

  • input_output_index : the index of input or output. For example, output_0 means that the file is the data of the first output Tensor of the operator.

  • slot: the id of the slot.

  • format: the format of the data.

  • data_id: the id of constant data.

For multi-graph networks, due to the control flow, some subgraphs may not be executed, but Dump only saves the executed nodes, so the {graph_id} in the .pb file name in the graphs directory does not necessarily exist in the {graph_id} directory under {net_name}.

Only when saved_data is "statistic" or "full", statistic.csv is generated. Only when saved_data is "tensor" or "full", {op_type}. {op_name}. {task_id}. {stream_id}. {timestamp}. {input_output_index}. {slot}. {format}.npy named complete tensor information is generated.

Only when save_kernel_args is True, {op_type}.{op_name}.json is generated and the params of the corresponding operators is saved.

The data file generated by the synchronous Dump is a binary file with the suffix .npy, and the file naming format is:

{op_type}.{op_name}.{task_id}.{stream_id}.{timestamp}.{input_output_index}.{slot}.{format}.npy

The constant data file generated by the synchronous Dump is in the same format as data file, whereas {op_type}, {task_id}, {stream_id}, {input_output_index}, {slot}, {format} are unchanged for all constant data. Note, non-Tensor type will not generate data file. This function is not supported in the Ascend scenario.

Parameter.data-{data_id}.0.0.{timestamp}.output.0.DefaultFormat.npy

The {iteration_id} directory may also save files starting with Parameter (parameters such as weight and bias will be saved as files starting with Parameter), while Parameter files will not be saved on Ascend.

User can use Numpy interface numpy.load to read the data.

The statistics file generated by the synchronous dump is named statistic.csv. This file stores key statistics for all tensors dumped under the same directory as itself (with the file names {op_type}.{op_name}.{task_id}.{stream_id}.{timestamp}.{input_output_index}.{slot}.{format}.npy). Each row in statistic.csv summarizes a single tensor, each row contains the statistics: Op Type, Op Name, Task ID, Stream ID, Timestamp, IO, Slot, Data Size, Data Type, Shape, and statistics items configured by the user. Note that opening this file with Excel may cause data to be displayed incorrectly. Please use commands like vi or cat, or use Excel to import csv from text for viewing.

The suffixes of the final execution graph files generated by synchronous Dump are .pb and .ir respectively, and the file naming format is:

ms_output_trace_code_graph_{graph_id}.pb
ms_output_trace_code_graph_{graph_id}.ir

The files with the suffix .ir can be opened and viewed by the vi command.

The suffix of the node execution sequence file generated by the synchronous Dump is .csv, and the file naming format is:

ms_execution_order_graph_{graph_id}.csv

The suffix of the graph execution history file is .csv. The file naming format is:

ms_global_execution_order_graph_{graph_id}.csv

This file stores the list of iterations in which the graph was executed. After the graph is compiled, it may be split into multiple sub-graphs. Since sub-graphs share the same graph execution history with root graph, only root graph will generate an execution history file. This function is not supported on Ascend.

.dump_metadata records the original training information(the directory is not available for Ascend backend), and data_dump.json saves the dump configuration set by the user.

Data Analysis Sample

In order to better demonstrate the process of using dump to save and analyze data, we provide a set of complete sample script , you only need to execute bash dump_sync_dump.sh for synchronous dump.

After the graph corresponding to the script is saved to the disk through the Dump function, the final execution graph file ms_output_trace_code_graph_{graph_id}.ir will be generated. This file saves the stack information of each operator in the corresponding graph, and records the generation script corresponding to the operator.

Take AlexNet script as an example:

...
def conv(in_channels, out_channels, kernel_size, stride=1, padding=0, pad_mode="valid"):
    weight = weight_variable()
    return nn.Conv2d(in_channels, out_channels,
                     kernel_size=kernel_size, stride=stride, padding=padding,
                     weight_init=weight, has_bias=False, pad_mode=pad_mode)


def fc_with_initialize(input_channels, out_channels):
    weight = weight_variable()
    bias = weight_variable()
    return nn.Dense(input_channels, out_channels, weight, bias)


def weight_variable():
    return TruncatedNormal(0.02)


class AlexNet(nn.Cell):
    """
    Alexnet
    """

    def __init__(self, num_classes=10, channel=3):
        super(AlexNet, self).__init__()
        self.conv1 = conv(channel, 96, 11, stride=4)
        self.conv2 = conv(96, 256, 5, pad_mode="same")
        self.conv3 = conv(256, 384, 3, pad_mode="same")
        self.conv4 = conv(384, 384, 3, pad_mode="same")
        self.conv5 = conv(384, 256, 3, pad_mode="same")
        self.relu = nn.ReLU()
        self.max_pool2d = nn.MaxPool2d(kernel_size=3, stride=2)
        self.flatten = nn.Flatten()
        self.fc1 = fc_with_initialize(6 * 6 * 256, 4096)
        self.fc2 = fc_with_initialize(4096, 4096)
        self.fc3 = fc_with_initialize(4096, num_classes)

    def construct(self, x):
        """
        The construct function.

        Args:
           x(int): Input of the network.

        Returns:
           Tensor, the output of the network.
        """
        x = self.conv1(x)
        x = self.relu(x)
        x = self.max_pool2d(x)
        x = self.conv2(x)
        x = self.relu(x)
        x = self.max_pool2d(x)
        x = self.conv3(x)
        x = self.relu(x)
        x = self.conv4(x)
        x = self.relu(x)
        x = self.conv5(x)
        x = self.relu(x)
        x = self.max_pool2d(x)
        x = self.flatten(x)
        x = self.fc1(x)
        x = self.relu(x)
        x = self.fc2(x)
        x = self.relu(x)
        x = self.fc3(x)
        return x
...

If the user wants to view the code at line 175 in the script:

x = self.conv3(x)

After executing the network training, you can find multiple operator information corresponding to the line of code from the final execution graph (ms_output_trace_code_graph_{graph_id}.ir file). The content of the file corresponding to Conv2D-op12 is as follows:

  %20(equivoutput) = Conv2D(%17, %19) {instance name: conv2d} primitive_attrs: {IsFeatureMapInputList: (0), kernel_size: (3, 3), mode: 1, out_channel: 384, input_names: [
x, w],    pri_format: NC1HWC0, pad: (0, 0, 0, 0), visited: true, pad_mod: same, format: NCHW,  pad_list: (1, 1, 1, 1), precision_flag: reduce, groups: 1, output_used_num:
(1), stream_id:     0, stride: (1, 1, 1, 1), group: 1, dilation: (1, 1, 1, 1), output_names: [output], IsFeatureMapOutput: true, ms_function_graph: true}
       : (<Tensor[Float32], (32, 256, 13, 13)>, <Tensor[Float32], (384, 256, 3, 3)>) -> (<Tensor[Float32], (32, 384, 13, 13)>)
       : (<Float16xNC1HWC0[const vector][32, 16, 13, 13, 16]>, <Float16xFracZ[const vector][144, 24, 16, 16]>) -> (<Float32xNC1HWC0[const vector][32, 24, 13, 13, 16]>)
       : full_name_with_scope: (Default/network-WithLossCell/_backbone-AlexNet/conv3-Conv2d/Conv2D-op12)
       ...
       # In file ./tain_alexnet.py(175)/        x = self.conv3(x)/
       ...

The meanings of the lines in the file content shown above are as follows:

  • The input and output of the operator on the Host side (the first line) and the Device side (the second line, some operators may not exist). It can be seen from the execution graph that the operator has two inputs (left side of the arrow) and one output (right side of the arrow).

       : (<Tensor[Float32], (32, 256, 13, 13)>, <Tensor[Float32], (384, 256, 3, 3)>) -> (<Tensor[Float32], (32, 384, 13, 13)>)
   : (<Float16xNC1HWC0[const vector][32, 16, 13, 13, 16]>, <Float16xFracZ[const vector][144, 24, 16, 16]>) -> (<Float32xNC1HWC0[const vector][32, 24, 13, 13, 16]>)

  • Operator name. It can be seen from the execution graph that the full name of the operator in the final execution graph is Default/network-WithLossCell/_backbone-AlexNet/conv3-Conv2d/Conv2D-op12.

    : (Default/network-WithLossCell/_backbone-AlexNet/conv3-Conv2d/Conv2D-op12)

  • The training script code corresponding to the operator. By searching the training script code to be queried, multiple matching operators can be found.

    # In file {Absolute path of model_zoo}/official/cv/alexnet/src/alexnet.py(175)/        x = self.conv3(x)/

Through the operator name and input and output information, you can find the only corresponding Tensor data file. For example, if you want to view the dump file corresponding to the first output data of the Conv2D-op12 operator, you can obtain the following information:

  • operator_name: Conv2D-op12.

  • input_output_index: output.0 indicates that the file is the data of the first output Tensor of the operator.

  • slot: 0, this tensor only has one slot.

Search for the corresponding file name in the data object file directory saved by Dump: Conv2d.Conv2D-op12.0.0.1623124369613540.output.0.DefaultFormat.npy.

When restoring data, execute:

import numpy
numpy.load("Conv2D.Conv2D-op12.0.0.1623124369613540.output.0.DefaultFormat.npy")

Generate the numpy.array data.

Asynchronous Dump

MindSpore provides debugging capabilities for large networks through asynchronous dumps on Ascend.

Dump Step

  1. Create configuration file:data_dump.json.

    The name and location of the JSON file can be customized.

    {
    "common_dump_settings": {
        "op_debug_mode": 0,
        "dump_mode": 0,
        "path": "/absolute_path",
        "net_name": "ResNet50",
        "iteration": "0|5-8|100-120",
        "saved_data": "tensor",
        "input_output": 0,
        "kernels": ["Default/Conv-op12"],
        "support_device": [0,1,2,3,4,5,6,7],
        "statistic_category": ["max", "min", "l2norm"],
        "file_format": "npy"
    }
}

    • op_debug_mode: This attribute is used for operator overflow debugging. 0: disable overflow check function; 3: enable overflow check function; 4: enable the lightweight exception dump function (this feature only takes effect when using ACL dump). Set it to 0 when Dump data is processed. If it is not set to 0, only the data of the overflow operator or exception operator will be dumped.

    • dump_mode: 0: all operator data in the network dumped out; 1: dump kernels data in kernels list. When overflow detection is enabled, the setting of this field becomes invalid, and Dump only saves the data of the overflow node. Specified data dump is supported only when "dump_mode' is set to 0.

    • path: The absolute path to save Dump data. When the graph compilation level is O0, MindSpore will create a new subdirectory for each step in the path directory.

    • net_name: The customized net name: "ResNet50".

    • iteration: Specify the iterations to dump, type is string. Use "|" to separate the step data of different intervals to be saved. For example, "0 | 5-8 | 100-120" represents dump the data of the 1st, 6th to 9th, and 101st to 121st steps. If iteration set to "all", data of every iteration will be dumped. Specified iteration dump is supported only when "op_debug_mode" is set to 0, not supported when when "op_debug_mode" is set to 3 or 4.

    • saved_data: Specify what data is to be dumped, type is string. Use "tensor" to dump tensor data, use "statistic" to dump tensor statistics, use "full" to dump both tensor data and statistics. Default setting is "tensor". Asynchronous statistics dump is only supported when file_format is set to npy, using "statistic" or "full" when file_format is set to bin will result in exception. Statistic dump is only supported when "op_debug_mode" is set to 0.

    • input_output: When set to 0, it means to Dump the operator's input and output; when set to 1, it means to Dump the operator's input; setting it to 2 means to Dump the output of the operator.

    • kernels: This item can be configured in two formats:

      1. List of operator names. Specifying operator needs to first set the environment variable for saving the graph file to save the graph, and then obtain the operator name from the saved graph file. Please refer to the documentation on Ascend Developer Zone DUMP_GE_GRAPH , DUMP_GRAPH_LEVEL and DUMP_GRAPH_PATH for details about the environment variable for saving the graph file.

      2. Regular expressions of operator names. When the string conforms to the format of "name-regex(xxx)", it would be considered a regular expression. For example, "name-regex(Default/.+)" can match all operators with names starting with "Default/".

    • support_device: Supported devices, default setting is [0,1,2,3,4,5,6,7]. You can specify specific device ids to dump specific device data.

    • statistic_category: This attribute is used by users to configure the category of statistical information to be saved, and only takes effect when saving statistical information is enabled(i.e.saved_data is set to statistic or full). The type is a string list, where the optional values of the strings are as follows:

      • "max": represents the maximum value of the elements in tensor;

      • "min": represents the minimum value of the elements in tensor;

      • "avg": represents the average value of elements in tensor;

      • "count": represents the number of the elements in tensor;

      • "negative zero count": represents the number of the elements which is less then zero in tensor;

      • "positive zero count": represents the number of the elements which is greater then zero in tensor;

      • "nan count": represents the number of Nan elements in the tensor;

      • "negative inf count": represents the number of -Inf elements in the tensor;

      • "positive inf count": represents the number of +Inf elements in the tensor;

      • "zero count": represents the number of zero elements in the tensor;

      • "md5": represents the MD5 value of the tensor;

      • "l2norm": represents L2Norm value of the tensor.

      This field is optional, with default values of ["max", "min", "l2norm"].

    • file_format: Dump file type. It can be either npy and bin. npy: data will be dumped in npy files as host format. bin: data will be dumped in protobuf file as device format and need to be transformed to parse using the provided data analysis tool. Please refer to Asynchronous Dump Data Analysis Sample for details. The default value is bin.

  2. Set Dump environment variable.

    export MINDSPORE_DUMP_CONFIG=${Absolute path of data_dump.json}

    If the path field is not set or set to an empty string in the Dump configuration file, you also need to configure the environment variable MS_DIAGNOSTIC_DATA_PATH.

    export MS_DIAGNOSTIC_DATA_PATH=${yyy}

    Then "$MS_DIAGNOSTIC_DATA_PATH/debug_dump" is regarded as path. If the path field in configuration file is not empty, it is still used as the path to save Dump data.

    • Set the environment variables before executing the training script. Setting environment variables during training will not take effect.

    • Dump environment variables need to be configured before calling mindspore.communication.init.

  3. Execute the training script to dump data.

    You can set set_context(reserve_class_name_in_scope=False) in your training script to avoid dump failure because of file name is too long.

  4. Refer to Asynchronous Dump Data Analysis Sample to analyze the Dump data file.

  • If you need to dump all or part of the operator, you can modify the dump_mode option in the json configuration file to 0 or 1.

  • Due to the slow Dump speed, enabling Dump in large model scenarios can extend the communication interval between different cards, leading to communication operator timeouts. This issue can be resolved by adjusting the timeout duration for the communication operators. For the Ascend backend, you can set the HCCL_EXEC_TIMEOUT environment variable. For detailed instructions, please refer to the Ascend CANN documentation.

Introduction to Data Object Directory and Data File

When the graph compilation level is not O0 or O1, the Dump directory structure is as follows, where the main feature is the {step_id} directory, which represents user side training step id:

{path}/
    - {step_id}/
        - {time}/
            - {device_id}/
                - {model_id}/
                    - {iteration_id}/
                        statistic.csv
                        {op_type}.{op_name}.{task_id}.{stream_id}.{timestamp}
                        Opdebug.Node_OpDebug.{task_id}.{stream_id}.{timestamp}
                        mapping.csv

When the graph compilation level is O0 or O1, the Dump directory structure is as follows. In this scenario, the dump files for aclop and aclnn operators will be saved in {device_id} directory, and the dump files for communication operators such as "ResuceSum" will be saved in {iteration_id} directory:

{path}/
    - {step_id}/
        - {time}/
            - {device_id}/
                - {model_name}/
                    - {model_id}/
                        - {iteration_id}/
                            statistic.csv
                            {op_type}.{op_name}.{task_id}.{stream_id}.{timestamp} //aclop ops
                            {op_name}.{op_type}.{task_id}.{stream_id}.{timestamp} //aclnn ops
                            mapping.csv
                statistic.csv
                {op_type}.{op_name}.{task_id}.{stream_id}.{timestamp} //aclop ops
                {op_name}.{op_type}.{task_id}.{stream_id}.{timestamp} //aclnn ops
                mapping.csv

  • path: the absolute path set in the data_dump.json configuration file.

  • device_id: the id of the device.

  • model_name: the model name generated by MindSpore.

  • model_id: the id of the model.

  • graph_id: the id of the training graph.

  • iteration_id: the iteration of the training.

  • op_type: the type of the operator.

  • op_name: the name of the operator.

  • task_id: the id of the task, if unable to fetch the value, the default is set to 65535.

  • stream_id: the id of the stream, if unable to fetch the value, the default is set to 65535.

  • timestamp: the time stamp.

  • step_id: user side training step id.

The overflow file (file Opdebug.Node_OpDebug.{task_id}.{stream_id}.{timestamp}) is only saved when overflow dump is enabled and overflow is detected.

If set file_format to npy, the operator file will be saved as a npy format file, and the overflow file will be saved as a json format file. The file naming formats are:

{op_type}.{op_name}.{task_id}.{stream_id}.{timestamp}.{input_output_index}.{slot}.{format}.npy
Opdebug.Node_OpDebug.{task_id}.{stream_id}.{timestamp}.output.0.json

If the length of the tensor file name defined according to the naming rules exceeds the OS file name length limit (usually 255 characters), the tensor file will be renamed to a string of random numbers. The mapping relationship will be written to the file 'mapping.csv' in the same directory.

If set file_format to npy, it can be loaded by numpy.load.

If not configured file_format or set file_format to bin, after the training is started, the original data file generated by asynchronous Dump or overflow files generated by overflow detection are in protobuf format. They need to be parsed using the data analysis tool that comes with the HiSilicon Run package. For details, please refer to How to view dump data files.

The data format on the Device side may be different from the definition in the calculation diagram on the Host side. The bin file data format of the asynchronous dump is the Device side format. If you want to convert to the Host side format, you can refer to How to convert dump data file format.

If the file is saved in bin format, the file naming format is:

{op_type}.{op_name}.{task_id}.{stream_id}.{timestamp}

Take the Conv2D-op12 of AlexNet network as an example: Conv2D.Default_network-WithLossCell__backbone-AlexNet_conv3-Conv2d_Conv2D-op12.2.7.161243956333802, where Conv2D is {op_type}, Default_network-WithLossCell__backbone-AlexNet_conv3-Conv2d_Conv2D-op12 is {op_name}, and 2 is {task_id' }, 7 is {stream_id' }, 161243956333802 is {timestamp}.

If ".", "/", "", and spaces appear in op_type and op_name, they will be converted to underscores.

The original data file generated by dump can also be parsed by using the data parsing tool DumpParser of MindSpore Insight. Please refer to DumpParser Introduction for the usage of DumpParser. The data format parsed by MindSpore Insight is exactly the same as that of synchronous dump.

If setting file_format to npy, the naming convention of data files generated by asynchronous dump is the same as those of synchronous dump. Please refer to Introduction to Synchronous Dump Data File. The overflow file generated by overflow detection is in the json format, and the content analysis of the overflow file can refer to the Analyzing the Data File of an Overflow/Underflow Operator .

The saved_data option only takes effect when file_format is "npy". If saved_data is "statistic" or "full", tensor statistics will be dumped in statistic.csv. When saved_data is "tensor" or "full", full tensor data will be dumped in {op_type}.{op_name}.{task_id}.{stream_id}.{timestamp}.{input_output_index}.{slot}.{format}.npy. The format of the statistic file will be the same as that of synchonous dump. Please refer to Introduction to Synchronous Dump Data File.

The constant dump file, final execution graph file and execution order file naming rules generated by asynchronous Dump are the same as that of synchronous Dump. You can refer to Introduction to Synchronous Dump Data File.

Data Analysis Sample

Asynchronous dump does not automatically save .ir files. To view .ir files, you can use MindSpore IR save switch set_comtext(save_graphs=2) before executing the use case. After executing the use case, you can view the saved tracecode_graph_ xxx} file, which can be opened with vi. Please refer to the data analysis example of synchronous dump for the file viewing method. When the graph compilation level is O0 or O1, the operator files saved by asynchronous dump are different from the operator names in the graph file. Therefore, asynchronous dump is not recommended for this scenario, and synchronous dump is recommended. When the compilation level of the graph is O2, since the .ir file is not the final execution graph, it cannot be guaranteed that the operator names in the operator file correspond one-to-one with those in the .ir file. Please refer to the documentation on Ascend Developer Zone DUMP_GE_GRAPH , DUMP_GRAPH_LEVEL and DUMP_GRAPH_PATH to save the final execution graph.

Through the asynchronous Dump function, the data files generated by the operator asynchronous Dump can be obtained. If file_format in the Dump configure file is set to "npy", then the step 1, 2 in the follows steps can be skipped. If file_format is not set or set to "bin", the tensor files need to be converted to .npy format.

  1. Parse the dumped file using msaccucmp.py provied in the run package, the path where the msaccucmp.py file is located may be different on different environments. You can find it through the find command:

    find ${run_path} -name "msaccucmp.py"

    • run_path: The installation path of the run package.

  2. After finding the msaccucmp.py, go to the /absolute_path directory and run the following command to parse the Dump data:

    python ${The absolute path of msaccucmp.py} convert -d {file path of dump} -out {file path of output}

    The {file path of dump} can be path to a single .bin file, or the folder that include the .bin files.

    If you need to convert the data format, please refer to the user instructions link https://www.hiascend.com/document/detail/en/CANNCommunityEdition/600alphaX/developmenttools/devtool/atlasaccuracy_16_0077.html.

    For example, the data file generated by Dump is:

    Conv2D.Default_network-WithLossCell__backbone-AlexNet_conv3-Conv2d_Conv2D-op12.2.7.161243956333802

    Then execute:

    python3.7.5 msaccucmp.py convert -d /path/to/Conv2D.Default_network-WithLossCell__backbone-AlexNet_conv3-Conv2d_Conv2D-op12.2.7.161243956333802 -out ./output -f NCHW -t npy

    All input and output data for this operator can be generated under ./output. Each data is saved as a file with the .npy suffix in the format NCHW. The result is as follows:

    Conv2D.Default_network-WithLossCell__backbone-AlexNet_conv3-Conv2d_Conv2D-op12.2.7.161243956333802.input.0.32x256x13x13.npy
Conv2D.Default_network-WithLossCell__backbone-AlexNet_conv3-Conv2d_Conv2D-op12.2.7.161243956333802.input.1.384x256x3x3.npy
Conv2D.Default_network-WithLossCell__backbone-AlexNet_conv3-Conv2d_Conv2D-op12.2.7.161243956333802.output.0.32x384x13x13.npy

    At the end of the file name, you can see which input or output the file is the operator, and the dimensional information of the data. For example, by the first .npy file name

    Conv2D.Default_network-WithLossCell__backbone-AlexNet_conv3-Conv2d_Conv2D-op12.2.7.161243956333802.input.0.32x256x13x13.npy

    It can be seen that the file is the 0th input of the operator, and the dimension information of the data is 32x256x13x13.

  3. The corresponding data can be read through numpy.load("file_name"). For example:

    import numpy
numpy.load("Conv2D.Default_network-WithLossCell__backbone-AlexNet_conv3-Conv2d_Conv2D-op12.2.7.161243956333802.input.0.32x256x13x13.npy")

Other Description

Other Dump Function

In some special scenarios, the GE dump mode can be applied under development guidance.

To enable GE dump, set the environment variable MINDSPORE_DUMP_CONFIG and ENABLE_MS_GE_DUMP to 1. This mode applies only to the scenario where the compilation level of the graph is O2. The format of the configuration file is the same as that of the acl dump configuration file. The op_debug_mode field cannot be set to 4. Other parameters are the same as those of the acl dump configuration file.

export ENABLE_MS_GE_DUMP=1

When GE dump is enabled, and the graph compilation level is O2, the Dump directory structure of the graph pattern is as follows:

{path}/
    - {time}/
        - {device_id}/
            - {model_name}/
                - {model_id}/
                    - {iteration_id}/
                        statistic.csv
                        {op_type}.{op_name}.{task_id}.{stream_id}.{timestamp}
                        Opdebug.Node_OpDebug.{task_id}.{stream_id}.{timestamp}
                        mapping.csv

Among them, the meanings of path, time, device_id, model_name, model_id, iteration_id, op_type, op_name, task_id, stream_id, and timestamp are the same as those of ACL dump.

This method will be abandoned in the future and is not recommended for use.

Notices

  • When an operator of type bfloat16 is saved to the npy file, it will be converted to type float32.

  • Dump only supports saving data with type of bool, int, int8, in16, int32, int64, uint, uint8, uint16, uint32, uint64, float, float16, float32, float64, bfloat16, double, complex64 and complex128.

  • Complex64 and complex128 only support saving as npy files, not as statistics information.

  • The Print operator has an input parameter with type of string, which is not a data type supported by Dump. Therefore, when the Print operator is included in the script, there will be an error log, which will not affect the saving data of other types.

  • When ACL dump is enabled, lite exception dump is not supported by using set_context(ascend_config={"exception_dump": "2"), while full exception dump is supported by using set_context(ascend_config={"exception_dump": "1").