mindspore 2.0.0rc1__cp38-cp38-manylinux1_x86_64.whl → 2.2.0__cp38-cp38-manylinux1_x86_64.whl
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
Potentially problematic release.
This version of mindspore might be problematic. Click here for more details.
- mindspore/.commit_id +1 -1
- mindspore/Third_Party_Open_Source_Software_Notice +2 -2
- mindspore/__init__.py +5 -2
- mindspore/_akg/akg/build_module.py +5 -6
- mindspore/_akg/akg/composite/build_module.py +49 -16
- mindspore/_akg/akg/composite/split_stitch.py +10 -11
- mindspore/_akg/akg/config/repository.json +195 -0
- mindspore/_akg/akg/global_configs.py +5 -1
- mindspore/_akg/akg/ms/info_version_adapt.py +67 -1
- mindspore/_akg/akg/tvm/api.py +4 -3
- mindspore/_akg/akg/tvm/autotvm/__init__.py +1 -2
- mindspore/_akg/akg/tvm/autotvm/graph_tuner/base_graph_tuner.py +1 -5
- mindspore/_akg/akg/tvm/autotvm/measure/__init__.py +1 -1
- mindspore/_akg/akg/tvm/autotvm/measure/measure.py +1 -10
- mindspore/_akg/akg/tvm/autotvm/measure/measure_methods.py +1 -372
- mindspore/_akg/akg/tvm/build_module.py +16 -1
- mindspore/_akg/akg/tvm/contrib/graph_runtime.py +0 -53
- mindspore/_akg/akg/tvm/hybrid/parser.py +7 -6
- mindspore/_akg/akg/tvm/ir_builder.py +1 -1
- mindspore/_akg/akg/tvm/module.py +1 -2
- mindspore/_akg/akg/tvm/stmt.py +2 -2
- mindspore/_akg/akg/utils/composite_op_helper.py +9 -10
- mindspore/_akg/akg/utils/kernel_exec.py +58 -260
- mindspore/_akg/akg/utils/op_dsl.py +17 -1
- mindspore/_akg/akg/utils/result_analysis.py +4 -24
- mindspore/_akg/akg/utils/tbe_codegen_utils.py +198 -0
- mindspore/_c_dataengine.cpython-38-x86_64-linux-gnu.so +0 -0
- mindspore/_c_expression.cpython-38-x86_64-linux-gnu.so +0 -0
- mindspore/_c_mindrecord.cpython-38-x86_64-linux-gnu.so +0 -0
- mindspore/_check_jit_forbidden_api.py +5 -1
- mindspore/_checkparam.py +79 -62
- mindspore/_extends/graph_kernel/__init__.py +0 -1
- mindspore/_extends/graph_kernel/model/graph_split.py +2 -0
- mindspore/_extends/graph_kernel/model/model_builder.py +9 -50
- mindspore/_extends/graph_kernel/splitter.py +1 -9
- mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +128 -21
- mindspore/_extends/parallel_compile/akg_compiler/build_tbe_kernel.py +2 -2
- mindspore/_extends/parallel_compile/akg_compiler/tbe_topi.py +4 -2
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py +18 -13
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_helper.py +13 -9
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_job.py +1 -1
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_job_manager.py +1 -1
- mindspore/_extends/parse/__init__.py +19 -17
- mindspore/_extends/parse/namespace.py +7 -36
- mindspore/_extends/parse/parser.py +375 -189
- mindspore/_extends/parse/resources.py +36 -41
- mindspore/_extends/parse/standard_method.py +350 -245
- mindspore/_extends/parse/trope.py +2 -12
- mindspore/_extends/remote/kernel_build_server.py +24 -7
- mindspore/_extends/remote/kernel_build_server_akg_v2.py +55 -0
- mindspore/_install_custom.py +43 -0
- mindspore/_mindspore_offline_debug.cpython-38-x86_64-linux-gnu.so +0 -0
- mindspore/amp.py +85 -19
- mindspore/bin/cache_admin +0 -0
- mindspore/bin/cache_server +0 -0
- mindspore/boost/base.py +2 -2
- mindspore/boost/boost.py +27 -32
- mindspore/boost/boost_cell_wrapper.py +37 -13
- mindspore/boost/grad_accumulation.py +1 -1
- mindspore/boost/grad_freeze.py +34 -6
- mindspore/boost/group_loss_scale_manager.py +15 -14
- mindspore/boost/less_batch_normalization.py +28 -3
- mindspore/common/__init__.py +15 -11
- mindspore/common/_auto_dynamic.py +68 -0
- mindspore/common/_jit_fallback_utils.py +111 -0
- mindspore/common/_register_for_adapter.py +17 -5
- mindspore/common/_register_for_tensor.py +2 -2
- mindspore/common/_stub_tensor.py +18 -15
- mindspore/common/_utils.py +31 -7
- mindspore/common/api.py +269 -101
- mindspore/common/auto_dynamic_shape.py +498 -0
- mindspore/common/dtype.py +61 -21
- mindspore/common/dump.py +9 -7
- mindspore/common/initializer.py +106 -76
- mindspore/common/jit_config.py +35 -14
- mindspore/common/lazy_inline.py +187 -0
- mindspore/common/mindir_util.py +101 -0
- mindspore/common/mutable.py +10 -13
- mindspore/common/parameter.py +246 -55
- mindspore/common/seed.py +13 -7
- mindspore/common/sparse_tensor.py +29 -33
- mindspore/common/tensor.py +907 -251
- mindspore/communication/__init__.py +7 -4
- mindspore/communication/_comm_helper.py +84 -4
- mindspore/communication/management.py +160 -88
- mindspore/config/op_info.config +99 -75
- mindspore/config/super_bar_config.json +36 -4
- mindspore/context.py +526 -219
- mindspore/dataset/__init__.py +9 -46
- mindspore/dataset/audio/__init__.py +4 -19
- mindspore/dataset/audio/transforms.py +545 -233
- mindspore/dataset/audio/utils.py +21 -18
- mindspore/dataset/callback/ds_callback.py +42 -13
- mindspore/dataset/core/config.py +158 -100
- mindspore/dataset/core/validator_helpers.py +1 -63
- mindspore/dataset/debug/debug_hook.py +45 -13
- mindspore/dataset/debug/pre_defined_hook.py +5 -5
- mindspore/dataset/engine/__init__.py +0 -5
- mindspore/dataset/engine/cache_client.py +38 -15
- mindspore/dataset/engine/datasets.py +615 -278
- mindspore/dataset/engine/datasets_audio.py +154 -283
- mindspore/dataset/engine/datasets_standard_format.py +104 -116
- mindspore/dataset/engine/datasets_text.py +443 -326
- mindspore/dataset/engine/datasets_user_defined.py +251 -164
- mindspore/dataset/engine/datasets_vision.py +839 -1443
- mindspore/dataset/engine/iterators.py +11 -4
- mindspore/dataset/engine/obs/obs_mindrecord_dataset.py +7 -3
- mindspore/dataset/engine/obs/util.py +3 -0
- mindspore/dataset/engine/offload.py +6 -6
- mindspore/dataset/engine/queue.py +15 -14
- mindspore/dataset/engine/samplers.py +39 -23
- mindspore/dataset/engine/serializer_deserializer.py +22 -6
- mindspore/dataset/engine/validators.py +21 -331
- mindspore/dataset/text/__init__.py +5 -33
- mindspore/dataset/text/transforms.py +334 -165
- mindspore/dataset/text/utils.py +215 -145
- mindspore/dataset/transforms/__init__.py +1 -1
- mindspore/dataset/transforms/c_transforms.py +3 -2
- mindspore/dataset/transforms/py_transforms_util.py +40 -12
- mindspore/dataset/transforms/transforms.py +174 -71
- mindspore/dataset/utils/browse_dataset.py +25 -17
- mindspore/dataset/utils/line_reader.py +24 -21
- mindspore/dataset/vision/__init__.py +5 -26
- mindspore/dataset/vision/c_transforms.py +177 -165
- mindspore/dataset/vision/py_transforms.py +114 -119
- mindspore/dataset/vision/py_transforms_util.py +54 -51
- mindspore/dataset/vision/transforms.py +1127 -381
- mindspore/dataset/vision/utils.py +54 -38
- mindspore/dataset/vision/validators.py +12 -2
- mindspore/experimental/map_parameter.py +38 -4
- mindspore/{dataset/datapreprocess → experimental/optim}/__init__.py +14 -4
- mindspore/experimental/optim/adam.py +192 -0
- mindspore/experimental/optim/adamw.py +181 -0
- mindspore/experimental/optim/lr_scheduler.py +1427 -0
- mindspore/experimental/optim/optimizer.py +252 -0
- mindspore/experimental/optim/sgd.py +147 -0
- mindspore/gen_ops.py +273 -0
- mindspore/include/OWNERS +1 -2
- mindspore/include/api/context.h +21 -1
- mindspore/include/api/data_type.h +2 -1
- mindspore/include/api/graph.h +0 -15
- mindspore/include/api/kernel.h +2 -0
- mindspore/include/api/kernel_api.h +37 -12
- mindspore/include/api/model.h +29 -42
- mindspore/include/api/model_group.h +14 -3
- mindspore/include/api/model_parallel_runner.h +18 -2
- mindspore/include/api/serialization.h +26 -0
- mindspore/include/api/status.h +1 -0
- mindspore/include/api/types.h +38 -4
- mindspore/include/c_api/ms/abstract.h +67 -0
- mindspore/include/c_api/ms/attribute.h +197 -0
- mindspore/include/c_api/ms/base/handle_types.h +43 -0
- mindspore/include/c_api/ms/base/macros.h +32 -0
- mindspore/include/c_api/ms/base/status.h +33 -0
- mindspore/include/c_api/ms/base/types.h +282 -0
- mindspore/include/c_api/ms/context.h +102 -0
- mindspore/include/c_api/ms/graph.h +160 -0
- mindspore/include/c_api/ms/node.h +606 -0
- mindspore/include/c_api/ms/tensor.h +161 -0
- mindspore/include/c_api/ms/value.h +84 -0
- mindspore/include/c_api/status_c.h +3 -0
- mindspore/include/dataset/constants.h +6 -12
- mindspore/include/dataset/execute.h +23 -13
- mindspore/include/dataset/text.h +26 -26
- mindspore/include/dataset/transforms.h +25 -31
- mindspore/include/dataset/vision.h +60 -60
- mindspore/include/dataset/vision_ascend.h +5 -6
- mindspore/include/dataset/vision_lite.h +17 -17
- mindspore/include/mindapi/base/format.h +0 -1
- mindspore/include/mindapi/base/type_id.h +2 -1
- mindspore/include/mindapi/base/types.h +5 -1
- mindspore/lib/libdnnl.so.2 +0 -0
- mindspore/lib/libjemalloc.so.2 +0 -0
- mindspore/lib/libmindspore.so +0 -0
- mindspore/lib/libmindspore_backend.so +0 -0
- mindspore/lib/libmindspore_common.so +0 -0
- mindspore/lib/libmindspore_core.so +0 -0
- mindspore/lib/libmindspore_glog.so.0 +0 -0
- mindspore/lib/libmindspore_gpr.so.15 +0 -0
- mindspore/lib/libmindspore_grpc++.so.1 +0 -0
- mindspore/lib/libmindspore_grpc.so.15 +0 -0
- mindspore/lib/libmindspore_shared_lib.so +0 -0
- mindspore/lib/libmpi_adapter.so +0 -0
- mindspore/lib/libnnacl.so +0 -0
- mindspore/lib/libopencv_core.so.4.5 +0 -0
- mindspore/lib/libopencv_imgcodecs.so.4.5 +0 -0
- mindspore/lib/libopencv_imgproc.so.4.5 +0 -0
- mindspore/lib/libps_cache.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_aicpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/aicpu_kernel/impl/libcust_cpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_impl/cpu/config/cust_aicpu_kernel.json +9000 -0
- mindspore/lib/plugin/ascend/custom_aicpu_ops/op_proto/libcust_op_proto.so +0 -0
- mindspore/lib/plugin/ascend/libakg.so +0 -0
- mindspore/lib/plugin/ascend/libascend_collective.so +0 -0
- mindspore/lib/plugin/ascend/libdvpp_utils.so +0 -0
- mindspore/lib/plugin/ascend/libhccl_plugin.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_aicpu_kernels.so +0 -0
- mindspore/lib/plugin/ascend/libmindspore_cpu_kernels.so +0 -0
- mindspore/lib/plugin/cpu/libakg.so +0 -0
- mindspore/lib/plugin/gpu/libcuda_ops.so.10 +0 -0
- mindspore/lib/plugin/gpu/libcuda_ops.so.11 +0 -0
- mindspore/lib/plugin/gpu10.1/libakg.so +0 -0
- mindspore/lib/plugin/gpu10.1/libnccl.so.2 +0 -0
- mindspore/lib/plugin/gpu10.1/libnvidia_collective.so +0 -0
- mindspore/lib/plugin/gpu11.1/libakg.so +0 -0
- mindspore/lib/plugin/gpu11.1/libnccl.so.2 +0 -0
- mindspore/lib/plugin/gpu11.1/libnvidia_collective.so +0 -0
- mindspore/lib/plugin/gpu11.6/libakg.so +0 -0
- mindspore/lib/plugin/gpu11.6/libnccl.so.2 +0 -0
- mindspore/lib/plugin/gpu11.6/libnvidia_collective.so +0 -0
- mindspore/lib/plugin/libmindspore_ascend.so.1 +0 -0
- mindspore/lib/plugin/libmindspore_ascend.so.2 +0 -0
- mindspore/lib/plugin/libmindspore_gpu.so.10.1 +0 -0
- mindspore/lib/plugin/libmindspore_gpu.so.11.1 +0 -0
- mindspore/lib/plugin/libmindspore_gpu.so.11.6 +0 -0
- mindspore/log.py +9 -6
- mindspore/mindrecord/filereader.py +33 -4
- mindspore/mindrecord/filewriter.py +70 -35
- mindspore/mindrecord/mindpage.py +40 -34
- mindspore/mindrecord/shardreader.py +1 -1
- mindspore/mindrecord/shardsegment.py +1 -1
- mindspore/mindrecord/tools/cifar100_to_mr.py +25 -18
- mindspore/mindrecord/tools/cifar10_to_mr.py +25 -18
- mindspore/mindrecord/tools/csv_to_mr.py +29 -13
- mindspore/mindrecord/tools/imagenet_to_mr.py +24 -10
- mindspore/mindrecord/tools/mnist_to_mr.py +24 -11
- mindspore/mindrecord/tools/tfrecord_to_mr.py +31 -26
- mindspore/nn/cell.py +463 -169
- mindspore/nn/dynamic_lr.py +47 -43
- mindspore/nn/layer/activation.py +225 -82
- mindspore/nn/layer/basic.py +121 -79
- mindspore/nn/layer/channel_shuffle.py +21 -21
- mindspore/nn/layer/combined.py +33 -26
- mindspore/nn/layer/container.py +277 -22
- mindspore/nn/layer/conv.py +441 -304
- mindspore/nn/layer/dense.py +19 -13
- mindspore/nn/layer/embedding.py +62 -49
- mindspore/nn/layer/flash_attention.py +264 -0
- mindspore/nn/layer/image.py +50 -39
- mindspore/nn/layer/math.py +62 -51
- mindspore/nn/layer/normalization.py +219 -167
- mindspore/nn/layer/padding.py +58 -70
- mindspore/nn/layer/pooling.py +334 -287
- mindspore/nn/layer/rnn_cells.py +53 -38
- mindspore/nn/layer/rnns.py +59 -56
- mindspore/nn/layer/thor_layer.py +52 -44
- mindspore/nn/layer/timedistributed.py +6 -4
- mindspore/nn/layer/transformer.py +284 -164
- mindspore/nn/learning_rate_schedule.py +34 -25
- mindspore/nn/loss/__init__.py +3 -2
- mindspore/nn/loss/loss.py +554 -311
- mindspore/nn/optim/ada_grad.py +12 -9
- mindspore/nn/optim/adadelta.py +14 -11
- mindspore/nn/optim/adafactor.py +19 -16
- mindspore/nn/optim/adam.py +62 -47
- mindspore/nn/optim/adamax.py +13 -10
- mindspore/nn/optim/adasum.py +12 -8
- mindspore/nn/optim/asgd.py +10 -9
- mindspore/nn/optim/ftrl.py +20 -17
- mindspore/nn/optim/lamb.py +16 -12
- mindspore/nn/optim/lars.py +8 -6
- mindspore/nn/optim/lazyadam.py +25 -20
- mindspore/nn/optim/momentum.py +10 -7
- mindspore/nn/optim/optimizer.py +61 -9
- mindspore/nn/optim/proximal_ada_grad.py +14 -13
- mindspore/nn/optim/rmsprop.py +17 -13
- mindspore/nn/optim/rprop.py +30 -17
- mindspore/nn/optim/sgd.py +40 -23
- mindspore/nn/optim/thor.py +24 -26
- mindspore/nn/probability/bijector/bijector.py +11 -11
- mindspore/nn/probability/bijector/exp.py +1 -1
- mindspore/nn/probability/bijector/gumbel_cdf.py +3 -3
- mindspore/nn/probability/bijector/invert.py +1 -1
- mindspore/nn/probability/bijector/power_transform.py +29 -29
- mindspore/nn/probability/bijector/scalar_affine.py +3 -3
- mindspore/nn/probability/bijector/softplus.py +5 -5
- mindspore/nn/probability/bnn_layers/bnn_cell_wrapper.py +4 -2
- mindspore/nn/probability/bnn_layers/conv_variational.py +13 -13
- mindspore/nn/probability/bnn_layers/dense_variational.py +12 -12
- mindspore/nn/probability/bnn_layers/layer_distribution.py +9 -8
- mindspore/nn/probability/distribution/_utils/custom_ops.py +19 -3
- mindspore/nn/probability/distribution/_utils/utils.py +1 -1
- mindspore/nn/probability/distribution/bernoulli.py +9 -9
- mindspore/nn/probability/distribution/beta.py +8 -8
- mindspore/nn/probability/distribution/categorical.py +23 -15
- mindspore/nn/probability/distribution/cauchy.py +5 -6
- mindspore/nn/probability/distribution/distribution.py +3 -3
- mindspore/nn/probability/distribution/exponential.py +4 -4
- mindspore/nn/probability/distribution/gamma.py +10 -10
- mindspore/nn/probability/distribution/geometric.py +8 -8
- mindspore/nn/probability/distribution/gumbel.py +8 -9
- mindspore/nn/probability/distribution/half_normal.py +5 -5
- mindspore/nn/probability/distribution/laplace.py +5 -5
- mindspore/nn/probability/distribution/log_normal.py +12 -11
- mindspore/nn/probability/distribution/logistic.py +8 -8
- mindspore/nn/probability/distribution/normal.py +6 -5
- mindspore/nn/probability/distribution/poisson.py +10 -11
- mindspore/nn/probability/distribution/student_t.py +8 -9
- mindspore/nn/probability/distribution/transformed_distribution.py +5 -5
- mindspore/nn/probability/distribution/uniform.py +11 -11
- mindspore/nn/reinforcement/tensor_array.py +2 -2
- mindspore/nn/sparse/sparse.py +9 -9
- mindspore/nn/wrap/cell_wrapper.py +188 -63
- mindspore/nn/wrap/grad_reducer.py +21 -12
- mindspore/nn/wrap/loss_scale.py +136 -49
- mindspore/numpy/__init__.py +4 -4
- mindspore/numpy/array_creations.py +55 -56
- mindspore/numpy/array_ops.py +134 -35
- mindspore/numpy/logic_ops.py +66 -20
- mindspore/numpy/math_ops.py +142 -139
- mindspore/numpy/utils_const.py +2 -2
- mindspore/offline_debug/convert_async.py +2 -2
- mindspore/ops/_grad_experimental/__init__.py +7 -5
- mindspore/ops/_grad_experimental/grad_array_ops.py +231 -348
- mindspore/ops/{_grad → _grad_experimental}/grad_base.py +1 -33
- mindspore/ops/{_grad → _grad_experimental}/grad_comm_ops.py +25 -13
- mindspore/ops/{_grad/__init__.py → _grad_experimental/grad_debug_ops.py} +15 -7
- mindspore/ops/{_grad → _grad_experimental}/grad_implementations.py +17 -11
- mindspore/ops/_grad_experimental/grad_inner_ops.py +33 -52
- mindspore/ops/_grad_experimental/grad_math_ops.py +151 -1224
- mindspore/ops/_grad_experimental/grad_nn_ops.py +141 -414
- mindspore/ops/{_grad → _grad_experimental}/grad_quant_ops.py +10 -6
- mindspore/ops/_grad_experimental/grad_sparse.py +317 -2
- mindspore/ops/_grad_experimental/grad_sparse_ops.py +3 -13
- mindspore/ops/{_grad → _grad_experimental}/taylor_rule.py +1 -1
- mindspore/ops/_op_impl/_custom_op/dsd_back_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/flash_attention/__init__.py +0 -0
- mindspore/ops/_op_impl/_custom_op/flash_attention/attention.py +406 -0
- mindspore/{_extends/graph_kernel/expanders/complex/__init__.py → ops/_op_impl/_custom_op/flash_attention/constants.py} +27 -8
- mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_bwd.py +467 -0
- mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_fwd.py +563 -0
- mindspore/ops/_op_impl/_custom_op/flash_attention/flash_attention_impl.py +193 -0
- mindspore/ops/_op_impl/_custom_op/flash_attention/tik_ops_utils.py +435 -0
- mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/__init__.py +0 -0
- mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/sparse_tiling.py +45 -0
- mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/strategy.py +67 -0
- mindspore/ops/_op_impl/_custom_op/flash_attention/tiling_strategy/wukong_tiling.py +62 -0
- mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_left_impl.py +2 -2
- mindspore/ops/_op_impl/aicpu/__init__.py +41 -1
- mindspore/ops/_op_impl/aicpu/adaptive_max_pool_2d.py +37 -0
- mindspore/ops/_op_impl/aicpu/bias_add_grad.py +0 -1
- mindspore/ops/_op_impl/aicpu/cast.py +52 -0
- mindspore/ops/_op_impl/aicpu/coalesce.py +2 -0
- mindspore/ops/_op_impl/aicpu/col2im.py +3 -1
- mindspore/ops/_op_impl/aicpu/count_nonzero.py +43 -0
- mindspore/ops/_op_impl/aicpu/dropout_genmask.py +6 -0
- mindspore/ops/_op_impl/aicpu/eps.py +32 -0
- mindspore/ops/_op_impl/aicpu/eye.py +4 -4
- mindspore/ops/_op_impl/aicpu/fft_with_size.py +6 -0
- mindspore/ops/_op_impl/aicpu/fill_diagonal.py +5 -0
- mindspore/ops/_op_impl/aicpu/gamma.py +2 -2
- mindspore/ops/_op_impl/aicpu/im2col.py +3 -5
- mindspore/ops/_op_impl/aicpu/lgamma.py +1 -0
- mindspore/ops/_op_impl/aicpu/log_uniform_candidate_sampler.py +6 -3
- mindspore/ops/_op_impl/aicpu/lu.py +39 -0
- mindspore/ops/_op_impl/aicpu/lu_unpack_grad.py +0 -1
- mindspore/ops/_op_impl/aicpu/masked_scatter.py +1 -0
- mindspore/ops/_op_impl/aicpu/masked_select_grad.py +3 -0
- mindspore/ops/_op_impl/aicpu/matrix_band_part.py +59 -0
- mindspore/ops/_op_impl/aicpu/matrix_power.py +6 -1
- mindspore/ops/_op_impl/aicpu/median.py +1 -0
- mindspore/ops/_op_impl/aicpu/multinomial.py +9 -9
- mindspore/ops/_op_impl/aicpu/not_equal.py +0 -5
- mindspore/ops/_op_impl/aicpu/pad_v3.py +3 -1
- mindspore/ops/_op_impl/aicpu/pad_v3_grad.py +2 -0
- mindspore/ops/_op_impl/aicpu/parameterized_truncated_normal.py +15 -7
- mindspore/ops/_op_impl/aicpu/random_categorical.py +39 -19
- mindspore/ops/_op_impl/aicpu/random_choice_with_mask.py +5 -2
- mindspore/ops/_op_impl/aicpu/random_poisson.py +103 -52
- mindspore/ops/_op_impl/aicpu/random_shuffle.py +17 -15
- mindspore/ops/_op_impl/aicpu/resize_bilinear_grad.py +0 -1
- mindspore/ops/_op_impl/aicpu/resize_nearest_neighbor_v2.py +0 -6
- mindspore/ops/_op_impl/aicpu/resize_nearest_neighbor_v2_grad.py +0 -7
- mindspore/ops/_op_impl/aicpu/scatter_nd.py +2 -0
- mindspore/ops/_op_impl/aicpu/sequence_concat.py +40 -0
- mindspore/ops/_op_impl/aicpu/sequence_stack.py +40 -0
- mindspore/ops/_op_impl/aicpu/{sparseaddmm.py → sparse_addmm.py} +2 -2
- mindspore/ops/_op_impl/aicpu/{sparsesparsemaximum.py → sparse_sparse_maximum.py} +4 -4
- mindspore/ops/_op_impl/aicpu/standard_laplace.py +5 -4
- mindspore/ops/_op_impl/aicpu/standard_normal.py +5 -4
- mindspore/ops/_op_impl/aicpu/truncated_normal.py +9 -7
- mindspore/ops/_op_impl/aicpu/uniform.py +5 -3
- mindspore/ops/_op_impl/aicpu/uniform_candidate_sampler.py +8 -4
- mindspore/ops/_op_impl/aicpu/uniform_int.py +5 -5
- mindspore/ops/_op_impl/aicpu/uniform_real.py +4 -4
- mindspore/ops/_op_impl/aicpu/upsample_nearest_3d.py +14 -6
- mindspore/ops/_op_impl/aicpu/upsample_nearest_3d_grad.py +22 -8
- mindspore/ops/_op_impl/aicpu/upsample_trilinear_3d.py +11 -6
- mindspore/ops/_op_impl/aicpu/upsample_trilinear_3d_grad.py +21 -10
- mindspore/ops/_op_impl/tbe/__init__.py +6 -4
- mindspore/ops/_op_impl/tbe/atomic_addr_clean.py +1 -1
- mindspore/ops/_op_impl/tbe/avg_pool.py +2 -2
- mindspore/ops/_op_impl/tbe/avg_pool_3d.py +3 -3
- mindspore/ops/_op_impl/tbe/avg_pool_3d_grad.py +4 -4
- mindspore/ops/_op_impl/tbe/avg_pool_ds.py +2 -2
- mindspore/ops/_op_impl/tbe/avg_pool_grad.py +3 -3
- mindspore/ops/_op_impl/tbe/avg_pool_grad_vm.py +3 -3
- mindspore/ops/_op_impl/tbe/batch_to_space.py +1 -1
- mindspore/ops/_op_impl/tbe/batch_to_space_nd.py +2 -2
- mindspore/ops/_op_impl/tbe/bn_infer.py +2 -2
- mindspore/ops/_op_impl/tbe/bn_infer_ds.py +3 -2
- mindspore/ops/_op_impl/tbe/broadcast_to.py +1 -1
- mindspore/ops/_op_impl/tbe/depthwise_conv2d.py +3 -3
- mindspore/ops/_op_impl/tbe/expand_dims.py +1 -1
- mindspore/ops/_op_impl/tbe/gather_v2.py +56 -0
- mindspore/ops/_op_impl/tbe/im2col.py +4 -4
- mindspore/ops/_op_impl/tbe/inplace_index_add.py +7 -3
- mindspore/ops/_op_impl/tbe/mem_set.py +38 -0
- mindspore/ops/_op_impl/tbe/scatter_nd_add.py +3 -0
- mindspore/ops/_op_impl/tbe/scatter_nd_d.py +1 -1
- mindspore/ops/_op_impl/tbe/space_to_batch.py +1 -1
- mindspore/ops/_op_impl/tbe/space_to_batch_nd.py +2 -2
- mindspore/ops/_op_impl/tbe/trans_data_ds.py +2 -0
- mindspore/ops/_primitive_cache.py +1 -1
- mindspore/ops/_tracefunc.py +241 -0
- mindspore/ops/_utils/utils.py +10 -2
- mindspore/ops/_vmap/vmap_array_ops.py +5 -3
- mindspore/ops/_vmap/vmap_base.py +5 -4
- mindspore/ops/_vmap/vmap_convolution_ops.py +1 -1
- mindspore/ops/_vmap/vmap_grad_math_ops.py +6 -4
- mindspore/ops/_vmap/vmap_grad_nn_ops.py +11 -6
- mindspore/ops/_vmap/vmap_math_ops.py +5 -2
- mindspore/ops/_vmap/vmap_nn_ops.py +135 -11
- mindspore/ops/arg_dtype_cast.py +54 -0
- mindspore/ops/composite/__init__.py +7 -5
- mindspore/ops/composite/base.py +78 -34
- mindspore/ops/composite/math_ops.py +5 -695
- mindspore/ops/composite/multitype_ops/_compile_utils.py +403 -97
- mindspore/ops/composite/multitype_ops/_constexpr_utils.py +28 -22
- mindspore/ops/composite/multitype_ops/add_impl.py +69 -7
- mindspore/ops/composite/multitype_ops/bitwise_and_impl.py +2 -1
- mindspore/ops/composite/multitype_ops/bitwise_or_impl.py +2 -1
- mindspore/ops/composite/multitype_ops/bitwise_xor_impl.py +2 -0
- mindspore/ops/composite/multitype_ops/div_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/floordiv_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/getitem_impl.py +48 -10
- mindspore/ops/composite/multitype_ops/greater_equal_impl.py +2 -0
- mindspore/ops/composite/multitype_ops/greater_impl.py +2 -0
- mindspore/ops/composite/multitype_ops/left_shift_impl.py +2 -0
- mindspore/ops/composite/multitype_ops/less_equal_impl.py +2 -0
- mindspore/ops/composite/multitype_ops/less_impl.py +2 -0
- mindspore/ops/composite/multitype_ops/logic_not_impl.py +2 -2
- mindspore/ops/composite/multitype_ops/mod_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/mul_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/negative_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/not_in_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/ones_like_impl.py +6 -0
- mindspore/ops/composite/multitype_ops/pow_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/right_shift_impl.py +2 -0
- mindspore/ops/composite/multitype_ops/setitem_impl.py +10 -7
- mindspore/ops/composite/multitype_ops/sub_impl.py +1 -0
- mindspore/ops/composite/multitype_ops/uadd_impl.py +2 -0
- mindspore/ops/composite/multitype_ops/zeros_like_impl.py +9 -0
- mindspore/ops/deprecated.py +304 -0
- mindspore/ops/function/__init__.py +41 -4
- mindspore/ops/function/array_func.py +1108 -467
- mindspore/ops/function/clip_func.py +94 -27
- mindspore/ops/function/debug_func.py +3 -1
- mindspore/ops/function/grad/grad_func.py +82 -73
- mindspore/ops/function/image_func.py +28 -12
- mindspore/ops/function/linalg_func.py +135 -39
- mindspore/ops/function/math_func.py +3779 -894
- mindspore/ops/function/nn_func.py +1584 -657
- mindspore/ops/function/parameter_func.py +13 -3
- mindspore/ops/function/random_func.py +247 -153
- mindspore/ops/function/sparse_func.py +14 -11
- mindspore/ops/function/sparse_unary_func.py +173 -47
- mindspore/ops/function/spectral_func.py +8 -4
- mindspore/ops/function/vmap_func.py +8 -7
- mindspore/ops/functional.py +47 -16
- mindspore/ops/op_info_register.py +346 -86
- mindspore/ops/operations/__init__.py +38 -22
- mindspore/ops/operations/_grad_ops.py +145 -149
- mindspore/ops/operations/_inner_ops.py +298 -56
- mindspore/ops/operations/_ms_kernel.py +3 -3
- mindspore/ops/operations/_quant_ops.py +24 -28
- mindspore/ops/operations/_rl_inner_ops.py +9 -7
- mindspore/ops/operations/_scalar_ops.py +115 -0
- mindspore/ops/operations/_sequence_ops.py +148 -10
- mindspore/ops/operations/_tensor_array.py +1 -1
- mindspore/ops/operations/_thor_ops.py +2 -2
- mindspore/ops/operations/array_ops.py +1239 -561
- mindspore/ops/operations/comm_ops.py +166 -90
- mindspore/ops/operations/control_ops.py +3 -3
- mindspore/ops/operations/custom_ops.py +124 -102
- mindspore/ops/operations/debug_ops.py +24 -11
- mindspore/ops/operations/image_ops.py +86 -71
- mindspore/ops/operations/inner_ops.py +18 -13
- mindspore/ops/operations/linalg_ops.py +30 -11
- mindspore/ops/operations/math_ops.py +1730 -435
- mindspore/ops/operations/nn_ops.py +1953 -943
- mindspore/ops/operations/other_ops.py +65 -43
- mindspore/ops/operations/random_ops.py +258 -98
- mindspore/ops/operations/rl_ops.py +4 -36
- mindspore/ops/operations/sparse_ops.py +38 -33
- mindspore/ops/operations/spectral_ops.py +8 -4
- mindspore/ops/primitive.py +66 -44
- mindspore/ops/signature.py +5 -5
- mindspore/parallel/_auto_parallel_context.py +80 -19
- mindspore/parallel/_cost_model_context.py +42 -0
- mindspore/parallel/_offload_context.py +162 -72
- mindspore/parallel/_parallel_serialization.py +2 -2
- mindspore/parallel/_ps_context.py +16 -4
- mindspore/parallel/_recovery_context.py +2 -1
- mindspore/parallel/_tensor.py +15 -13
- mindspore/parallel/_transformer/layers.py +8 -6
- mindspore/parallel/_transformer/loss.py +1 -0
- mindspore/parallel/_transformer/moe.py +7 -7
- mindspore/parallel/_transformer/op_parallel_config.py +12 -1
- mindspore/parallel/_transformer/transformer.py +34 -14
- mindspore/parallel/_utils.py +36 -14
- mindspore/parallel/algo_parameter_config.py +114 -20
- mindspore/parallel/checkpoint_transform.py +16 -18
- mindspore/parallel/shard.py +16 -13
- mindspore/profiler/__init__.py +1 -1
- mindspore/profiler/common/struct_type.py +3 -3
- mindspore/profiler/common/util.py +3 -2
- mindspore/profiler/envprofiling.py +11 -4
- mindspore/profiler/parser/aicpu_data_parser.py +5 -3
- mindspore/profiler/parser/ascend_flops_generator.py +94 -0
- mindspore/profiler/parser/ascend_fpbp_generator.py +76 -0
- mindspore/profiler/parser/ascend_hccl_generator.py +288 -0
- mindspore/profiler/parser/ascend_msprof_exporter.py +213 -0
- mindspore/profiler/parser/ascend_msprof_generator.py +199 -0
- mindspore/profiler/parser/ascend_op_generator.py +276 -0
- mindspore/profiler/parser/ascend_steptrace_generator.py +94 -0
- mindspore/profiler/parser/ascend_timeline_generator.py +110 -54
- mindspore/profiler/parser/base_timeline_generator.py +11 -7
- mindspore/profiler/parser/cpu_gpu_timeline_generator.py +45 -46
- mindspore/profiler/parser/flops_parser.py +15 -11
- mindspore/profiler/parser/framework_parser.py +92 -73
- mindspore/profiler/parser/hccl_parser.py +16 -12
- mindspore/profiler/parser/integrator.py +22 -11
- mindspore/profiler/parser/memory_usage_parser.py +36 -11
- mindspore/profiler/parser/minddata_analyzer.py +12 -14
- mindspore/profiler/parser/minddata_pipeline_parser.py +1 -1
- mindspore/profiler/parser/msadvisor_parser.py +8 -4
- mindspore/profiler/parser/op_intermediate_parser.py +5 -2
- mindspore/profiler/parser/optime_parser.py +1 -1
- mindspore/profiler/parser/profiler_info.py +4 -5
- mindspore/profiler/parser/step_trace_parser.py +11 -14
- mindspore/profiler/profiling.py +678 -377
- mindspore/rewrite/api/node.py +211 -54
- mindspore/rewrite/api/node_type.py +5 -0
- mindspore/rewrite/api/pattern_engine.py +22 -23
- mindspore/rewrite/api/scoped_value.py +20 -17
- mindspore/rewrite/api/symbol_tree.py +252 -106
- mindspore/rewrite/api/tree_node_helper.py +3 -0
- mindspore/rewrite/ast_helpers/__init__.py +2 -1
- mindspore/rewrite/ast_helpers/ast_finder.py +129 -0
- mindspore/rewrite/ast_helpers/ast_modifier.py +116 -104
- mindspore/rewrite/ast_transformers/flatten_recursive_stmt.py +97 -46
- mindspore/rewrite/common/rewrite_elog.py +5 -1
- mindspore/rewrite/namer.py +51 -51
- mindspore/rewrite/namespace.py +14 -5
- mindspore/{ops/bprop_mindir → rewrite/node}/__init__.py +9 -4
- mindspore/rewrite/node/call_function.py +79 -0
- mindspore/rewrite/node/cell_container.py +135 -0
- mindspore/rewrite/node/control_flow.py +88 -0
- mindspore/rewrite/{node.py → node/node.py} +313 -247
- mindspore/rewrite/node/node_manager.py +254 -0
- mindspore/rewrite/node/node_topological_manager.py +243 -0
- mindspore/rewrite/parsers/arguments_parser.py +22 -21
- mindspore/rewrite/parsers/assign_parser.py +225 -239
- mindspore/rewrite/parsers/attribute_parser.py +9 -7
- mindspore/rewrite/parsers/class_def_parser.py +179 -218
- mindspore/rewrite/parsers/constant_parser.py +9 -6
- mindspore/rewrite/parsers/container_parser.py +9 -7
- mindspore/rewrite/parsers/for_parser.py +36 -15
- mindspore/rewrite/parsers/function_def_parser.py +23 -20
- mindspore/rewrite/parsers/if_parser.py +28 -24
- mindspore/rewrite/parsers/module_parser.py +202 -25
- mindspore/rewrite/{parser.py → parsers/parser.py} +4 -2
- mindspore/rewrite/{parser_register.py → parsers/parser_register.py} +1 -1
- mindspore/rewrite/parsers/return_parser.py +6 -6
- mindspore/rewrite/sparsify/sparse_transformer.py +12 -3
- mindspore/rewrite/sparsify/sparsify.py +4 -1
- mindspore/rewrite/sparsify/utils.py +11 -5
- mindspore/rewrite/symbol_tree.py +577 -732
- mindspore/rewrite/symbol_tree_builder.py +9 -175
- mindspore/rewrite/symbol_tree_dumper.py +2 -2
- mindspore/run_check/_check_version.py +46 -39
- mindspore/run_check/run_check.py +3 -2
- mindspore/{scipy/sparse → safeguard}/__init__.py +4 -5
- mindspore/safeguard/rewrite_obfuscation.py +517 -0
- mindspore/scipy/__init__.py +1 -1
- mindspore/scipy/linalg.py +67 -61
- mindspore/scipy/ops.py +5 -41
- mindspore/scipy/ops_grad.py +3 -2
- mindspore/scipy/ops_wrapper.py +5 -5
- mindspore/scipy/optimize/line_search.py +8 -8
- mindspore/scipy/optimize/linear_sum_assignment.py +4 -4
- mindspore/scipy/optimize/minimize.py +16 -12
- mindspore/scipy/utils.py +1 -52
- mindspore/scipy/utils_const.py +4 -4
- mindspore/train/__init__.py +4 -4
- mindspore/train/_utils.py +13 -5
- mindspore/train/amp.py +410 -148
- mindspore/train/anf_ir_pb2.py +16 -4
- mindspore/train/callback/_backup_and_restore.py +8 -11
- mindspore/train/callback/_callback.py +80 -3
- mindspore/train/callback/_checkpoint.py +82 -51
- mindspore/train/callback/_early_stop.py +12 -15
- mindspore/train/callback/_history.py +1 -1
- mindspore/train/callback/_lambda_callback.py +13 -13
- mindspore/train/callback/_landscape.py +21 -17
- mindspore/train/callback/_loss_monitor.py +9 -10
- mindspore/train/callback/_on_request_exit.py +16 -33
- mindspore/train/callback/_reduce_lr_on_plateau.py +21 -24
- mindspore/train/callback/_summary_collector.py +44 -30
- mindspore/train/callback/_time_monitor.py +62 -12
- mindspore/train/data_sink.py +10 -16
- mindspore/train/dataset_helper.py +154 -86
- mindspore/train/loss_scale_manager.py +14 -9
- mindspore/train/metrics/__init__.py +10 -2
- mindspore/train/metrics/accuracy.py +1 -1
- mindspore/train/metrics/auc.py +1 -1
- mindspore/train/metrics/bleu_score.py +2 -2
- mindspore/train/metrics/confusion_matrix.py +14 -14
- mindspore/train/metrics/cosine_similarity.py +3 -3
- mindspore/train/metrics/dice.py +1 -1
- mindspore/train/metrics/fbeta.py +1 -1
- mindspore/train/metrics/hausdorff_distance.py +8 -6
- mindspore/train/metrics/mean_surface_distance.py +5 -4
- mindspore/train/metrics/metric.py +49 -17
- mindspore/train/metrics/occlusion_sensitivity.py +4 -4
- mindspore/train/metrics/perplexity.py +1 -1
- mindspore/train/metrics/precision.py +2 -2
- mindspore/train/metrics/recall.py +2 -3
- mindspore/train/metrics/roc.py +7 -7
- mindspore/train/metrics/root_mean_square_surface_distance.py +5 -4
- mindspore/train/metrics/topk.py +7 -4
- mindspore/train/mind_ir_pb2.py +193 -48
- mindspore/train/model.py +377 -133
- mindspore/train/serialization.py +697 -245
- mindspore/train/summary/_summary_adapter.py +5 -2
- mindspore/train/summary/_writer_pool.py +4 -3
- mindspore/train/summary/summary_record.py +25 -23
- mindspore/train/train_thor/convert_utils.py +39 -23
- mindspore/train/train_thor/dataset_helper.py +4 -3
- mindspore/train/train_thor/model_thor.py +8 -8
- mindspore/version.py +1 -1
- {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/METADATA +7 -8
- {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/RECORD +647 -818
- {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/entry_points.txt +0 -1
- mindspore/_akg/akg/tvm/contrib/debugger/__init__.py +0 -16
- mindspore/_akg/akg/tvm/contrib/debugger/debug_result.py +0 -274
- mindspore/_akg/akg/tvm/contrib/debugger/debug_runtime.py +0 -259
- mindspore/_akg/akg/tvm/contrib/peak.py +0 -341
- mindspore/_akg/akg/tvm/contrib/rpc.py +0 -25
- mindspore/_akg/akg/tvm/contrib/xcode.py +0 -257
- mindspore/_akg/akg/tvm/exec/__init__.py +0 -17
- mindspore/_akg/akg/tvm/exec/autotvm_log_editor.py +0 -60
- mindspore/_akg/akg/tvm/exec/measure_peak.py +0 -48
- mindspore/_akg/akg/tvm/exec/query_rpc_tracker.py +0 -48
- mindspore/_akg/akg/tvm/exec/rpc_proxy.py +0 -98
- mindspore/_akg/akg/tvm/exec/rpc_server.py +0 -88
- mindspore/_akg/akg/tvm/exec/rpc_tracker.py +0 -62
- mindspore/_akg/akg/tvm/rpc/__init__.py +0 -29
- mindspore/_akg/akg/tvm/rpc/base.py +0 -182
- mindspore/_akg/akg/tvm/rpc/client.py +0 -436
- mindspore/_akg/akg/tvm/rpc/proxy.py +0 -595
- mindspore/_akg/akg/tvm/rpc/server.py +0 -413
- mindspore/_akg/akg/tvm/rpc/tornado_util.py +0 -121
- mindspore/_akg/akg/tvm/rpc/tracker.py +0 -431
- mindspore/_extends/graph_kernel/expander.py +0 -80
- mindspore/_extends/graph_kernel/expanders/__init__.py +0 -57
- mindspore/_extends/graph_kernel/expanders/_utils.py +0 -269
- mindspore/_extends/graph_kernel/expanders/addn.py +0 -33
- mindspore/_extends/graph_kernel/expanders/batchnorm.py +0 -152
- mindspore/_extends/graph_kernel/expanders/batchnorm_grad.py +0 -105
- mindspore/_extends/graph_kernel/expanders/bias_add_grad.py +0 -49
- mindspore/_extends/graph_kernel/expanders/clip_by_norm_no_div_sum.py +0 -33
- mindspore/_extends/graph_kernel/expanders/complex/abs.py +0 -30
- mindspore/_extends/graph_kernel/expanders/complex/add.py +0 -44
- mindspore/_extends/graph_kernel/expanders/complex/div.py +0 -62
- mindspore/_extends/graph_kernel/expanders/complex/mul.py +0 -52
- mindspore/_extends/graph_kernel/expanders/complex/real_div.py +0 -62
- mindspore/_extends/graph_kernel/expanders/complex/sub.py +0 -45
- mindspore/_extends/graph_kernel/expanders/conv2d.py +0 -200
- mindspore/_extends/graph_kernel/expanders/dropout_grad.py +0 -30
- mindspore/_extends/graph_kernel/expanders/equal_count.py +0 -50
- mindspore/_extends/graph_kernel/expanders/erfc.py +0 -35
- mindspore/_extends/graph_kernel/expanders/expand_dims.py +0 -50
- mindspore/_extends/graph_kernel/expanders/fused_adam.py +0 -44
- mindspore/_extends/graph_kernel/expanders/fused_adam_weight_decay.py +0 -47
- mindspore/_extends/graph_kernel/expanders/fused_mul_add.py +0 -28
- mindspore/_extends/graph_kernel/expanders/gather.py +0 -43
- mindspore/_extends/graph_kernel/expanders/gelu_grad.py +0 -70
- mindspore/_extends/graph_kernel/expanders/gkdropout.py +0 -40
- mindspore/_extends/graph_kernel/expanders/identity.py +0 -25
- mindspore/_extends/graph_kernel/expanders/layernorm.py +0 -93
- mindspore/_extends/graph_kernel/expanders/layernorm_grad.py +0 -113
- mindspore/_extends/graph_kernel/expanders/logsoftmax.py +0 -46
- mindspore/_extends/graph_kernel/expanders/logsoftmax_grad.py +0 -36
- mindspore/_extends/graph_kernel/expanders/matmul.py +0 -80
- mindspore/_extends/graph_kernel/expanders/maximum_grad.py +0 -59
- mindspore/_extends/graph_kernel/expanders/minimum_grad.py +0 -80
- mindspore/_extends/graph_kernel/expanders/oneslike.py +0 -26
- mindspore/_extends/graph_kernel/expanders/reduce_mean.py +0 -43
- mindspore/_extends/graph_kernel/expanders/relu_grad.py +0 -32
- mindspore/_extends/graph_kernel/expanders/sigmoid_cross_entropy_with_logits.py +0 -41
- mindspore/_extends/graph_kernel/expanders/sigmoid_cross_entropy_with_logits_grad.py +0 -35
- mindspore/_extends/graph_kernel/expanders/sigmoid_grad.py +0 -31
- mindspore/_extends/graph_kernel/expanders/slice.py +0 -35
- mindspore/_extends/graph_kernel/expanders/softmax_cross_entropy_with_logits.py +0 -42
- mindspore/_extends/graph_kernel/expanders/softmax_grad_ext.py +0 -41
- mindspore/_extends/graph_kernel/expanders/softsign.py +0 -28
- mindspore/_extends/graph_kernel/expanders/sqrt_grad.py +0 -29
- mindspore/_extends/graph_kernel/expanders/square_sum_all.py +0 -44
- mindspore/_extends/graph_kernel/expanders/square_sum_v1.py +0 -37
- mindspore/_extends/graph_kernel/expanders/squared_difference.py +0 -43
- mindspore/_extends/graph_kernel/expanders/tanh_grad.py +0 -31
- mindspore/_extends/graph_kernel/expanders/tile.py +0 -54
- mindspore/_extends/graph_kernel/model/op_infer.py +0 -506
- mindspore/_extends/parse/jit_fallback_modules.py +0 -51
- mindspore/dataset/datapreprocess/preprocess_imagenet_validate_dataset.py +0 -54
- mindspore/dataset/engine/graphdata.py +0 -1586
- mindspore/include/api/net.h +0 -142
- mindspore/ops/_grad/grad_array_ops.py +0 -1347
- mindspore/ops/_grad/grad_clip_ops.py +0 -84
- mindspore/ops/_grad/grad_debug_ops.py +0 -68
- mindspore/ops/_grad/grad_inner_ops.py +0 -235
- mindspore/ops/_grad/grad_math_ops.py +0 -1684
- mindspore/ops/_grad/grad_nn_ops.py +0 -1529
- mindspore/ops/_grad/grad_other_ops.py +0 -89
- mindspore/ops/_grad/grad_sequence_ops.py +0 -296
- mindspore/ops/_grad/grad_sparse.py +0 -323
- mindspore/ops/_grad_experimental/grad_image_ops.py +0 -249
- mindspore/ops/_grad_experimental/grad_linalg_ops.py +0 -195
- mindspore/ops/_grad_experimental/grad_scalar_ops.py +0 -112
- mindspore/ops/bprop_mindir/AdaptiveAvgPool2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/AdaptiveMaxPool2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ApproximateEqual_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/Argmax_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/Argmin_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/AssignSub_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/Assign_bprop.mindir +0 -17
- mindspore/ops/bprop_mindir/AvgPool3D_bprop.mindir +0 -150
- mindspore/ops/bprop_mindir/AvgPool_bprop.mindir +0 -66
- mindspore/ops/bprop_mindir/BCEWithLogitsLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BNTrainingReduce_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/BatchNormGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BatchToSpaceND_bprop.mindir +0 -28
- mindspore/ops/bprop_mindir/BiasAddGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BinaryCrossEntropy_bprop.mindir +0 -33
- mindspore/ops/bprop_mindir/BroadcastTo_bprop.mindir +0 -306
- mindspore/ops/bprop_mindir/Broadcast_bprop.mindir +0 -13
- mindspore/ops/bprop_mindir/CTCLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Concat_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Conv2DBackpropFilter_bprop.mindir +0 -240
- mindspore/ops/bprop_mindir/Conv2DBackpropInput_bprop.mindir +0 -247
- mindspore/ops/bprop_mindir/Conv2DTranspose_bprop.mindir +0 -247
- mindspore/ops/bprop_mindir/Conv3DTranspose_bprop.mindir +0 -315
- mindspore/ops/bprop_mindir/Conv3D_bprop.mindir +0 -278
- mindspore/ops/bprop_mindir/DType_bprop.mindir +0 -14
- mindspore/ops/bprop_mindir/DeformableOffsets_bprop.mindir +0 -58
- mindspore/ops/bprop_mindir/Depend_bprop.mindir +0 -13
- mindspore/ops/bprop_mindir/DepthToSpace_bprop.mindir +0 -23
- mindspore/ops/bprop_mindir/DepthwiseConv2dNative_bprop.mindir +0 -138
- mindspore/ops/bprop_mindir/DiagPart_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/Dropout2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Dropout3D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DropoutDoMask_bprop.mindir +0 -25
- mindspore/ops/bprop_mindir/DropoutGenMask_bprop.mindir +0 -18
- mindspore/ops/bprop_mindir/DropoutGrad_bprop.mindir +0 -27
- mindspore/ops/bprop_mindir/Dropout_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DynamicGRUV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DynamicRNN_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/DynamicShape_bprop.mindir +0 -14
- mindspore/ops/bprop_mindir/Elu_bprop.mindir +0 -16
- mindspore/ops/bprop_mindir/EmbeddingLookup_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Equal_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/ExpandDims_bprop.mindir +0 -58
- mindspore/ops/bprop_mindir/FastGeLU_bprop.mindir +0 -16
- mindspore/ops/bprop_mindir/Flatten_bprop.mindir +0 -54
- mindspore/ops/bprop_mindir/FloorDiv_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/GatherD_bprop.mindir +0 -26
- mindspore/ops/bprop_mindir/GatherNd_bprop.mindir +0 -57
- mindspore/ops/bprop_mindir/Gather_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/GreaterEqual_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/Greater_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/HSigmoid_bprop.mindir +0 -16
- mindspore/ops/bprop_mindir/HSwish_bprop.mindir +0 -16
- mindspore/ops/bprop_mindir/IOU_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/InstanceNorm_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/IsFinite_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/IsInf_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/IsNan_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/KLDivLoss_bprop.mindir +0 -126
- mindspore/ops/bprop_mindir/L2Loss_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/L2Normalize_bprop.mindir +0 -30
- mindspore/ops/bprop_mindir/LRN_bprop.mindir +0 -43
- mindspore/ops/bprop_mindir/LayerNormGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/LessEqual_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/Less_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/LinSpace_bprop.mindir +0 -23
- mindspore/ops/bprop_mindir/Load_bprop.mindir +0 -13
- mindspore/ops/bprop_mindir/LogSoftmax_bprop.mindir +0 -23
- mindspore/ops/bprop_mindir/LogicalAnd_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/LogicalNot_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/MaskedSelect_bprop.mindir +0 -21
- mindspore/ops/bprop_mindir/MaxPool3DGradGrad_bprop.mindir +0 -74
- mindspore/ops/bprop_mindir/MaxPool3DGrad_bprop.mindir +0 -74
- mindspore/ops/bprop_mindir/MaxPool3D_bprop.mindir +0 -75
- mindspore/ops/bprop_mindir/MaxPoolGradGrad_bprop.mindir +0 -65
- mindspore/ops/bprop_mindir/MaxPoolWithArgmax_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Maximum_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Minimum_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/MirrorPad_bprop.mindir +0 -27
- mindspore/ops/bprop_mindir/Mish_bprop.mindir +0 -35
- mindspore/ops/bprop_mindir/MulNoNan_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/NLLLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/NonZero_bprop.mindir +0 -14
- mindspore/ops/bprop_mindir/NotEqual_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/OneHot_bprop.mindir +0 -26
- mindspore/ops/bprop_mindir/OnesLike_bprop.mindir +0 -14
- mindspore/ops/bprop_mindir/PReLU_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Pad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Padding_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/RNNTLoss_bprop.mindir +0 -29
- mindspore/ops/bprop_mindir/ROIAlign_bprop.mindir +0 -82
- mindspore/ops/bprop_mindir/Range_bprop.mindir +0 -22
- mindspore/ops/bprop_mindir/Rank_bprop.mindir +0 -14
- mindspore/ops/bprop_mindir/ReLU6_bprop.mindir +0 -16
- mindspore/ops/bprop_mindir/ReLUV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ReduceAll_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/ReduceAny_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/ReluGrad_bprop.mindir +0 -20
- mindspore/ops/bprop_mindir/Reshape_bprop.mindir +0 -60
- mindspore/ops/bprop_mindir/ResizeBilinear_bprop.mindir +0 -29
- mindspore/ops/bprop_mindir/ResizeNearestNeighbor_bprop.mindir +0 -89
- mindspore/ops/bprop_mindir/ReverseSequence_bprop.mindir +0 -52
- mindspore/ops/bprop_mindir/ReverseV2_bprop.mindir +0 -22
- mindspore/ops/bprop_mindir/Round_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/ScatterMax_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ScatterMin_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ScatterNdUpdate_bprop.mindir +0 -22
- mindspore/ops/bprop_mindir/ScatterNd_bprop.mindir +0 -24
- mindspore/ops/bprop_mindir/ScatterNonAliasingAdd_bprop.mindir +0 -22
- mindspore/ops/bprop_mindir/ScatterUpdate_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SeLU_bprop.mindir +0 -21
- mindspore/ops/bprop_mindir/Select_bprop.mindir +0 -31
- mindspore/ops/bprop_mindir/Shape_bprop.mindir +0 -14
- mindspore/ops/bprop_mindir/SigmoidCrossEntropyWithLogits_bprop.mindir +0 -21
- mindspore/ops/bprop_mindir/SigmoidGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Sigmoid_bprop.mindir +0 -16
- mindspore/ops/bprop_mindir/Sign_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/Slice_bprop.mindir +0 -26
- mindspore/ops/bprop_mindir/SmoothL1Loss_bprop.mindir +0 -36
- mindspore/ops/bprop_mindir/SoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Softplus_bprop.mindir +0 -16
- mindspore/ops/bprop_mindir/Softsign_bprop.mindir +0 -33
- mindspore/ops/bprop_mindir/Sort_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SpaceToBatchND_bprop.mindir +0 -28
- mindspore/ops/bprop_mindir/SpaceToDepth_bprop.mindir +0 -23
- mindspore/ops/bprop_mindir/SparseGatherV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/SparseSoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Split_bprop.mindir +0 -22
- mindspore/ops/bprop_mindir/Squeeze_bprop.mindir +0 -54
- mindspore/ops/bprop_mindir/StridedSliceGrad_bprop.mindir +0 -95
- mindspore/ops/bprop_mindir/StridedSlice_bprop.mindir +0 -98
- mindspore/ops/bprop_mindir/Switch_bprop.mindir +0 -29
- mindspore/ops/bprop_mindir/TanhGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Tanh_bprop.mindir +0 -66
- mindspore/ops/bprop_mindir/TensorScatterAdd_bprop.mindir +0 -22
- mindspore/ops/bprop_mindir/TensorScatterUpdate_bprop.mindir +0 -29
- mindspore/ops/bprop_mindir/TensorShape_bprop.mindir +0 -14
- mindspore/ops/bprop_mindir/Tile_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TopK_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TransShape_bprop.mindir +0 -23
- mindspore/ops/bprop_mindir/TruncateDiv_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir +0 -20
- mindspore/ops/bprop_mindir/Unique_bprop.mindir +0 -16
- mindspore/ops/bprop_mindir/Unstack_bprop.mindir +0 -22
- mindspore/ops/bprop_mindir/UpsampleNearest3D_bprop.mindir +0 -32
- mindspore/ops/bprop_mindir/UpsampleTrilinear3D_bprop.mindir +0 -38
- mindspore/ops/bprop_mindir/ZerosLike_bprop.mindir +0 -15
- mindspore/ops/bprop_mindir/generate_mindir.py +0 -114
- mindspore/rewrite/node_visitor.py +0 -44
- mindspore/rewrite/topological_manager.py +0 -203
- mindspore/scipy/sparse/linalg.py +0 -192
- {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/WHEEL +0 -0
- {mindspore-2.0.0rc1.dist-info → mindspore-2.2.0.dist-info}/top_level.txt +0 -0
|
@@ -0,0 +1,76 @@
|
|
|
1
|
+
# Copyright 2023 Huawei Technologies Co., Ltd
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
# ============================================================================
|
|
15
|
+
"""fp bp and is_training_mode_flag analyse model"""
|
|
16
|
+
import json
|
|
17
|
+
import logging
|
|
18
|
+
import os
|
|
19
|
+
import stat
|
|
20
|
+
from collections import defaultdict
|
|
21
|
+
|
|
22
|
+
import numpy as np
|
|
23
|
+
from mindspore.profiler.common.exceptions.exceptions import ProfilerIOException
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
class AscendFPBPGenerator:
|
|
27
|
+
"""Generate ascend fp bp data from DataFrame."""
|
|
28
|
+
|
|
29
|
+
def __init__(self, op_summary, steptrace):
|
|
30
|
+
self.op_summary = op_summary
|
|
31
|
+
self.steptrace = steptrace
|
|
32
|
+
self.points = None
|
|
33
|
+
|
|
34
|
+
def parse(self):
|
|
35
|
+
"""Analyse the op_summary and steptrace data generate fpbp data."""
|
|
36
|
+
is_training_mode_flag = False
|
|
37
|
+
points = None
|
|
38
|
+
|
|
39
|
+
steptrace = self.steptrace[self.steptrace['Iteration ID'] == 1]
|
|
40
|
+
steptrace = steptrace[(steptrace['FP Start'] != 0) & (steptrace['BP End'] != 0)]
|
|
41
|
+
if steptrace.size:
|
|
42
|
+
is_training_mode_flag = True
|
|
43
|
+
op_summary = self.op_summary[np.argsort(self.op_summary['Task Start Time'])]
|
|
44
|
+
fp_index = np.searchsorted(op_summary['Task Start Time'], steptrace['FP Start'], side='right') - 1
|
|
45
|
+
bp_index = np.searchsorted(op_summary['Task Start Time'] + op_summary['Task Duration'], steptrace['BP End'],
|
|
46
|
+
side='left') - 1
|
|
47
|
+
points = defaultdict(dict)
|
|
48
|
+
for i in range(steptrace.size):
|
|
49
|
+
model_id = f"model_{steptrace[i]['Iteration ID']}"
|
|
50
|
+
points[model_id]['fp_start'] = op_summary[fp_index[i]]['Op Name']
|
|
51
|
+
points[model_id]['bp_end'] = op_summary[bp_index[i]]['Op Name']
|
|
52
|
+
self.points = defaultdict()
|
|
53
|
+
self.points['fp_start'] = op_summary[fp_index[0]]['Op Name']
|
|
54
|
+
self.points['bp_end'] = op_summary[bp_index[0]]['Op Name']
|
|
55
|
+
|
|
56
|
+
return points, is_training_mode_flag
|
|
57
|
+
|
|
58
|
+
def write(self, step_trace_point_info_path):
|
|
59
|
+
"""
|
|
60
|
+
Write the flops.csv and flops_summary.json
|
|
61
|
+
|
|
62
|
+
Args:
|
|
63
|
+
step_trace_point_info_path(str): step_trace_point_info.json path.
|
|
64
|
+
"""
|
|
65
|
+
if not self.points:
|
|
66
|
+
return
|
|
67
|
+
try:
|
|
68
|
+
with os.fdopen(os.open(step_trace_point_info_path,
|
|
69
|
+
os.O_WRONLY | os.O_CREAT | os.O_TRUNC, stat.S_IWUSR | stat.S_IRUSR),
|
|
70
|
+
'w') as json_file:
|
|
71
|
+
json.dump(self.points, json_file)
|
|
72
|
+
except (IOError, OSError) as err:
|
|
73
|
+
logging.critical('Errot occurred when write step trace point info file: %s', err)
|
|
74
|
+
raise ProfilerIOException() from err
|
|
75
|
+
if os.path.exists(step_trace_point_info_path):
|
|
76
|
+
os.chmod(step_trace_point_info_path, stat.S_IREAD | stat.S_IWRITE)
|
|
@@ -0,0 +1,288 @@
|
|
|
1
|
+
# Copyright 2023 Huawei Technologies Co., Ltd
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
# ============================================================================
|
|
15
|
+
"""hccl analyse model"""
|
|
16
|
+
import copy
|
|
17
|
+
import csv
|
|
18
|
+
import fnmatch
|
|
19
|
+
import json
|
|
20
|
+
import logging
|
|
21
|
+
import os
|
|
22
|
+
import stat
|
|
23
|
+
|
|
24
|
+
import numpy as np
|
|
25
|
+
from mindspore.profiler.common.exceptions.exceptions import ProfilerIOException
|
|
26
|
+
|
|
27
|
+
|
|
28
|
+
def find_files(directory, pattern):
|
|
29
|
+
"""Find files with feature 'pattern' from the directory"""
|
|
30
|
+
file_list = []
|
|
31
|
+
for root, _, files in os.walk(directory):
|
|
32
|
+
files.sort(key=lambda x: os.path.getctime(os.path.join(directory, x)))
|
|
33
|
+
for basename in files:
|
|
34
|
+
if fnmatch.fnmatch(basename, pattern):
|
|
35
|
+
filename = os.path.join(root, basename)
|
|
36
|
+
file_list.append(filename)
|
|
37
|
+
return file_list
|
|
38
|
+
|
|
39
|
+
|
|
40
|
+
def calculate_average(data):
|
|
41
|
+
"""Calculate hccl data average"""
|
|
42
|
+
result = data
|
|
43
|
+
if isinstance(data, dict):
|
|
44
|
+
result = {}
|
|
45
|
+
for key, value in data.items():
|
|
46
|
+
result[key] = calculate_average(value)
|
|
47
|
+
elif isinstance(data, list):
|
|
48
|
+
if all(isinstance(item, (dict, list)) for item in data):
|
|
49
|
+
if isinstance(data[0], dict):
|
|
50
|
+
result_dict = {}
|
|
51
|
+
keys = set()
|
|
52
|
+
for item in data:
|
|
53
|
+
keys.update(item.keys())
|
|
54
|
+
for key in keys:
|
|
55
|
+
values = []
|
|
56
|
+
for item in data:
|
|
57
|
+
values.append(item.get(key))
|
|
58
|
+
result_dict[key] = calculate_average(values)
|
|
59
|
+
result = result_dict
|
|
60
|
+
elif isinstance(data[0], list):
|
|
61
|
+
transposed_list = np.array(data).T.tolist()
|
|
62
|
+
result = [calculate_average(sub_list) for sub_list in transposed_list]
|
|
63
|
+
return result
|
|
64
|
+
|
|
65
|
+
|
|
66
|
+
def count_average(data):
|
|
67
|
+
"""Count average"""
|
|
68
|
+
result = data
|
|
69
|
+
if isinstance(data, list):
|
|
70
|
+
if all(isinstance(x, (int, float)) for x in data):
|
|
71
|
+
if data:
|
|
72
|
+
result = sum(data) / len(data)
|
|
73
|
+
else:
|
|
74
|
+
result = 0
|
|
75
|
+
else:
|
|
76
|
+
result = [count_average(x) for x in data]
|
|
77
|
+
elif isinstance(data, dict):
|
|
78
|
+
result = {key: count_average(value) for key, value in data.items()}
|
|
79
|
+
return result
|
|
80
|
+
|
|
81
|
+
|
|
82
|
+
class AscendHCCLGenerator:
|
|
83
|
+
"""Generate ascend hccl data from files."""
|
|
84
|
+
|
|
85
|
+
def __init__(self, source_path):
|
|
86
|
+
self.root_path = source_path
|
|
87
|
+
self.hccl_raw = []
|
|
88
|
+
self.hccl_data_df = np.dtype(
|
|
89
|
+
[('name', object), ('pid', int), ('tid', int), ('ts', float), ('te', float), ('dur', float), ('ph', object),
|
|
90
|
+
('task_type', object), ('link_info', object), ('transport_type', object), ('size', int), ('tag', object)])
|
|
91
|
+
|
|
92
|
+
@staticmethod
|
|
93
|
+
def _cost_analyse(iteration):
|
|
94
|
+
"""analyse communication cost and wait cost"""
|
|
95
|
+
communication_cost = np.sum(iteration[iteration['name'] != 'Notify_Wait']['dur'])
|
|
96
|
+
wait_cost = np.sum(iteration[iteration['name'] == 'Notify_Wait']['dur'])
|
|
97
|
+
return communication_cost, wait_cost
|
|
98
|
+
|
|
99
|
+
@staticmethod
|
|
100
|
+
def _rdma_analyse(groupby_transport):
|
|
101
|
+
"""rdma analyse"""
|
|
102
|
+
thread_groups = np.unique(groupby_transport['tid'])
|
|
103
|
+
thread_information = []
|
|
104
|
+
for thread_index in thread_groups:
|
|
105
|
+
groupby_thread = groupby_transport[groupby_transport['tid'] == thread_index]
|
|
106
|
+
rdma_communication_time = 0
|
|
107
|
+
rdma_communication_size = 0
|
|
108
|
+
rdma_communication_wait_time = 0
|
|
109
|
+
start_index = 0
|
|
110
|
+
end_index = groupby_thread.size - 2
|
|
111
|
+
while start_index < end_index:
|
|
112
|
+
first_task_type = groupby_thread[start_index]['task_type']
|
|
113
|
+
if first_task_type == 'RDMASend':
|
|
114
|
+
second_index = start_index + 1
|
|
115
|
+
third_index = start_index + 2
|
|
116
|
+
second_task_type = groupby_thread[second_index]['task_type']
|
|
117
|
+
third_task_type = groupby_thread[third_index]['task_type']
|
|
118
|
+
if second_task_type == 'RDMASend' and third_task_type == 'Notify Wait':
|
|
119
|
+
rdma_send_cost = groupby_thread[start_index]['dur']
|
|
120
|
+
notify_record_cost = groupby_thread[second_index]['dur']
|
|
121
|
+
notify_wait_cost = groupby_thread[third_index]['dur']
|
|
122
|
+
rdma_communication_time += rdma_send_cost + notify_record_cost + notify_wait_cost
|
|
123
|
+
rdma_communication_wait_time += notify_wait_cost
|
|
124
|
+
rdma_communication_size += groupby_thread[start_index]['size'] + groupby_thread[second_index][
|
|
125
|
+
'size']
|
|
126
|
+
start_index += 2
|
|
127
|
+
start_index += 1
|
|
128
|
+
rdma_communication_wait_time = rdma_communication_wait_time / 1e3
|
|
129
|
+
rdma_communication_size = rdma_communication_size / 1e3
|
|
130
|
+
rdma_communication_time = rdma_communication_time / 1e3
|
|
131
|
+
rdma_bandwidth = rdma_communication_size / (rdma_communication_time / 1e3) \
|
|
132
|
+
if rdma_communication_size else 0
|
|
133
|
+
thread_information.append(
|
|
134
|
+
[rdma_communication_time, rdma_communication_size, rdma_bandwidth, rdma_communication_wait_time])
|
|
135
|
+
if len(thread_information) > 1:
|
|
136
|
+
thread_information = np.sum(thread_information, axis=0).tolist()
|
|
137
|
+
|
|
138
|
+
return thread_information
|
|
139
|
+
|
|
140
|
+
def parse(self):
|
|
141
|
+
"""Analyse the original hccl data generator hccl data."""
|
|
142
|
+
file_list = find_files(self.root_path, "hccl_*.json")
|
|
143
|
+
|
|
144
|
+
for hccl_file in file_list:
|
|
145
|
+
_, relative_path = os.path.split(hccl_file)
|
|
146
|
+
iteration_id = int(relative_path.split('_')[3])
|
|
147
|
+
with open(hccl_file) as f:
|
|
148
|
+
_, hccl_detail_data = self._original_data_analyse(json.load(f))
|
|
149
|
+
raw = self._iteration_analyse(hccl_detail_data, iteration_id)
|
|
150
|
+
self.hccl_raw.append(raw)
|
|
151
|
+
self.hccl_raw = sorted(self.hccl_raw, key=lambda x: x[0])
|
|
152
|
+
self.hccl_raw.append(copy.deepcopy(self.hccl_raw[-1]))
|
|
153
|
+
self.hccl_raw[-1][0] = '-'
|
|
154
|
+
for _, value in self.hccl_raw[-1][4].items():
|
|
155
|
+
value[0] = '-'
|
|
156
|
+
|
|
157
|
+
def write(self, hccl_raw_path):
|
|
158
|
+
"""
|
|
159
|
+
Write the flops.csv and flops_summary.json
|
|
160
|
+
|
|
161
|
+
Args:
|
|
162
|
+
hccl_raw_path(str): hccl_raw.csv path.
|
|
163
|
+
"""
|
|
164
|
+
try:
|
|
165
|
+
with os.fdopen(os.open(hccl_raw_path,
|
|
166
|
+
os.O_WRONLY | os.O_CREAT | os.O_TRUNC, stat.S_IWUSR | stat.S_IRUSR), 'w',
|
|
167
|
+
newline='') as hccl_row:
|
|
168
|
+
writer = csv.writer(hccl_row)
|
|
169
|
+
writer.writerow(
|
|
170
|
+
['step_num', 'communication_cost', 'wait_cost', 'link_info', 'communication_operator_cost'])
|
|
171
|
+
for row in self.hccl_raw:
|
|
172
|
+
row[3] = json.dumps(row[3])
|
|
173
|
+
row[4] = json.dumps(row[4])
|
|
174
|
+
writer.writerows(self.hccl_raw)
|
|
175
|
+
except (IOError, OSError) as err:
|
|
176
|
+
logging.critical('Errot occurred when write aicore detail file: %s', err)
|
|
177
|
+
raise ProfilerIOException() from err
|
|
178
|
+
if os.path.exists(hccl_raw_path):
|
|
179
|
+
os.chmod(hccl_raw_path, stat.S_IREAD | stat.S_IWRITE)
|
|
180
|
+
|
|
181
|
+
def _original_data_analyse(self, original_data):
|
|
182
|
+
"""analyse original data"""
|
|
183
|
+
target_data = []
|
|
184
|
+
for row in original_data:
|
|
185
|
+
if row.get('ph') == 'X':
|
|
186
|
+
name = row.get('name')
|
|
187
|
+
pid = row.get('pid')
|
|
188
|
+
tid = row.get('tid')
|
|
189
|
+
ts = row.get('ts')
|
|
190
|
+
dur = row.get('dur')
|
|
191
|
+
te = ts + dur
|
|
192
|
+
ph = row.get('ph')
|
|
193
|
+
task_type = row.get('args', {}).get('task type', '')
|
|
194
|
+
src_rank = row.get('args', {}).get('src rank', 0)
|
|
195
|
+
dst_rank = row.get('args', {}).get('dst rank', 0)
|
|
196
|
+
if src_rank == int('0xffffffff', 16):
|
|
197
|
+
src_rank = dst_rank
|
|
198
|
+
if dst_rank == int('0xffffffff', 16):
|
|
199
|
+
dst_rank = src_rank
|
|
200
|
+
transport_type = row.get('args', {}).get('transport type', '')
|
|
201
|
+
if transport_type == 'LOCAL':
|
|
202
|
+
src_rank, dst_rank = dst_rank, src_rank
|
|
203
|
+
link_info = str(src_rank) + '-' + str(dst_rank)
|
|
204
|
+
size = row.get('args', {}).get('size(Byte)', 0)
|
|
205
|
+
size = size if isinstance(size, int) else int(size, 16)
|
|
206
|
+
target_data.append(
|
|
207
|
+
tuple([name, pid, tid, ts, te, dur, ph, task_type, link_info, transport_type, size, -1]))
|
|
208
|
+
hccl_data = np.array(target_data, dtype=self.hccl_data_df)
|
|
209
|
+
|
|
210
|
+
hccl_abstract_data = hccl_data[hccl_data['task_type'] == '']
|
|
211
|
+
hccl_detail_data = hccl_data[hccl_data['task_type'] != '']
|
|
212
|
+
|
|
213
|
+
hccl_abstract_data = hccl_abstract_data[np.argsort(hccl_abstract_data['ts'])]
|
|
214
|
+
hccl_detail_data = hccl_detail_data[np.argsort(hccl_detail_data['ts'])]
|
|
215
|
+
hccl_detail_data['ts'] = hccl_detail_data['ts']
|
|
216
|
+
tag = np.searchsorted(hccl_abstract_data['ts'], hccl_detail_data['ts'], side='right') - 1
|
|
217
|
+
hccl_detail_data['tag'] = [x[-1] for x in np.char.split(hccl_abstract_data[tag]['name'].astype(str), sep='/')]
|
|
218
|
+
return hccl_abstract_data, hccl_detail_data
|
|
219
|
+
|
|
220
|
+
def _iteration_analyse(self, hccl_detail_data, iteration):
|
|
221
|
+
"""analyse data by iteration """
|
|
222
|
+
communication_cost, wait_cost = self._cost_analyse(hccl_detail_data)
|
|
223
|
+
link_info = self._link_info_analyse(hccl_detail_data)
|
|
224
|
+
communication_operator_cost = self._communication_operator_cost_analyse(hccl_detail_data, iteration)
|
|
225
|
+
return [iteration, communication_cost, wait_cost, link_info, communication_operator_cost]
|
|
226
|
+
|
|
227
|
+
def _link_info_analyse(self, hccl_detail_data):
|
|
228
|
+
"""analyse link info data"""
|
|
229
|
+
groupby_iteration = hccl_detail_data[hccl_detail_data['task_type'] != 'Notify Record']
|
|
230
|
+
link_info_groups = np.unique(groupby_iteration['link_info'])
|
|
231
|
+
link_info_information = dict()
|
|
232
|
+
for link_info_index in link_info_groups:
|
|
233
|
+
groupby_link_info = groupby_iteration[groupby_iteration['link_info'] == link_info_index]
|
|
234
|
+
transport_groups = np.unique(groupby_iteration['transport_type'])
|
|
235
|
+
transport_information = dict()
|
|
236
|
+
for transport_index in transport_groups:
|
|
237
|
+
groupby_transport = groupby_link_info[groupby_link_info['transport_type'] == transport_index]
|
|
238
|
+
if transport_index == 'SDMA' and groupby_transport.size > 0:
|
|
239
|
+
groupby_sdma = \
|
|
240
|
+
groupby_transport[np.isin(groupby_transport['task_type'], ['Memcpy', 'Reduce Inline'])][
|
|
241
|
+
['dur', 'size']]
|
|
242
|
+
sdma_communication_time = np.sum(groupby_sdma['dur']) * 1e-3
|
|
243
|
+
sdma_communication_size = np.sum(groupby_sdma['size']) * 1e-3
|
|
244
|
+
sdma_bandwidth = sdma_communication_size / sdma_communication_time * 1e-3 \
|
|
245
|
+
if sdma_communication_time != 0 else 0
|
|
246
|
+
transport_information['SDMA'] = [sdma_communication_time, sdma_communication_size, sdma_bandwidth]
|
|
247
|
+
elif transport_index == 'RDMA' and groupby_transport.size > 0:
|
|
248
|
+
transport_information['RDMA'] = self._rdma_analyse(groupby_transport)
|
|
249
|
+
link_info_information[link_info_index] = transport_information
|
|
250
|
+
return link_info_information
|
|
251
|
+
|
|
252
|
+
def _communication_operator_cost_analyse(self, hccl_detail_data, iteration_index):
|
|
253
|
+
"""analyse communication operator cost"""
|
|
254
|
+
groupby_iteration = hccl_detail_data[hccl_detail_data['task_type'] != 'Notify Record']
|
|
255
|
+
tag_groups = np.unique(groupby_iteration['tag'])
|
|
256
|
+
tag_information = dict()
|
|
257
|
+
for tag_index in tag_groups:
|
|
258
|
+
groupby_tag = groupby_iteration[groupby_iteration['tag'] == tag_index]
|
|
259
|
+
link_groups = np.unique(groupby_iteration['link_info'])
|
|
260
|
+
link_info_information = dict()
|
|
261
|
+
for link_info_index in link_groups:
|
|
262
|
+
groupby_link_info = groupby_tag[groupby_tag['link_info'] == link_info_index]
|
|
263
|
+
transport_groups = np.unique(groupby_link_info['transport_type'])
|
|
264
|
+
transport_information = dict()
|
|
265
|
+
for transport_index in transport_groups:
|
|
266
|
+
groupby_transport = groupby_link_info[groupby_link_info['transport_type'] == transport_index]
|
|
267
|
+
if transport_index == 'SDMA':
|
|
268
|
+
groupby_sdma = \
|
|
269
|
+
groupby_transport[np.isin(groupby_transport['task_type'], ['Memcpy', 'Reduce Inline'])][
|
|
270
|
+
['dur', 'size']]
|
|
271
|
+
sdma_communication_time = np.sum(groupby_sdma['dur']) * 1e-3
|
|
272
|
+
sdma_communication_size = np.sum(groupby_sdma['size']) * 1e-3
|
|
273
|
+
sdma_bandwidth = sdma_communication_size / sdma_communication_time * 1e-3 \
|
|
274
|
+
if sdma_communication_time != 0 else 0
|
|
275
|
+
transport_information['SDMA'] = [
|
|
276
|
+
sdma_communication_time, sdma_communication_size,
|
|
277
|
+
sdma_bandwidth
|
|
278
|
+
]
|
|
279
|
+
elif transport_index == 'RDMA':
|
|
280
|
+
transport_information['RDMA'] = self._rdma_analyse(groupby_transport)
|
|
281
|
+
link_info_information[link_info_index] = transport_information
|
|
282
|
+
communication_cost = np.sum(groupby_tag[groupby_tag['name'] != 'Notify_Wait']['dur'])
|
|
283
|
+
wait_cost = np.sum(groupby_tag[groupby_tag['name'] == 'Notify_Wait']['dur'])
|
|
284
|
+
tag_information[tag_index] = [
|
|
285
|
+
str(iteration_index), communication_cost, wait_cost,
|
|
286
|
+
link_info_information
|
|
287
|
+
]
|
|
288
|
+
return tag_information
|
|
@@ -0,0 +1,213 @@
|
|
|
1
|
+
# Copyright 2023 Huawei Technologies Co., Ltd
|
|
2
|
+
#
|
|
3
|
+
# Licensed under the Apache License, Version 2.0 (the "License");
|
|
4
|
+
# you may not use this file except in compliance with the License.
|
|
5
|
+
# You may obtain a copy of the License at
|
|
6
|
+
#
|
|
7
|
+
# http://www.apache.org/licenses/LICENSE-2.0
|
|
8
|
+
#
|
|
9
|
+
# Unless required by applicable law or agreed to in writing, software
|
|
10
|
+
# distributed under the License is distributed on an "AS IS" BASIS,
|
|
11
|
+
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
12
|
+
# See the License for the specific language governing permissions and
|
|
13
|
+
# limitations under the License.
|
|
14
|
+
# ============================================================================
|
|
15
|
+
"""msprof PROF data export api file"""
|
|
16
|
+
import os
|
|
17
|
+
import shutil
|
|
18
|
+
from collections import defaultdict
|
|
19
|
+
from subprocess import CalledProcessError, TimeoutExpired
|
|
20
|
+
from subprocess import Popen, PIPE
|
|
21
|
+
import csv
|
|
22
|
+
from mindspore import log as logger
|
|
23
|
+
from mindspore.profiler.common.util import get_file_path
|
|
24
|
+
|
|
25
|
+
|
|
26
|
+
class AscendMsprofExporter:
|
|
27
|
+
"""
|
|
28
|
+
msprof exporter. export cann edge profiling data.
|
|
29
|
+
|
|
30
|
+
args:
|
|
31
|
+
source_path: the root path of PROF_* files
|
|
32
|
+
|
|
33
|
+
files under prof_root_dir is like:
|
|
34
|
+
profiler/PROF_*/device_{id}/data/xxx
|
|
35
|
+
profiler/PROF_*/host/data/xxx
|
|
36
|
+
...
|
|
37
|
+
|
|
38
|
+
usage:
|
|
39
|
+
>> ms_exporter = AscendMsprofExporter("path/to/profiler/data")
|
|
40
|
+
>> ms_exporter.export(start_time)
|
|
41
|
+
"""
|
|
42
|
+
|
|
43
|
+
_hiai_msprof_tail = "Ascend/latest/tools/profiler/bin"
|
|
44
|
+
_msprof_cmd = "msprof"
|
|
45
|
+
_ascend_mark = "Ascend"
|
|
46
|
+
_summary_dir = "summary"
|
|
47
|
+
_timeline_dir = "timeline"
|
|
48
|
+
_step_trace_mark = "step_trace"
|
|
49
|
+
_op_summary_mark = "op_summary"
|
|
50
|
+
_op_statistic_mark = "op_statistic"
|
|
51
|
+
|
|
52
|
+
def __init__(self, source_path, time_out=3000):
|
|
53
|
+
self._time_out = time_out
|
|
54
|
+
self.source_path = source_path
|
|
55
|
+
self.prof_root_dir = os.path.abspath(os.path.join(self.source_path, os.path.pardir))
|
|
56
|
+
|
|
57
|
+
self._check_msprof_env()
|
|
58
|
+
|
|
59
|
+
def export(self, model_iteration_dict=None):
|
|
60
|
+
"""start_time is the time to collect PROF data"""
|
|
61
|
+
|
|
62
|
+
if not model_iteration_dict:
|
|
63
|
+
model_iteration_dict = self._generate_step_trace(self.prof_root_dir, self.source_path)
|
|
64
|
+
|
|
65
|
+
if model_iteration_dict:
|
|
66
|
+
for model_id, value in model_iteration_dict.items():
|
|
67
|
+
for iteration_id in value:
|
|
68
|
+
msprof_export_cmd = self._msprof_command_generator(self.prof_root_dir, model_id, iteration_id)
|
|
69
|
+
self._run_cmd(msprof_export_cmd)
|
|
70
|
+
self._check_export_files(self.source_path, model_iteration_dict)
|
|
71
|
+
|
|
72
|
+
def _run_cmd(self, cmd, raise_error=True):
|
|
73
|
+
"""run shell command"""
|
|
74
|
+
try:
|
|
75
|
+
proc = Popen(cmd, stdout=PIPE, stderr=PIPE, text=True)
|
|
76
|
+
except (FileNotFoundError, PermissionError, CalledProcessError) as exc:
|
|
77
|
+
raise RuntimeError(exc)
|
|
78
|
+
try:
|
|
79
|
+
outs, errs = proc.communicate(timeout=self._time_out)
|
|
80
|
+
except TimeoutExpired:
|
|
81
|
+
proc.kill()
|
|
82
|
+
msg = "The possible cause is that too much data is collected " \
|
|
83
|
+
"and the export time is too long."
|
|
84
|
+
logger.error(msg)
|
|
85
|
+
raise TimeoutError(msg)
|
|
86
|
+
logger.info(outs)
|
|
87
|
+
if raise_error and errs != "":
|
|
88
|
+
raise RuntimeError(errs)
|
|
89
|
+
return outs
|
|
90
|
+
|
|
91
|
+
def _msprof_command_generator(self, output, model_id=None, iter_id=None):
|
|
92
|
+
"""msprof export helper"""
|
|
93
|
+
export_cmd = [self._msprof_cmd, "--export=on", "--output={}".format(output)]
|
|
94
|
+
if isinstance(model_id, int) and model_id >= 0:
|
|
95
|
+
export_cmd.append("--model-id={}".format(model_id))
|
|
96
|
+
if isinstance(iter_id, int) and iter_id >= 0:
|
|
97
|
+
export_cmd.append("--iteration-id={}".format(iter_id))
|
|
98
|
+
return export_cmd
|
|
99
|
+
|
|
100
|
+
def _check_msprof_env(self):
|
|
101
|
+
"""Check the existence of msprof binary tool"""
|
|
102
|
+
|
|
103
|
+
def _check_msprof(temp_path: str):
|
|
104
|
+
if not os.path.isdir(temp_path):
|
|
105
|
+
return False
|
|
106
|
+
sub_files = os.listdir(temp_path)
|
|
107
|
+
if self._msprof_cmd in sub_files:
|
|
108
|
+
return True
|
|
109
|
+
return False
|
|
110
|
+
|
|
111
|
+
msprof_cmd = ["which", self._msprof_cmd]
|
|
112
|
+
outs = self._run_cmd(msprof_cmd, raise_error=False)
|
|
113
|
+
if outs != "":
|
|
114
|
+
return
|
|
115
|
+
logger.warning("[Profiler]The msprof command was not found. Searching from environment variables...")
|
|
116
|
+
|
|
117
|
+
msprof_path = None
|
|
118
|
+
envs = os.environ
|
|
119
|
+
if envs.get("ASCEND_TOOLKIT_HOME"):
|
|
120
|
+
temp_path = os.path.join(envs.get("ASCEND_TOOLKIT_HOME"), "bin")
|
|
121
|
+
if _check_msprof(temp_path):
|
|
122
|
+
msprof_path = temp_path
|
|
123
|
+
|
|
124
|
+
if not msprof_path and envs.get("PATH"):
|
|
125
|
+
path_list = envs.get("PATH").split(":")
|
|
126
|
+
for path in path_list:
|
|
127
|
+
if self._ascend_mark in path:
|
|
128
|
+
prefix = path.split(self._ascend_mark)[0]
|
|
129
|
+
temp_path = os.path.join(prefix, self._hiai_msprof_tail)
|
|
130
|
+
if _check_msprof(temp_path):
|
|
131
|
+
msprof_path = temp_path
|
|
132
|
+
break
|
|
133
|
+
if msprof_path:
|
|
134
|
+
envs["PATH"] = msprof_path + ":" + envs.get("PATH", "")
|
|
135
|
+
else:
|
|
136
|
+
raise FileNotFoundError("The msprof command was not found!")
|
|
137
|
+
|
|
138
|
+
logger.info("The msprof command has been added to the path!")
|
|
139
|
+
|
|
140
|
+
def _generate_step_trace(self, prof_path, device_path):
|
|
141
|
+
""""generate model_id iteration_id dict"""
|
|
142
|
+
|
|
143
|
+
summary_path = os.path.join(device_path, self._summary_dir)
|
|
144
|
+
timeline_path = os.path.join(device_path, self._timeline_dir)
|
|
145
|
+
|
|
146
|
+
msprof_export_cmd = self._msprof_command_generator(prof_path)
|
|
147
|
+
self._run_cmd(msprof_export_cmd)
|
|
148
|
+
|
|
149
|
+
if not os.path.isdir(summary_path):
|
|
150
|
+
msg = "Path {} is not a existing directory. Make sure there is " \
|
|
151
|
+
"valid profiling data directory!".format(summary_path)
|
|
152
|
+
raise FileNotFoundError(msg)
|
|
153
|
+
|
|
154
|
+
step_trace_file = get_file_path(summary_path, self._step_trace_mark)
|
|
155
|
+
|
|
156
|
+
if not step_trace_file:
|
|
157
|
+
logger.info("Do not found step trace csv file in {} .".format(summary_path))
|
|
158
|
+
return None
|
|
159
|
+
|
|
160
|
+
step_trace = defaultdict(list)
|
|
161
|
+
with open(step_trace_file, newline='', mode='r') as csvfile:
|
|
162
|
+
reader = csv.reader(csvfile, delimiter=',', quotechar='"')
|
|
163
|
+
header = next(reader)
|
|
164
|
+
for index, value in enumerate(header):
|
|
165
|
+
if value == 'Model ID':
|
|
166
|
+
Model_ID = index
|
|
167
|
+
if value == 'Iteration ID':
|
|
168
|
+
Iteration_ID = index
|
|
169
|
+
for row in reader:
|
|
170
|
+
step_trace[int(row[Model_ID])].append(int(row[Iteration_ID]))
|
|
171
|
+
|
|
172
|
+
if os.path.isdir(summary_path):
|
|
173
|
+
shutil.rmtree(summary_path)
|
|
174
|
+
if os.path.isdir(timeline_path):
|
|
175
|
+
shutil.rmtree(timeline_path)
|
|
176
|
+
|
|
177
|
+
return step_trace
|
|
178
|
+
|
|
179
|
+
def _check_export_files(self, source_path, step_trace):
|
|
180
|
+
"""Check the existence of op_summary & op_statistic files."""
|
|
181
|
+
summary_path = os.path.join(source_path, self._summary_dir)
|
|
182
|
+
if not os.path.isdir(summary_path):
|
|
183
|
+
raise RuntimeError("Path {} is not a existing directory.".format(summary_path))
|
|
184
|
+
summary_file_list = os.listdir(summary_path)
|
|
185
|
+
op_summary = set()
|
|
186
|
+
op_statistic = set()
|
|
187
|
+
|
|
188
|
+
for summary_file in summary_file_list:
|
|
189
|
+
if summary_file.startswith(self._op_summary_mark):
|
|
190
|
+
op_summary.add(summary_file)
|
|
191
|
+
elif summary_file.startswith(self._op_statistic_mark):
|
|
192
|
+
op_statistic.add(summary_file)
|
|
193
|
+
|
|
194
|
+
if not op_summary:
|
|
195
|
+
raise RuntimeError("The op_summary file was not found, perhaps the original data was not collected.")
|
|
196
|
+
if not op_statistic:
|
|
197
|
+
raise RuntimeError("The op_statistics file was not found, perhaps the original data was not collected.")
|
|
198
|
+
|
|
199
|
+
device_id = source_path.split('_')[-1].replace("/", "")
|
|
200
|
+
|
|
201
|
+
for model_id, value in step_trace.items():
|
|
202
|
+
for iteration_id in value:
|
|
203
|
+
tag = f"_{device_id}_{model_id}_{iteration_id}.csv"
|
|
204
|
+
op_summary_file_name = self._op_summary_mark + tag
|
|
205
|
+
op_statistic_file = self._op_statistic_mark + tag
|
|
206
|
+
if op_summary_file_name not in op_summary:
|
|
207
|
+
logger.warning("[Profiler]The file {} was not found, " \
|
|
208
|
+
"perhaps the original data was not collected.".format(op_summary_file_name))
|
|
209
|
+
if op_statistic_file not in op_statistic:
|
|
210
|
+
logger.warning("[Profiler]The file {} was not found, " \
|
|
211
|
+
"perhaps the original data was not collected.".format(op_statistic_file))
|
|
212
|
+
|
|
213
|
+
logger.info("Finish checking files.")
|