mindspore 2.0.0a0__cp37-none-any.whl → 2.0.0rc1__cp37-none-any.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 +9064 -0
- mindspore/__init__.py +4 -2
- mindspore/_akg/akg/composite/build_module.py +11 -0
- mindspore/_akg/akg/config/repository_cuda.json +11 -0
- mindspore/_akg/akg/tvm/contrib/nvcc.py +4 -3
- mindspore/_c_dataengine.cpython-37m-aarch64-linux-gnu.so +0 -0
- mindspore/_c_expression.cpython-37m-aarch64-linux-gnu.so +0 -0
- mindspore/_c_mindrecord.cpython-37m-aarch64-linux-gnu.so +0 -0
- mindspore/_check_jit_forbidden_api.py +102 -0
- mindspore/_checkparam.py +1066 -1001
- mindspore/_extends/parallel_compile/akg_compiler/akg_process.py +4 -3
- mindspore/_extends/parallel_compile/akg_compiler/tbe_topi.py +50 -48
- mindspore/_extends/parallel_compile/akg_compiler/util.py +9 -4
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_adapter.py +4 -4
- mindspore/_extends/parallel_compile/tbe_compiler/tbe_helper.py +9 -4
- mindspore/_extends/parse/__init__.py +5 -3
- mindspore/_extends/parse/namespace.py +16 -1
- mindspore/_extends/parse/parser.py +107 -22
- mindspore/_extends/parse/resources.py +0 -7
- mindspore/_extends/parse/standard_method.py +885 -413
- mindspore/_mindspore_offline_debug.cpython-37m-aarch64-linux-gnu.so +0 -0
- mindspore/amp.py +52 -57
- mindspore/bin/cache_admin +0 -0
- mindspore/bin/cache_server +0 -0
- mindspore/boost/boost.py +2 -2
- mindspore/boost/boost_cell_wrapper.py +38 -20
- mindspore/boost/dim_reduce.py +3 -3
- mindspore/boost/group_loss_scale_manager.py +1 -1
- mindspore/common/__init__.py +4 -6
- mindspore/common/_decorator.py +2 -0
- mindspore/common/_register_for_adapter.py +55 -0
- mindspore/common/_stub_tensor.py +201 -0
- mindspore/common/_utils.py +41 -7
- mindspore/common/api.py +215 -141
- mindspore/common/dtype.py +8 -1
- mindspore/common/dump.py +2 -2
- mindspore/common/initializer.py +4 -2
- mindspore/common/jit_config.py +17 -13
- mindspore/common/mutable.py +33 -13
- mindspore/common/parameter.py +23 -21
- mindspore/common/seed.py +8 -24
- mindspore/common/sparse_tensor.py +62 -41
- mindspore/common/tensor.py +852 -1154
- mindspore/communication/__init__.py +2 -2
- mindspore/communication/_comm_helper.py +11 -4
- mindspore/communication/management.py +22 -21
- mindspore/config/op_info.config +501 -1008
- mindspore/config/super_bar_config.json +512 -0
- mindspore/context.py +201 -23
- mindspore/dataset/__init__.py +6 -6
- mindspore/dataset/audio/__init__.py +7 -7
- mindspore/dataset/audio/transforms.py +670 -30
- mindspore/dataset/audio/utils.py +47 -4
- mindspore/dataset/audio/validators.py +223 -1
- mindspore/dataset/callback/ds_callback.py +2 -2
- mindspore/dataset/core/config.py +210 -14
- mindspore/dataset/core/validator_helpers.py +2 -2
- mindspore/{parallel/nn/layers.py → dataset/debug/__init__.py} +7 -8
- mindspore/dataset/debug/debug_hook.py +65 -0
- mindspore/dataset/debug/pre_defined_hook.py +67 -0
- mindspore/dataset/engine/__init__.py +7 -3
- mindspore/dataset/engine/cache_client.py +1 -1
- mindspore/dataset/engine/datasets.py +322 -66
- mindspore/dataset/engine/datasets_audio.py +80 -76
- mindspore/dataset/engine/datasets_standard_format.py +51 -38
- mindspore/dataset/engine/datasets_text.py +232 -118
- mindspore/dataset/engine/datasets_user_defined.py +41 -17
- mindspore/dataset/engine/datasets_vision.py +746 -225
- mindspore/dataset/engine/graphdata.py +75 -10
- mindspore/dataset/engine/iterators.py +45 -5
- mindspore/dataset/engine/offload.py +48 -28
- mindspore/dataset/engine/validators.py +117 -8
- mindspore/dataset/text/__init__.py +6 -5
- mindspore/dataset/text/transforms.py +86 -3
- mindspore/dataset/text/utils.py +6 -4
- mindspore/dataset/text/validators.py +25 -0
- mindspore/dataset/transforms/__init__.py +3 -2
- mindspore/dataset/transforms/c_transforms.py +1 -1
- mindspore/dataset/transforms/transforms.py +2 -2
- mindspore/dataset/utils/__init__.py +2 -1
- mindspore/dataset/utils/line_reader.py +121 -0
- mindspore/dataset/vision/__init__.py +2 -3
- mindspore/dataset/vision/c_transforms.py +9 -9
- mindspore/dataset/vision/py_transforms.py +5 -5
- mindspore/dataset/vision/py_transforms_util.py +2 -0
- mindspore/dataset/vision/transforms.py +160 -161
- mindspore/dataset/vision/utils.py +3 -3
- mindspore/experimental/map_parameter.py +38 -26
- mindspore/include/OWNERS +0 -1
- mindspore/include/api/callback/callback.h +9 -13
- mindspore/include/api/callback/ckpt_saver.h +2 -2
- mindspore/include/api/callback/loss_monitor.h +2 -2
- mindspore/include/api/callback/lr_scheduler.h +5 -5
- mindspore/include/api/callback/time_monitor.h +2 -2
- mindspore/include/api/callback/train_accuracy.h +4 -6
- mindspore/include/api/cfg.h +19 -6
- mindspore/include/api/context.h +44 -9
- mindspore/include/api/delegate.h +1 -1
- mindspore/include/api/metrics/accuracy.h +2 -2
- mindspore/include/api/metrics/metrics.h +4 -3
- mindspore/include/api/model.h +9 -4
- mindspore/include/api/model_parallel_runner.h +2 -2
- mindspore/include/api/net.h +12 -11
- mindspore/include/api/serialization.h +19 -3
- mindspore/include/api/types.h +3 -3
- mindspore/include/dataset/constants.h +7 -0
- mindspore/include/dataset/text.h +59 -0
- mindspore/include/mindapi/base/type_id.h +1 -0
- mindspore/lib/libdnnl.so.2 +0 -0
- mindspore/lib/libicudata.so.69 +0 -0
- mindspore/lib/libicui18n.so.69 +0 -0
- mindspore/lib/libicuuc.so.69 +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/libmpi_collective.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/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/libmindspore_ascend.so.1 +0 -0
- mindspore/lib/plugin/{libmindspore_ascend.so → libmindspore_ascend.so.2} +0 -0
- mindspore/log.py +1 -1
- mindspore/mindrecord/filereader.py +18 -0
- mindspore/mindrecord/filewriter.py +197 -34
- mindspore/mindrecord/shardreader.py +9 -0
- mindspore/mindrecord/shardwriter.py +1 -1
- mindspore/mindrecord/tools/cifar100_to_mr.py +3 -3
- mindspore/mindrecord/tools/cifar10_to_mr.py +3 -3
- mindspore/mindrecord/tools/csv_to_mr.py +3 -3
- mindspore/mindrecord/tools/imagenet_to_mr.py +16 -11
- mindspore/mindrecord/tools/mnist_to_mr.py +2 -2
- mindspore/mindrecord/tools/tfrecord_to_mr.py +6 -6
- mindspore/nn/__init__.py +0 -4
- mindspore/nn/cell.py +204 -132
- mindspore/nn/dynamic_lr.py +1 -1
- mindspore/nn/grad/cell_grad.py +7 -6
- mindspore/nn/layer/__init__.py +5 -4
- mindspore/nn/layer/activation.py +40 -89
- mindspore/nn/layer/basic.py +255 -624
- mindspore/nn/layer/channel_shuffle.py +7 -6
- mindspore/nn/layer/combined.py +1 -1
- mindspore/nn/layer/container.py +41 -4
- mindspore/nn/layer/conv.py +64 -28
- mindspore/nn/layer/dense.py +9 -8
- mindspore/nn/layer/embedding.py +27 -25
- mindspore/nn/layer/image.py +53 -46
- mindspore/nn/layer/math.py +97 -105
- mindspore/nn/layer/normalization.py +117 -86
- mindspore/nn/layer/padding.py +185 -95
- mindspore/nn/layer/pooling.py +817 -414
- mindspore/nn/layer/rnn_cells.py +10 -15
- mindspore/nn/layer/rnns.py +37 -38
- mindspore/nn/layer/thor_layer.py +11 -12
- mindspore/nn/layer/timedistributed.py +5 -5
- mindspore/nn/layer/transformer.py +701 -0
- mindspore/nn/learning_rate_schedule.py +8 -8
- mindspore/nn/loss/__init__.py +5 -4
- mindspore/nn/loss/loss.py +334 -199
- mindspore/nn/optim/ada_grad.py +6 -6
- mindspore/nn/optim/adadelta.py +2 -3
- mindspore/nn/optim/adafactor.py +4 -5
- mindspore/nn/optim/adam.py +126 -62
- mindspore/nn/optim/adamax.py +3 -4
- mindspore/nn/optim/adasum.py +6 -6
- mindspore/nn/optim/asgd.py +2 -2
- mindspore/nn/optim/ftrl.py +67 -38
- mindspore/nn/optim/lamb.py +4 -5
- mindspore/nn/optim/lars.py +2 -2
- mindspore/nn/optim/lazyadam.py +43 -4
- mindspore/nn/optim/momentum.py +6 -5
- mindspore/nn/optim/optimizer.py +3 -1
- mindspore/nn/optim/proximal_ada_grad.py +2 -2
- mindspore/nn/optim/rmsprop.py +1 -1
- mindspore/nn/optim/rprop.py +8 -9
- mindspore/nn/optim/sgd.py +19 -13
- mindspore/nn/optim/thor.py +10 -15
- mindspore/nn/probability/__init__.py +0 -2
- mindspore/nn/probability/bijector/bijector.py +4 -4
- mindspore/nn/probability/bijector/invert.py +1 -1
- mindspore/nn/probability/bijector/softplus.py +2 -2
- mindspore/nn/probability/bnn_layers/dense_variational.py +1 -1
- mindspore/nn/probability/bnn_layers/layer_distribution.py +2 -2
- mindspore/nn/probability/distribution/_utils/utils.py +9 -15
- mindspore/nn/probability/distribution/bernoulli.py +3 -3
- mindspore/nn/probability/distribution/beta.py +1 -1
- mindspore/nn/probability/distribution/categorical.py +5 -7
- mindspore/nn/probability/distribution/cauchy.py +3 -3
- mindspore/nn/probability/distribution/distribution.py +2 -2
- mindspore/nn/probability/distribution/exponential.py +2 -2
- mindspore/nn/probability/distribution/gamma.py +3 -3
- mindspore/nn/probability/distribution/geometric.py +1 -1
- mindspore/nn/probability/distribution/gumbel.py +3 -3
- mindspore/nn/probability/distribution/half_normal.py +15 -11
- mindspore/nn/probability/distribution/laplace.py +16 -13
- mindspore/nn/probability/distribution/logistic.py +2 -2
- mindspore/nn/probability/distribution/normal.py +1 -1
- mindspore/nn/probability/distribution/poisson.py +1 -1
- mindspore/nn/probability/distribution/student_t.py +20 -15
- mindspore/nn/probability/distribution/transformed_distribution.py +4 -4
- mindspore/nn/probability/distribution/uniform.py +2 -2
- mindspore/nn/reinforcement/_tensors_queue.py +3 -3
- mindspore/nn/reinforcement/tensor_array.py +2 -2
- mindspore/nn/sparse/sparse.py +2 -2
- mindspore/nn/wrap/cell_wrapper.py +27 -10
- mindspore/nn/wrap/grad_reducer.py +2 -2
- mindspore/nn/wrap/loss_scale.py +40 -24
- mindspore/numpy/array_creations.py +33 -22
- mindspore/numpy/array_ops.py +35 -30
- mindspore/numpy/logic_ops.py +6 -27
- mindspore/numpy/math_ops.py +22 -19
- mindspore/numpy/utils.py +1 -1
- mindspore/numpy/utils_const.py +108 -58
- mindspore/ops/_constants.py +0 -6
- mindspore/ops/_grad/__init__.py +2 -1
- mindspore/ops/_grad/grad_array_ops.py +86 -117
- mindspore/ops/_grad/grad_base.py +23 -1
- mindspore/ops/_grad/grad_clip_ops.py +2 -3
- mindspore/ops/_grad/grad_comm_ops.py +34 -24
- mindspore/ops/_grad/grad_implementations.py +9 -45
- mindspore/ops/_grad/grad_inner_ops.py +47 -4
- mindspore/ops/_grad/grad_math_ops.py +142 -117
- mindspore/ops/_grad/grad_nn_ops.py +71 -165
- mindspore/ops/_grad/grad_sequence_ops.py +296 -0
- mindspore/ops/_grad/grad_sparse.py +7 -6
- mindspore/ops/_grad_experimental/__init__.py +1 -0
- mindspore/ops/_grad_experimental/grad_array_ops.py +150 -15
- mindspore/ops/_grad_experimental/grad_image_ops.py +16 -7
- mindspore/ops/_grad_experimental/grad_inner_ops.py +1 -22
- mindspore/ops/_grad_experimental/grad_linalg_ops.py +4 -11
- mindspore/ops/_grad_experimental/grad_math_ops.py +210 -89
- mindspore/ops/_grad_experimental/grad_nn_ops.py +26 -22
- mindspore/ops/_grad_experimental/grad_scalar_ops.py +112 -0
- mindspore/ops/_grad_experimental/grad_sparse_ops.py +49 -8
- mindspore/ops/_op_impl/_custom_op/batch_matmul_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold.py +2 -2
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold2.py +2 -2
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold2_grad_reduce.py +4 -4
- mindspore/ops/_op_impl/_custom_op/batchnorm_fold_grad.py +3 -3
- mindspore/ops/_op_impl/_custom_op/cholesky_trsm_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/correction_mul.py +2 -2
- mindspore/ops/_op_impl/_custom_op/correction_mul_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/dsd_back_impl.py +1 -5
- mindspore/ops/_op_impl/_custom_op/dsd_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perchannel_grad_reduce.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_learned_scale_quant_perlayer_grad_reduce.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perchannel_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fake_quant_perlayer_grad.py +2 -2
- mindspore/ops/_op_impl/_custom_op/fused_abs_max1_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/img2col_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_left_impl.py +2 -2
- mindspore/ops/_op_impl/_custom_op/matmul_cube_dense_right_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_left_cast_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_fracz_right_mul_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/matmul_cube_impl.py +2 -2
- mindspore/ops/_op_impl/_custom_op/matmul_dds_impl.py +0 -4
- mindspore/ops/_op_impl/_custom_op/matrix_combine_impl.py +1 -1
- mindspore/ops/_op_impl/_custom_op/minmax_update_perchannel.py +2 -2
- mindspore/ops/_op_impl/_custom_op/minmax_update_perlayer.py +2 -2
- mindspore/ops/_op_impl/_custom_op/transpose02314_impl.py +1 -1
- mindspore/ops/_op_impl/aicpu/__init__.py +236 -4
- mindspore/ops/_op_impl/aicpu/abs.py +36 -0
- mindspore/ops/_op_impl/aicpu/{adaptive_avg_pool_2d_v1.py → adaptive_avg_pool_2d.py} +6 -5
- mindspore/ops/_op_impl/aicpu/adaptive_avg_pool_2d_grad.py +34 -0
- mindspore/ops/_op_impl/aicpu/add.py +43 -0
- mindspore/ops/_op_impl/aicpu/addcdiv.py +0 -32
- mindspore/ops/_op_impl/aicpu/addcmul.py +0 -84
- mindspore/ops/_op_impl/aicpu/affine_grid_grad.py +35 -0
- mindspore/ops/_op_impl/aicpu/batch_matmul.py +43 -43
- mindspore/ops/_op_impl/aicpu/bernoulli.py +48 -0
- mindspore/{compression/common/__init__.py → ops/_op_impl/aicpu/bessel_i0.py} +15 -8
- mindspore/ops/_op_impl/aicpu/channel_shuffle.py +40 -0
- mindspore/ops/_op_impl/aicpu/conj.py +11 -0
- mindspore/ops/_op_impl/aicpu/cumulative_logsumexp.py +0 -3
- mindspore/ops/_op_impl/aicpu/deformable_offsets.py +38 -0
- mindspore/ops/_op_impl/aicpu/deformable_offsets_grad.py +43 -0
- mindspore/ops/_op_impl/aicpu/{adaptive_avg_pool_2d_grad_v1.py → digamma.py} +7 -9
- mindspore/ops/_op_impl/aicpu/flatten.py +1 -0
- mindspore/ops/_op_impl/aicpu/fmax.py +36 -0
- mindspore/ops/_op_impl/aicpu/fmin.py +37 -0
- mindspore/ops/_op_impl/aicpu/fractional_max_pool3d_with_fixed_ksize.py +1 -1
- mindspore/ops/_op_impl/aicpu/fse_decode.py +43 -0
- mindspore/ops/_op_impl/aicpu/greater.py +41 -0
- mindspore/ops/_op_impl/aicpu/greater_equal.py +41 -0
- mindspore/ops/_op_impl/aicpu/index_put.py +50 -0
- mindspore/ops/_op_impl/aicpu/less.py +41 -0
- mindspore/{nn/probability/infer/variational/__init__.py → ops/_op_impl/aicpu/lgamma.py} +16 -10
- mindspore/ops/_op_impl/aicpu/mirror_pad.py +0 -4
- mindspore/ops/_op_impl/aicpu/mirror_pad_grad.py +0 -4
- mindspore/ops/_op_impl/aicpu/mul.py +3 -1
- mindspore/ops/_op_impl/aicpu/multinomial.py +14 -6
- mindspore/ops/_op_impl/aicpu/nllloss.py +38 -0
- mindspore/ops/_op_impl/aicpu/nllloss_grad.py +39 -0
- mindspore/ops/_op_impl/aicpu/ones_like.py +0 -2
- mindspore/ops/_op_impl/aicpu/polar.py +32 -0
- mindspore/ops/_op_impl/aicpu/polygamma.py +34 -0
- mindspore/ops/_op_impl/aicpu/quant_dtype_cast.py +40 -0
- mindspore/ops/_op_impl/aicpu/quantile.py +35 -0
- mindspore/ops/_op_impl/aicpu/ragged_tensor_to_sparse.py +73 -0
- mindspore/ops/_op_impl/aicpu/randperm_v2.py +41 -0
- mindspore/ops/_op_impl/aicpu/resize_bicubic.py +2 -8
- mindspore/ops/_op_impl/aicpu/resize_bicubic_grad.py +1 -1
- mindspore/ops/_op_impl/aicpu/resize_v2.py +68 -0
- mindspore/ops/_op_impl/aicpu/resize_v2_grad.py +68 -0
- mindspore/ops/_op_impl/aicpu/scatter_elements.py +4 -0
- mindspore/ops/_op_impl/aicpu/scatter_nd_update.py +2 -0
- mindspore/ops/_op_impl/aicpu/sequence_add.py +34 -0
- mindspore/ops/_op_impl/aicpu/sequence_add_offset.py +34 -0
- mindspore/ops/_op_impl/aicpu/sequence_addn.py +38 -0
- mindspore/ops/_op_impl/aicpu/smooth_l1_loss.py +35 -0
- mindspore/ops/_op_impl/aicpu/smooth_l1_loss_grad.py +37 -0
- mindspore/ops/_op_impl/aicpu/sparse_apply_adagrad_da.py +0 -24
- mindspore/ops/_op_impl/aicpu/sparse_cross.py +42 -0
- mindspore/ops/_op_impl/aicpu/sparse_slice.py +4 -0
- mindspore/ops/_op_impl/aicpu/sparse_slice_grad.py +6 -0
- mindspore/ops/_op_impl/aicpu/tensor_scatter_update.py +59 -0
- mindspore/ops/_op_impl/aicpu/trans_data.py +1 -0
- mindspore/ops/_op_impl/aicpu/tril_indices.py +34 -0
- mindspore/ops/_op_impl/aicpu/uniform.py +34 -0
- mindspore/ops/_op_impl/aicpu/uniform_candidate_sampler.py +1 -0
- mindspore/ops/_op_impl/aicpu/unique_consecutive.py +10 -2
- mindspore/ops/_op_impl/cpu/dynamic_shape.py +5 -1
- mindspore/ops/_op_impl/cpu/sparse_slice.py +4 -0
- mindspore/ops/_op_impl/cpu/sparse_slice_grad.py +6 -0
- mindspore/ops/_op_impl/cpu/tensor_shape.py +5 -1
- mindspore/ops/_op_impl/tbe/__init__.py +27 -611
- mindspore/ops/_op_impl/tbe/assign_add_ds.py +1 -0
- mindspore/ops/_op_impl/tbe/atomic_addr_clean.py +1 -1
- mindspore/ops/_op_impl/tbe/avg_pool_3d_grad.py +1 -1
- mindspore/ops/_op_impl/tbe/batch_matmul_ds.py +1 -0
- mindspore/ops/_op_impl/tbe/batch_to_space.py +1 -1
- mindspore/ops/_op_impl/tbe/batch_to_space_nd.py +1 -1
- mindspore/ops/_op_impl/tbe/bn_infer_grad.py +4 -2
- mindspore/ops/_op_impl/tbe/bn_training_update.py +0 -1
- mindspore/ops/_op_impl/tbe/bn_training_update_ds.py +0 -1
- mindspore/ops/_op_impl/tbe/broadcast_to_ds.py +6 -4
- mindspore/ops/_op_impl/tbe/cast.py +0 -2
- mindspore/ops/_op_impl/tbe/cast_ds.py +3 -3
- mindspore/ops/_op_impl/tbe/data_format_dim_map_ds.py +1 -0
- mindspore/ops/_op_impl/tbe/depthwise_conv2d.py +2 -2
- mindspore/ops/_op_impl/tbe/dynamic_atomic_addr_clean.py +1 -1
- mindspore/ops/_op_impl/tbe/gather_nd.py +1 -0
- mindspore/ops/_op_impl/tbe/{index_add.py → inplace_index_add.py} +3 -6
- mindspore/ops/_op_impl/tbe/matmul_ds.py +2 -0
- mindspore/ops/_op_impl/tbe/npu_clear_float_status_v2.py +35 -0
- mindspore/ops/_op_impl/tbe/npu_get_float_status_v2.py +35 -0
- mindspore/ops/_op_impl/tbe/scatter_mul.py +2 -0
- mindspore/ops/_op_impl/tbe/scatter_nd_add.py +0 -2
- mindspore/ops/_op_impl/tbe/space_to_batch.py +1 -1
- mindspore/ops/_op_impl/tbe/space_to_batch_nd.py +1 -1
- mindspore/ops/_op_impl/tbe/trans_data_ds.py +15 -5
- mindspore/ops/_register_for_op.py +1 -0
- mindspore/ops/_utils/__init__.py +1 -2
- mindspore/ops/_utils/utils.py +19 -40
- mindspore/ops/_vmap/vmap_array_ops.py +116 -38
- mindspore/ops/_vmap/vmap_base.py +16 -9
- mindspore/ops/_vmap/vmap_convolution_ops.py +7 -10
- mindspore/ops/_vmap/vmap_grad_math_ops.py +4 -4
- mindspore/ops/_vmap/vmap_grad_nn_ops.py +7 -5
- mindspore/ops/_vmap/vmap_image_ops.py +12 -5
- mindspore/ops/_vmap/vmap_math_ops.py +46 -5
- mindspore/ops/_vmap/vmap_nn_ops.py +15 -21
- mindspore/ops/_vmap/vmap_random_ops.py +1 -1
- mindspore/ops/bprop_mindir/AdaptiveAvgPool2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/AdaptiveMaxPool2D_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/AvgPool3D_bprop.mindir +150 -0
- mindspore/ops/bprop_mindir/AvgPool_bprop.mindir +66 -0
- mindspore/ops/bprop_mindir/BCEWithLogitsLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BatchNormGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BiasAddGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/BinaryCrossEntropy_bprop.mindir +33 -0
- mindspore/ops/bprop_mindir/BroadcastTo_bprop.mindir +220 -106
- mindspore/ops/bprop_mindir/CTCLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Conv2DBackpropFilter_bprop.mindir +240 -0
- mindspore/ops/bprop_mindir/Conv2DBackpropInput_bprop.mindir +247 -0
- mindspore/ops/bprop_mindir/Conv2DTranspose_bprop.mindir +247 -0
- mindspore/ops/bprop_mindir/Conv3DTranspose_bprop.mindir +315 -0
- mindspore/ops/bprop_mindir/Conv3D_bprop.mindir +278 -0
- mindspore/ops/bprop_mindir/DeformableOffsets_bprop.mindir +58 -0
- mindspore/ops/bprop_mindir/DepthwiseConv2dNative_bprop.mindir +138 -0
- 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 +22 -23
- mindspore/ops/bprop_mindir/DropoutGenMask_bprop.mindir +16 -17
- mindspore/ops/bprop_mindir/DropoutGrad_bprop.mindir +27 -0
- 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/Elu_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/EmbeddingLookup_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ExpandDims_bprop.mindir +39 -41
- mindspore/ops/bprop_mindir/FastGeLU_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/Flatten_bprop.mindir +41 -43
- mindspore/ops/bprop_mindir/GatherNd_bprop.mindir +51 -57
- mindspore/ops/bprop_mindir/Gather_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/HSigmoid_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/HSwish_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/InstanceNorm_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/KLDivLoss_bprop.mindir +126 -0
- mindspore/ops/bprop_mindir/L2Loss_bprop.mindir +15 -0
- mindspore/ops/bprop_mindir/L2Normalize_bprop.mindir +30 -0
- mindspore/ops/bprop_mindir/LRN_bprop.mindir +43 -0
- mindspore/ops/bprop_mindir/LayerNormGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/LogSoftmax_bprop.mindir +23 -0
- mindspore/ops/bprop_mindir/MaxPool3DGradGrad_bprop.mindir +74 -0
- mindspore/ops/bprop_mindir/MaxPool3DGrad_bprop.mindir +74 -0
- mindspore/ops/bprop_mindir/MaxPool3D_bprop.mindir +75 -0
- mindspore/ops/bprop_mindir/MaxPoolGradGrad_bprop.mindir +65 -0
- mindspore/ops/bprop_mindir/MaxPoolWithArgmax_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/MirrorPad_bprop.mindir +27 -0
- mindspore/ops/bprop_mindir/Mish_bprop.mindir +35 -0
- mindspore/ops/bprop_mindir/MulNoNan_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/NLLLoss_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/OneHot_bprop.mindir +24 -25
- 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 +29 -0
- mindspore/ops/bprop_mindir/ROIAlign_bprop.mindir +82 -0
- mindspore/ops/bprop_mindir/ReLU6_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/ReLUV2_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ReluGrad_bprop.mindir +18 -19
- mindspore/ops/bprop_mindir/Reshape_bprop.mindir +53 -53
- mindspore/ops/bprop_mindir/ResizeBilinear_bprop.mindir +29 -0
- mindspore/ops/bprop_mindir/ResizeNearestNeighbor_bprop.mindir +77 -85
- mindspore/ops/bprop_mindir/SeLU_bprop.mindir +21 -0
- mindspore/ops/bprop_mindir/SigmoidCrossEntropyWithLogits_bprop.mindir +21 -0
- mindspore/ops/bprop_mindir/SigmoidGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Sigmoid_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/SmoothL1Loss_bprop.mindir +36 -0
- mindspore/ops/bprop_mindir/SoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Softplus_bprop.mindir +16 -0
- mindspore/ops/bprop_mindir/Softsign_bprop.mindir +33 -0
- mindspore/ops/bprop_mindir/SparseSoftmaxCrossEntropyWithLogits_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Squeeze_bprop.mindir +37 -39
- mindspore/ops/bprop_mindir/StridedSlice_bprop.mindir +70 -72
- mindspore/ops/bprop_mindir/TanhGrad_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/Tanh_bprop.mindir +66 -0
- mindspore/ops/bprop_mindir/Tile_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TopK_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/TupleGetItem_bprop.mindir +17 -17
- mindspore/ops/bprop_mindir/UpsampleNearest3D_bprop.mindir +32 -0
- mindspore/ops/bprop_mindir/UpsampleTrilinear3D_bprop.mindir +38 -0
- mindspore/ops/bprop_mindir/generate_mindir.py +2 -0
- mindspore/ops/composite/__init__.py +7 -8
- mindspore/ops/composite/base.py +101 -47
- mindspore/ops/composite/math_ops.py +188 -158
- mindspore/ops/composite/multitype_ops/_compile_utils.py +415 -170
- mindspore/ops/composite/multitype_ops/_constexpr_utils.py +142 -87
- mindspore/ops/composite/multitype_ops/add_impl.py +6 -1
- mindspore/ops/composite/multitype_ops/div_impl.py +2 -3
- mindspore/ops/composite/multitype_ops/getitem_impl.py +31 -3
- mindspore/ops/composite/multitype_ops/greater_equal_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/greater_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/in_impl.py +9 -0
- mindspore/ops/composite/multitype_ops/less_equal_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/less_impl.py +31 -0
- mindspore/ops/composite/multitype_ops/mul_impl.py +21 -5
- mindspore/ops/composite/multitype_ops/not_in_impl.py +9 -0
- mindspore/ops/composite/multitype_ops/ones_like_impl.py +2 -4
- mindspore/ops/composite/multitype_ops/setitem_impl.py +21 -3
- mindspore/ops/composite/multitype_ops/sub_impl.py +1 -1
- mindspore/ops/composite/multitype_ops/zeros_like_impl.py +35 -4
- mindspore/ops/function/__init__.py +152 -8
- mindspore/ops/function/array_func.py +2555 -674
- mindspore/ops/function/clip_func.py +209 -13
- mindspore/ops/function/debug_func.py +2 -2
- mindspore/ops/function/grad/__init__.py +2 -1
- mindspore/ops/function/grad/grad_func.py +147 -62
- mindspore/ops/function/image_func.py +54 -38
- mindspore/ops/function/linalg_func.py +167 -16
- mindspore/ops/function/math_func.py +4849 -1492
- mindspore/ops/function/nn_func.py +2573 -988
- mindspore/ops/function/other_func.py +115 -0
- mindspore/ops/function/parameter_func.py +3 -3
- mindspore/ops/function/random_func.py +790 -73
- mindspore/ops/function/sparse_func.py +98 -78
- mindspore/ops/function/sparse_unary_func.py +54 -53
- mindspore/ops/function/spectral_func.py +27 -24
- mindspore/ops/function/vmap_func.py +22 -2
- mindspore/ops/functional.py +97 -37
- mindspore/ops/op_info_register.py +70 -28
- mindspore/ops/operations/__init__.py +47 -14
- mindspore/ops/operations/_csr_ops.py +7 -7
- mindspore/ops/operations/_embedding_cache_ops.py +5 -5
- mindspore/ops/operations/_grad_ops.py +276 -187
- mindspore/ops/operations/_inner_ops.py +319 -113
- mindspore/ops/operations/_ms_kernel.py +10 -8
- mindspore/ops/operations/_ocr_ops.py +9 -9
- mindspore/ops/operations/_opaque_predicate_registry.py +4 -0
- mindspore/ops/operations/_quant_ops.py +137 -102
- mindspore/ops/operations/_rl_inner_ops.py +121 -60
- mindspore/ops/operations/_scalar_ops.py +466 -0
- mindspore/ops/operations/_sequence_ops.py +1004 -2
- mindspore/ops/operations/_tensor_array.py +10 -11
- mindspore/ops/operations/_thor_ops.py +1 -1
- mindspore/ops/operations/array_ops.py +801 -466
- mindspore/ops/operations/comm_ops.py +51 -49
- mindspore/ops/operations/control_ops.py +2 -2
- mindspore/ops/operations/custom_ops.py +123 -44
- mindspore/ops/operations/debug_ops.py +24 -24
- mindspore/ops/operations/image_ops.py +240 -153
- mindspore/ops/operations/inner_ops.py +34 -50
- mindspore/ops/operations/linalg_ops.py +31 -9
- mindspore/ops/operations/math_ops.py +988 -757
- mindspore/ops/operations/nn_ops.py +965 -819
- mindspore/ops/operations/other_ops.py +51 -40
- mindspore/ops/operations/random_ops.py +204 -122
- mindspore/ops/operations/rl_ops.py +8 -9
- mindspore/ops/operations/sparse_ops.py +254 -93
- mindspore/ops/operations/spectral_ops.py +35 -3
- mindspore/ops/primitive.py +111 -9
- mindspore/parallel/_auto_parallel_context.py +189 -83
- mindspore/parallel/_offload_context.py +185 -0
- mindspore/parallel/_parallel_serialization.py +99 -7
- mindspore/parallel/_ps_context.py +9 -5
- mindspore/parallel/_recovery_context.py +1 -1
- mindspore/parallel/_tensor.py +7 -1
- mindspore/{nn/transformer → parallel/_transformer}/__init__.py +6 -6
- mindspore/{nn/transformer → parallel/_transformer}/layers.py +6 -37
- mindspore/{nn/transformer → parallel/_transformer}/loss.py +4 -7
- mindspore/{nn/transformer → parallel/_transformer}/moe.py +20 -16
- mindspore/{nn/transformer → parallel/_transformer}/op_parallel_config.py +3 -3
- mindspore/{nn/transformer → parallel/_transformer}/transformer.py +48 -111
- mindspore/parallel/_utils.py +1 -2
- mindspore/parallel/algo_parameter_config.py +1 -1
- mindspore/parallel/checkpoint_transform.py +37 -34
- mindspore/parallel/shard.py +17 -18
- mindspore/profiler/common/validator/validate_path.py +2 -2
- mindspore/profiler/envprofiling.py +69 -47
- mindspore/profiler/parser/ascend_timeline_generator.py +49 -42
- mindspore/profiler/parser/base_timeline_generator.py +49 -56
- mindspore/profiler/parser/cpu_gpu_timeline_generator.py +98 -78
- mindspore/profiler/parser/hwts_log_parser.py +1 -1
- mindspore/profiler/parser/integrator.py +15 -14
- mindspore/profiler/parser/minddata_analyzer.py +2 -2
- mindspore/profiler/parser/msadvisor_analyzer.py +12 -25
- mindspore/profiler/parser/msadvisor_parser.py +2 -4
- mindspore/profiler/parser/optime_parser.py +17 -18
- mindspore/profiler/parser/profiler_info.py +2 -1
- mindspore/profiler/profiling.py +218 -186
- mindspore/rewrite/__init__.py +3 -1
- mindspore/rewrite/api/node.py +1 -114
- mindspore/rewrite/api/node_type.py +3 -0
- mindspore/rewrite/api/pattern_engine.py +31 -1
- mindspore/rewrite/api/scoped_value.py +4 -4
- mindspore/rewrite/api/symbol_tree.py +3 -78
- mindspore/rewrite/api/tree_node_helper.py +1 -1
- mindspore/rewrite/ast_creator_register.py +1 -0
- mindspore/rewrite/ast_helpers/__init__.py +2 -2
- mindspore/rewrite/ast_helpers/ast_creator.py +1 -2
- mindspore/rewrite/ast_helpers/ast_finder.py +65 -0
- mindspore/rewrite/ast_helpers/ast_modifier.py +11 -3
- mindspore/rewrite/ast_transformers/flatten_recursive_stmt.py +18 -2
- mindspore/rewrite/namespace.py +0 -2
- mindspore/rewrite/node.py +157 -11
- mindspore/rewrite/parsers/assign_parser.py +231 -53
- mindspore/rewrite/parsers/class_def_parser.py +187 -109
- mindspore/rewrite/parsers/for_parser.py +24 -14
- mindspore/rewrite/parsers/function_def_parser.py +21 -4
- mindspore/rewrite/parsers/if_parser.py +6 -2
- mindspore/rewrite/sparsify/__init__.py +0 -0
- mindspore/rewrite/sparsify/sparse_transformer.py +448 -0
- mindspore/rewrite/sparsify/sparsify.py +109 -0
- mindspore/rewrite/sparsify/utils.py +173 -0
- mindspore/rewrite/symbol_tree.py +256 -133
- mindspore/rewrite/symbol_tree_builder.py +38 -1
- mindspore/run_check/_check_version.py +69 -63
- mindspore/run_check/run_check.py +2 -1
- mindspore/scipy/linalg.py +10 -114
- mindspore/scipy/ops.py +2 -2
- mindspore/scipy/ops_wrapper.py +1 -1
- mindspore/scipy/optimize/_bfgs.py +1 -1
- mindspore/scipy/optimize/_lagrange.py +200 -0
- mindspore/scipy/optimize/line_search.py +3 -2
- mindspore/scipy/optimize/minimize.py +41 -2
- mindspore/scipy/sparse/__init__.py +2 -2
- mindspore/scipy/sparse/linalg.py +4 -464
- mindspore/scipy/utils.py +1 -1
- mindspore/scipy/utils_const.py +7 -1
- mindspore/train/__init__.py +1 -1
- mindspore/train/_utils.py +28 -5
- mindspore/train/amp.py +273 -102
- mindspore/train/callback/_backup_and_restore.py +5 -5
- mindspore/train/callback/_callback.py +2 -2
- mindspore/train/callback/_checkpoint.py +3 -3
- mindspore/train/callback/_early_stop.py +3 -3
- mindspore/train/callback/_lambda_callback.py +2 -2
- mindspore/train/callback/_landscape.py +29 -31
- mindspore/train/callback/_loss_monitor.py +3 -3
- mindspore/train/callback/_on_request_exit.py +3 -3
- mindspore/train/callback/_reduce_lr_on_plateau.py +4 -4
- mindspore/train/callback/_summary_collector.py +23 -16
- mindspore/train/callback/_time_monitor.py +3 -3
- mindspore/train/checkpoint_pb2.py +68 -8
- mindspore/train/data_sink.py +15 -3
- mindspore/train/dataset_helper.py +10 -15
- mindspore/train/loss_scale_manager.py +8 -11
- mindspore/train/metrics/__init__.py +1 -1
- mindspore/train/metrics/bleu_score.py +1 -1
- mindspore/train/metrics/confusion_matrix.py +1 -1
- mindspore/train/metrics/cosine_similarity.py +1 -1
- mindspore/train/metrics/dice.py +2 -2
- mindspore/train/metrics/fbeta.py +1 -1
- mindspore/train/metrics/hausdorff_distance.py +4 -3
- mindspore/train/metrics/mean_surface_distance.py +2 -2
- mindspore/train/metrics/occlusion_sensitivity.py +1 -1
- mindspore/train/metrics/perplexity.py +1 -1
- mindspore/train/metrics/precision.py +1 -1
- mindspore/train/metrics/recall.py +1 -1
- mindspore/train/metrics/roc.py +2 -2
- mindspore/train/metrics/root_mean_square_surface_distance.py +2 -2
- mindspore/train/mind_ir_pb2.py +116 -37
- mindspore/train/model.py +45 -28
- mindspore/train/serialization.py +295 -188
- mindspore/train/summary/_summary_adapter.py +1 -1
- mindspore/train/summary/summary_record.py +43 -13
- mindspore/train/train_thor/convert_utils.py +2 -2
- mindspore/train/train_thor/dataset_helper.py +3 -3
- mindspore/version.py +1 -1
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/METADATA +3 -2
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/RECORD +648 -574
- mindspore/compression/__init__.py +0 -19
- mindspore/compression/common/constant.py +0 -124
- mindspore/compression/export/__init__.py +0 -19
- mindspore/compression/export/quant_export.py +0 -515
- mindspore/compression/quant/__init__.py +0 -28
- mindspore/compression/quant/qat.py +0 -634
- mindspore/compression/quant/quant_utils.py +0 -462
- mindspore/compression/quant/quantizer.py +0 -68
- mindspore/nn/layer/quant.py +0 -1868
- mindspore/nn/layer/rnn_utils.py +0 -90
- mindspore/nn/probability/dpn/__init__.py +0 -22
- mindspore/nn/probability/dpn/vae/__init__.py +0 -25
- mindspore/nn/probability/dpn/vae/cvae.py +0 -140
- mindspore/nn/probability/dpn/vae/vae.py +0 -124
- mindspore/nn/probability/infer/__init__.py +0 -22
- mindspore/nn/probability/infer/variational/elbo.py +0 -70
- mindspore/nn/probability/infer/variational/svi.py +0 -84
- mindspore/nn/probability/toolbox/__init__.py +0 -22
- mindspore/nn/probability/toolbox/anomaly_detection.py +0 -99
- mindspore/nn/probability/toolbox/uncertainty_evaluation.py +0 -364
- mindspore/nn/probability/transforms/__init__.py +0 -22
- mindspore/nn/probability/transforms/transform_bnn.py +0 -262
- mindspore/nn/probability/zhusuan/__init__.py +0 -18
- mindspore/nn/probability/zhusuan/framework/__init__.py +0 -18
- mindspore/nn/probability/zhusuan/framework/bn.py +0 -95
- mindspore/nn/probability/zhusuan/variational/__init__.py +0 -18
- mindspore/nn/probability/zhusuan/variational/elbo.py +0 -46
- mindspore/ops/_op_impl/aicpu/parallel_concat.py +0 -42
- mindspore/ops/_op_impl/tbe/gather_v2.py +0 -56
- mindspore/ops/bprop_mindir/AssignAdd_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/Cast_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/LogicalOr_bprop.mindir +0 -19
- mindspore/ops/bprop_mindir/MatMul_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/ReLU_bprop.mindir +0 -17
- mindspore/ops/bprop_mindir/Transpose_bprop.mindir +0 -0
- mindspore/ops/bprop_mindir/UpdateState_bprop.mindir +0 -15
- mindspore/ops/composite/array_ops.py +0 -241
- mindspore/ops/composite/clip_ops.py +0 -134
- mindspore/ops/composite/random_ops.py +0 -426
- mindspore/ops/composite/vmap_ops.py +0 -38
- mindspore/parallel/nn/__init__.py +0 -42
- mindspore/parallel/nn/loss.py +0 -22
- mindspore/parallel/nn/moe.py +0 -21
- mindspore/parallel/nn/op_parallel_config.py +0 -22
- mindspore/parallel/nn/transformer.py +0 -31
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/WHEEL +0 -0
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/entry_points.txt +0 -0
- {mindspore-2.0.0a0.dist-info → mindspore-2.0.0rc1.dist-info}/top_level.txt +0 -0
|
@@ -35,7 +35,7 @@ class EnvChecker(metaclass=ABCMeta):
|
|
|
35
35
|
|
|
36
36
|
@abstractmethod
|
|
37
37
|
def check_env(self):
|
|
38
|
-
|
|
38
|
+
"""check dependency"""
|
|
39
39
|
|
|
40
40
|
@abstractmethod
|
|
41
41
|
def set_env(self):
|
|
@@ -95,8 +95,8 @@ class GPUEnvChecker(EnvChecker):
|
|
|
95
95
|
version_match = True
|
|
96
96
|
if not version_match:
|
|
97
97
|
if self.v == "0":
|
|
98
|
-
logger.warning("Can not found cuda libs
|
|
99
|
-
"cuda version has been installed
|
|
98
|
+
logger.warning("Can not found cuda libs. Please confirm that the correct "
|
|
99
|
+
"cuda version has been installed. Refer to the "
|
|
100
100
|
"installation guidelines: https://www.mindspore.cn/install")
|
|
101
101
|
else:
|
|
102
102
|
logger.warning(f"MindSpore version {__version__} and cuda version {self.v} does not match, "
|
|
@@ -106,17 +106,17 @@ class GPUEnvChecker(EnvChecker):
|
|
|
106
106
|
nvcc_version = self._get_nvcc_version(False)
|
|
107
107
|
if nvcc_version and (nvcc_version not in self.version):
|
|
108
108
|
logger.warning(f"MindSpore version {__version__} and nvcc(cuda bin) version {nvcc_version} "
|
|
109
|
-
"does not match
|
|
109
|
+
"does not match. Please refer to the installation guide for version matching "
|
|
110
110
|
"information: https://www.mindspore.cn/install")
|
|
111
111
|
cudnn_version = self._get_cudnn_version()
|
|
112
112
|
if cudnn_version and int(cudnn_version) < 760:
|
|
113
113
|
logger.warning(f"MindSpore version {__version__} and cudDNN version {cudnn_version} "
|
|
114
|
-
"does not match
|
|
114
|
+
"does not match. Please refer to the installation guide for version matching "
|
|
115
115
|
"information: https://www.mindspore.cn/install. The recommended version is "
|
|
116
116
|
"CUDA10.1 with cuDNN7.6.x, CUDA11.1 with cuDNN8.0.x and CUDA11.6 with cuDNN8.5.x.")
|
|
117
117
|
if cudnn_version and int(cudnn_version) < 800 and int(str(self.v).split('.')[0]) > 10:
|
|
118
118
|
logger.warning(f"CUDA version {self.v} and cuDNN version {cudnn_version} "
|
|
119
|
-
"does not match
|
|
119
|
+
"does not match. Please refer to the installation guide for version matching "
|
|
120
120
|
"information: https://www.mindspore.cn/install. The recommended version is "
|
|
121
121
|
"CUDA11.1 with cuDNN8.0.x or CUDA11.6 with cuDNN8.5.x.")
|
|
122
122
|
|
|
@@ -211,8 +211,8 @@ class GPUEnvChecker(EnvChecker):
|
|
|
211
211
|
mindspore_path = os.path.join(current_path, "../lib/plugin")
|
|
212
212
|
try:
|
|
213
213
|
real_path = self.library_path
|
|
214
|
-
if real_path == []:
|
|
215
|
-
logger.error(f"{self.lib_key_to_lib_name[lib_name]} (need by mindspore-gpu) is not found
|
|
214
|
+
if real_path is None or real_path == []:
|
|
215
|
+
logger.error(f"{self.lib_key_to_lib_name[lib_name]} (need by mindspore-gpu) is not found. Please "
|
|
216
216
|
f"confirm that libmindspore_gpu.so is in directory:{mindspore_path} and the correct cuda "
|
|
217
217
|
"version has been installed, you can refer to the installation "
|
|
218
218
|
"guidelines: https://www.mindspore.cn/install")
|
|
@@ -223,10 +223,10 @@ class GPUEnvChecker(EnvChecker):
|
|
|
223
223
|
for i in result.split('\n'):
|
|
224
224
|
path = i.partition("=>")[2]
|
|
225
225
|
if path.lower().find("not found") > 0:
|
|
226
|
-
logger.error(f"Cuda {self.version} version({lib_name}*.so need by mindspore-gpu) is not found
|
|
227
|
-
"
|
|
226
|
+
logger.error(f"Cuda {self.version} version({lib_name}*.so need by mindspore-gpu) is not found. "
|
|
227
|
+
"Please confirm that the path of cuda is set to the env LD_LIBRARY_PATH, or check "
|
|
228
228
|
"whether the CUDA version in wheel package and the CUDA runtime in current device "
|
|
229
|
-
"matches
|
|
229
|
+
"matches. Please refer to the installation guidelines: "
|
|
230
230
|
"https://www.mindspore.cn/install")
|
|
231
231
|
continue
|
|
232
232
|
path = path.partition(lib_name)[0]
|
|
@@ -234,8 +234,8 @@ class GPUEnvChecker(EnvChecker):
|
|
|
234
234
|
path_list.append(os.path.abspath(path.strip() + "../"))
|
|
235
235
|
return np.unique(path_list)
|
|
236
236
|
except subprocess.TimeoutExpired:
|
|
237
|
-
logger.warning("Failed to check cuda version due to the ldd command timeout
|
|
238
|
-
"the correct cuda version has been installed
|
|
237
|
+
logger.warning("Failed to check cuda version due to the ldd command timeout. Please confirm that "
|
|
238
|
+
"the correct cuda version has been installed. For details, refer to the "
|
|
239
239
|
"installation guidelines: https://www.mindspore.cn/install")
|
|
240
240
|
return path_list
|
|
241
241
|
|
|
@@ -255,14 +255,14 @@ class AscendEnvChecker(EnvChecker):
|
|
|
255
255
|
|
|
256
256
|
def __init__(self, library_path):
|
|
257
257
|
self.library_path = library_path
|
|
258
|
-
self.version = ["
|
|
258
|
+
self.version = ["6.11", "6.3"]
|
|
259
259
|
atlas_nnae_version = "/usr/local/Ascend/nnae/latest/compiler/version.info"
|
|
260
260
|
atlas_toolkit_version = "/usr/local/Ascend/ascend-toolkit/latest/compiler/version.info"
|
|
261
261
|
hisi_fwk_version = "/usr/local/Ascend/latest/compiler/version.info"
|
|
262
262
|
if os.path.exists(atlas_nnae_version):
|
|
263
263
|
# atlas default path
|
|
264
264
|
self.fwk_path = "/usr/local/Ascend/nnae/latest"
|
|
265
|
-
self.op_impl_path = "/usr/local/Ascend/nnae/latest/opp/
|
|
265
|
+
self.op_impl_path = "/usr/local/Ascend/nnae/latest/opp/built-in/op_impl/ai_core/tbe"
|
|
266
266
|
self.tbe_path = self.fwk_path + "/lib64"
|
|
267
267
|
self.cce_path = self.fwk_path + "/compiler/ccec_compiler/bin"
|
|
268
268
|
self.fwk_version = atlas_nnae_version
|
|
@@ -271,7 +271,7 @@ class AscendEnvChecker(EnvChecker):
|
|
|
271
271
|
elif os.path.exists(atlas_toolkit_version):
|
|
272
272
|
# atlas default path
|
|
273
273
|
self.fwk_path = "/usr/local/Ascend/ascend-toolkit/latest"
|
|
274
|
-
self.op_impl_path = "/usr/local/Ascend/ascend-toolkit/latest/opp/
|
|
274
|
+
self.op_impl_path = "/usr/local/Ascend/ascend-toolkit/latest/opp/built-in/op_impl/ai_core/tbe"
|
|
275
275
|
self.tbe_path = self.fwk_path + "/lib64"
|
|
276
276
|
self.cce_path = self.fwk_path + "/compiler/ccec_compiler/bin"
|
|
277
277
|
self.fwk_version = atlas_toolkit_version
|
|
@@ -280,7 +280,7 @@ class AscendEnvChecker(EnvChecker):
|
|
|
280
280
|
elif os.path.exists(hisi_fwk_version):
|
|
281
281
|
# hisi default path
|
|
282
282
|
self.fwk_path = "/usr/local/Ascend/latest"
|
|
283
|
-
self.op_impl_path = "/usr/local/Ascend/latest/opp/
|
|
283
|
+
self.op_impl_path = "/usr/local/Ascend/latest/opp/built-in/op_impl/ai_core/tbe"
|
|
284
284
|
self.tbe_path = self.fwk_path + "/lib64"
|
|
285
285
|
self.cce_path = self.fwk_path + "/compiler/ccec_compiler/bin"
|
|
286
286
|
self.fwk_version = hisi_fwk_version
|
|
@@ -305,7 +305,7 @@ class AscendEnvChecker(EnvChecker):
|
|
|
305
305
|
|
|
306
306
|
# check content
|
|
307
307
|
self.path_check = "/compiler/ccec_compiler/bin"
|
|
308
|
-
self.python_path_check = "opp/
|
|
308
|
+
self.python_path_check = "opp/built-in/op_impl/ai_core/tbe"
|
|
309
309
|
self.ld_lib_path_check_fwk = "/lib64"
|
|
310
310
|
self.ld_lib_path_check_addons = "/add-ons"
|
|
311
311
|
self.ascend_opp_path_check = "/op"
|
|
@@ -317,8 +317,8 @@ class AscendEnvChecker(EnvChecker):
|
|
|
317
317
|
def check_version(self):
|
|
318
318
|
if not Path(self.fwk_version).is_file():
|
|
319
319
|
logger.warning("Using custom Ascend AI software package (Ascend Data Center Solution) path, package "
|
|
320
|
-
"version checking is skipped
|
|
321
|
-
"Center Solution) version is supported
|
|
320
|
+
"version checking is skipped. Please make sure Ascend AI software package (Ascend Data "
|
|
321
|
+
"Center Solution) version is supported. For details, refer to the installation guidelines "
|
|
322
322
|
"https://www.mindspore.cn/install")
|
|
323
323
|
return
|
|
324
324
|
|
|
@@ -327,11 +327,11 @@ class AscendEnvChecker(EnvChecker):
|
|
|
327
327
|
v_list = str([x for x in self.version])
|
|
328
328
|
logger.warning(f"MindSpore version {__version__} and Ascend AI software package (Ascend Data Center "
|
|
329
329
|
f"Solution)version {v} does not match, the version of software package expect one of "
|
|
330
|
-
f"{v_list}
|
|
330
|
+
f"{v_list}. Please refer to the match info on: https://www.mindspore.cn/install")
|
|
331
331
|
|
|
332
332
|
def check_deps_version(self):
|
|
333
333
|
"""
|
|
334
|
-
te
|
|
334
|
+
te and hccl wheel package version check
|
|
335
335
|
in order to update the change of 'LD_LIBRARY_PATH' env, run a sub process
|
|
336
336
|
"""
|
|
337
337
|
|
|
@@ -344,24 +344,22 @@ class AscendEnvChecker(EnvChecker):
|
|
|
344
344
|
if v not in supported_version:
|
|
345
345
|
attention_warning = True
|
|
346
346
|
logger.warning(f"MindSpore version {mindspore_version} and \"te\" wheel package version {v} does not "
|
|
347
|
-
"match,
|
|
348
|
-
|
|
349
|
-
v = '.'.join(topiver.version.split('.')[0:2])
|
|
350
|
-
if v not in supported_version:
|
|
351
|
-
attention_warning = True
|
|
352
|
-
logger.warning(f"MindSpore version {mindspore_version} and \"topi\" wheel package version {v} does not "
|
|
353
|
-
"match, reference to the match info on: https://www.mindspore.cn/install")
|
|
347
|
+
"match. For details, refer to the installation guidelines: "
|
|
348
|
+
"https://www.mindspore.cn/install")
|
|
354
349
|
from hccl import sys_version as hccl_version
|
|
355
350
|
v = '.'.join(hccl_version.__sys_version__.split('.')[0:2])
|
|
356
351
|
if v not in supported_version:
|
|
357
352
|
attention_warning = True
|
|
358
353
|
logger.warning(f"MindSpore version {mindspore_version} and \"hccl\" wheel package version {v} does not "
|
|
359
|
-
"match,
|
|
360
|
-
|
|
354
|
+
"match. For details, refer to the installation guidelines: "
|
|
355
|
+
"https://www.mindspore.cn/install")
|
|
356
|
+
# DO NOT modify exception type to any other, you DO NOT know what kind of exceptions the te will throw.
|
|
357
|
+
# pylint: disable=broad-except
|
|
358
|
+
except Exception as e:
|
|
361
359
|
logger.error("CheckFailed:", e.args)
|
|
362
|
-
logger.error("MindSpore relies on
|
|
363
|
-
"folder of the Ascend AI software package (Ascend Data Center Solution)
|
|
364
|
-
" they are installed correctly or not,
|
|
360
|
+
logger.error("MindSpore relies on whl packages of \"te\" and \"hccl\" in the \"latest\" "
|
|
361
|
+
"folder of the Ascend AI software package (Ascend Data Center Solution). Please check whether"
|
|
362
|
+
" they are installed correctly or not, refer to the match info on: "
|
|
365
363
|
"https://www.mindspore.cn/install")
|
|
366
364
|
if attention_warning:
|
|
367
365
|
warning_countdown = 3
|
|
@@ -390,9 +388,10 @@ class AscendEnvChecker(EnvChecker):
|
|
|
390
388
|
if Path(self.tbe_path).is_dir():
|
|
391
389
|
os.environ['LD_LIBRARY_PATH'] = self.tbe_path + ":" + os.environ['LD_LIBRARY_PATH']
|
|
392
390
|
else:
|
|
393
|
-
|
|
394
|
-
f"No such directory: {self.tbe_path}
|
|
391
|
+
logger.error(
|
|
392
|
+
f"No such directory: {self.tbe_path}. Please check if Ascend AI software package (Ascend Data "
|
|
395
393
|
"Center Solution) is installed correctly.")
|
|
394
|
+
return
|
|
396
395
|
|
|
397
396
|
# check te version after set te env
|
|
398
397
|
self.check_deps_version()
|
|
@@ -408,57 +407,63 @@ class AscendEnvChecker(EnvChecker):
|
|
|
408
407
|
|
|
409
408
|
os.environ['TBE_IMPL_PATH'] = self.op_impl_path
|
|
410
409
|
else:
|
|
411
|
-
|
|
412
|
-
f"No such directory: {self.op_impl_path}
|
|
410
|
+
logger.error(
|
|
411
|
+
f"No such directory: {self.op_impl_path}. Please check if Ascend AI software package (Ascend Data "
|
|
413
412
|
"Center Solution) is installed correctly.")
|
|
413
|
+
return
|
|
414
414
|
|
|
415
415
|
if Path(self.cce_path).is_dir():
|
|
416
416
|
os.environ['PATH'] = self.cce_path + ":" + os.environ['PATH']
|
|
417
417
|
else:
|
|
418
|
-
|
|
419
|
-
f"No such directory: {self.cce_path}
|
|
418
|
+
logger.error(
|
|
419
|
+
f"No such directory: {self.cce_path}. Please check if Ascend AI software package (Ascend Data Center "
|
|
420
420
|
"Solution) is installed correctly.")
|
|
421
|
+
return
|
|
421
422
|
|
|
422
423
|
if self.op_path is None:
|
|
423
424
|
pass
|
|
424
425
|
elif Path(self.op_path).is_dir():
|
|
425
426
|
os.environ['ASCEND_OPP_PATH'] = self.op_path
|
|
426
427
|
else:
|
|
427
|
-
|
|
428
|
-
f"No such directory: {self.op_path}
|
|
428
|
+
logger.error(
|
|
429
|
+
f"No such directory: {self.op_path}. Please check if Ascend AI software package (Ascend Data Center "
|
|
429
430
|
"Solution) is installed correctly.")
|
|
431
|
+
return
|
|
430
432
|
|
|
431
433
|
if self.aicpu_path is None:
|
|
432
434
|
pass
|
|
433
435
|
elif Path(self.aicpu_path).is_dir():
|
|
434
436
|
os.environ['ASCEND_AICPU_PATH'] = self.aicpu_path
|
|
435
437
|
else:
|
|
436
|
-
|
|
437
|
-
f"No such directory: {self.aicpu_path}
|
|
438
|
+
logger.error(
|
|
439
|
+
f"No such directory: {self.aicpu_path}. Please check if Ascend AI software package (Ascend Data Center"
|
|
438
440
|
" Solution) is installed correctly.")
|
|
441
|
+
return
|
|
439
442
|
|
|
440
443
|
def _check_env(self):
|
|
441
444
|
"""ascend dependence path check"""
|
|
442
445
|
if self.path is None or self.path_check not in self.path:
|
|
443
|
-
logger.warning("Can not find ccec_compiler(need by mindspore-ascend)
|
|
444
|
-
"PATH
|
|
446
|
+
logger.warning("Can not find ccec_compiler(need by mindspore-ascend). Please check whether the Environment "
|
|
447
|
+
"Variable PATH is set. For details, refer to the installation guidelines: "
|
|
448
|
+
"https://www.mindspore.cn/install")
|
|
445
449
|
|
|
446
450
|
if self.python_path is None or self.python_path_check not in self.python_path:
|
|
447
451
|
logger.warning(
|
|
448
|
-
"Can not find tbe
|
|
449
|
-
"PYTHONPATH
|
|
452
|
+
"Can not find the tbe operator implementation(need by mindspore-ascend). Please check whether the "
|
|
453
|
+
"Environment Variable PYTHONPATH is set. For details, refer to the installation guidelines: "
|
|
450
454
|
"https://www.mindspore.cn/install")
|
|
451
455
|
|
|
452
456
|
if self.ld_lib_path is None or not (self.ld_lib_path_check_fwk in self.ld_lib_path and
|
|
453
457
|
self.ld_lib_path_check_addons in self.ld_lib_path):
|
|
454
|
-
logger.warning("Can not find driver so(need by mindspore-ascend)
|
|
455
|
-
"LD_LIBRARY_PATH
|
|
456
|
-
"https://www.mindspore.cn/install")
|
|
458
|
+
logger.warning("Can not find driver so(need by mindspore-ascend). Please check whether the "
|
|
459
|
+
"Environment Variable LD_LIBRARY_PATH is set. For details, refer to the installation "
|
|
460
|
+
"guidelines: https://www.mindspore.cn/install")
|
|
457
461
|
|
|
458
462
|
if self.ascend_opp_path is None or self.ascend_opp_path_check not in self.ascend_opp_path:
|
|
459
463
|
logger.warning(
|
|
460
|
-
"Can not find opp path (need by mindspore-ascend)
|
|
461
|
-
"
|
|
464
|
+
"Can not find opp path (need by mindspore-ascend). Please check whether the Environment Variable "
|
|
465
|
+
"ASCEND_OPP_PATH is set. For details, refer to the installation guidelines: "
|
|
466
|
+
"https://www.mindspore.cn/install")
|
|
462
467
|
|
|
463
468
|
def _read_version(self, file_path):
|
|
464
469
|
"""get ascend version info"""
|
|
@@ -476,6 +481,7 @@ def check_env(device, _):
|
|
|
476
481
|
"""callback function for checking environment variables"""
|
|
477
482
|
if device.lower() == "ascend":
|
|
478
483
|
env_checker = AscendEnvChecker(None)
|
|
484
|
+
env_checker.check_version()
|
|
479
485
|
elif device.lower() == "gpu":
|
|
480
486
|
env_checker = GPUEnvChecker(None)
|
|
481
487
|
else:
|
|
@@ -515,15 +521,15 @@ def check_version_and_env_config():
|
|
|
515
521
|
import ctypes
|
|
516
522
|
ctypes.cdll.LoadLibrary("libgomp.so.1")
|
|
517
523
|
except OSError:
|
|
518
|
-
logger.warning(
|
|
519
|
-
|
|
520
|
-
|
|
521
|
-
|
|
522
|
-
|
|
523
|
-
|
|
524
|
-
|
|
525
|
-
|
|
526
|
-
|
|
524
|
+
logger.warning("Pre-Load Library libgomp.so.1 failed, which might cause TLS memory allocation failure. If "
|
|
525
|
+
"the failure occurs, please refer to the FAQ for a solution: "
|
|
526
|
+
"https://www.mindspore.cn/docs/en/r2.0/faq/installation.html.")
|
|
527
|
+
if not os.getenv("MS_DEV_CLOSE_VERSION_CHECK") is None:
|
|
528
|
+
return
|
|
529
|
+
MSContext.get_instance().register_check_env_callback(check_env)
|
|
530
|
+
MSContext.get_instance().register_set_env_callback(set_env)
|
|
531
|
+
MSContext.get_instance().set_param(ms_ctx_param.device_target,
|
|
532
|
+
MSContext.get_instance().get_param(ms_ctx_param.device_target))
|
|
527
533
|
|
|
528
534
|
|
|
529
535
|
def _set_pb_env():
|
|
@@ -532,7 +538,7 @@ def _set_pb_env():
|
|
|
532
538
|
logger.info("Current env variable `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=cpp`. "
|
|
533
539
|
"When the checkpoint file is too large, "
|
|
534
540
|
"it may cause memory limit error during load checkpoint file. "
|
|
535
|
-
"This can be solved by
|
|
541
|
+
"This can be solved by setting env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python`.")
|
|
536
542
|
elif os.getenv("PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION") is None:
|
|
537
543
|
logger.info("Setting the env `PROTOCOL_BUFFERS_PYTHON_IMPLEMENTATION=python` to prevent memory overflow "
|
|
538
544
|
"during save or load checkpoint file.")
|
mindspore/run_check/run_check.py
CHANGED
|
@@ -40,7 +40,8 @@ def _check_mul():
|
|
|
40
40
|
input_y = ms.Tensor(np.array([4.0, 5.0, 6.0]), ms.float32)
|
|
41
41
|
mul = ms.ops.Mul()
|
|
42
42
|
mul(input_x, input_y)
|
|
43
|
-
print(
|
|
43
|
+
print("The result of multiplication calculation is correct, MindSpore has been installed on platform "
|
|
44
|
+
f"[{ms.get_context('device_target')}] successfully!")
|
|
44
45
|
|
|
45
46
|
|
|
46
47
|
def run_check():
|
mindspore/scipy/linalg.py
CHANGED
|
@@ -26,7 +26,7 @@ from ..ops.operations.math_ops import Cholesky
|
|
|
26
26
|
from ..ops import functional as F
|
|
27
27
|
from ..ops import operations as P
|
|
28
28
|
|
|
29
|
-
__all__ = ['block_diag', '
|
|
29
|
+
__all__ = ['block_diag', 'inv', 'cho_factor', 'cholesky', 'cho_solve', 'eigh', 'lu_factor', 'lu']
|
|
30
30
|
|
|
31
31
|
|
|
32
32
|
def block_diag(*arrs):
|
|
@@ -56,7 +56,7 @@ def block_diag(*arrs):
|
|
|
56
56
|
ValueError: If there are Tensors with dimensions higher than 2 in all arguments.
|
|
57
57
|
|
|
58
58
|
Supported Platforms:
|
|
59
|
-
``
|
|
59
|
+
``GPU`` ``CPU``
|
|
60
60
|
|
|
61
61
|
Examples:
|
|
62
62
|
>>> import numpy as onp
|
|
@@ -96,110 +96,6 @@ def block_diag(*arrs):
|
|
|
96
96
|
return accum
|
|
97
97
|
|
|
98
98
|
|
|
99
|
-
def solve_triangular(a, b, trans=0, lower=False, unit_diagonal=False,
|
|
100
|
-
overwrite_b=False, debug=None, check_finite=True):
|
|
101
|
-
"""
|
|
102
|
-
Assuming a is a batched triangular matrix, solve the equation
|
|
103
|
-
|
|
104
|
-
.. math::
|
|
105
|
-
a x = b
|
|
106
|
-
|
|
107
|
-
Note:
|
|
108
|
-
- `solve_triangular` is not supported on Windows platform yet.
|
|
109
|
-
- Only `float32`, `float64`, `int32`, `int64` are supported Tensor dtypes. If Tensor with dtype `int32` or
|
|
110
|
-
`int64` is passed, it will be cast to :class:`mstype.float64`.
|
|
111
|
-
- The floating point error will accumulate when the size of input matrix gets larger. Substituting
|
|
112
|
-
result `x` back into :math:`a x = b` would be a way to evaluate the result. If the input shape is large
|
|
113
|
-
enough, using `float64` instead of `float32` is also a way to mitigate the error.
|
|
114
|
-
|
|
115
|
-
Args:
|
|
116
|
-
a (Tensor): A non-singular triangular matrix of shape :math:`(M, M)`.
|
|
117
|
-
b (Tensor): A Tensor of shape :math:`(M,)` or :math:`(M, N)`. Right-hand side matrix in :math:`a x = b`.
|
|
118
|
-
lower (bool, optional): Use only data contained in the lower triangle of `a`. Default: False.
|
|
119
|
-
trans (0, 1, 2, 'N', 'T', 'C', optional): Type of system to solve. Default: 0.
|
|
120
|
-
|
|
121
|
-
======== =========
|
|
122
|
-
trans system
|
|
123
|
-
======== =========
|
|
124
|
-
0 or 'N' a x = b
|
|
125
|
-
1 or 'T' a^T x = b
|
|
126
|
-
2 or 'C' a^H x = b
|
|
127
|
-
======== =========
|
|
128
|
-
unit_diagonal (bool, optional): If True, diagonal elements of :math:`a` are assumed to be 1 and
|
|
129
|
-
will not be referenced. Default: False.
|
|
130
|
-
overwrite_b (bool, optional): Allow overwriting data in :math:`b` (may enhance performance). Default: False.
|
|
131
|
-
debug (None): Not implemented now. Default: None.
|
|
132
|
-
check_finite (bool, optional): Whether to check that the input matrices contain only finite numbers.
|
|
133
|
-
Disabling may give a performance gain, but may result in problems
|
|
134
|
-
(crashes, non-termination) if the inputs do contain infinities or NaNs. Default: True.
|
|
135
|
-
|
|
136
|
-
Returns:
|
|
137
|
-
Tensor of shape :math:`(M,)` or :math:`(M, N)`,
|
|
138
|
-
which is the solution to the system :math:`a x = b`.
|
|
139
|
-
Shape of :math:`x` matches :math:`b`.
|
|
140
|
-
|
|
141
|
-
Raises:
|
|
142
|
-
TypeError: If `a` is not Tensor.
|
|
143
|
-
ValueError: If `a` is not 2 dimension.
|
|
144
|
-
TypeError: If `b` is not Tensor.
|
|
145
|
-
ValueError: If `b` is not 1 or 2 dimension.
|
|
146
|
-
TypeError: If dtype of `a` and `b` are not the same.
|
|
147
|
-
ValueError: If the shape of `a` and `b` are not matched.
|
|
148
|
-
TypeError: If `trans` is not int or str.
|
|
149
|
-
ValueError: If `trans` is not in set {0, 1, 2, 'N', 'T', 'C'}.
|
|
150
|
-
TypeError: If `lower` is not bool.
|
|
151
|
-
TypeError: If `unit_diagonal` is not bool.
|
|
152
|
-
TypeError: If `overwrite_b` is not bool.
|
|
153
|
-
TypeError: If `check_finite` is not bool.
|
|
154
|
-
ValueError: If `debug` is not None.
|
|
155
|
-
ValueError: If `a` is singular matrix.
|
|
156
|
-
|
|
157
|
-
Supported Platforms:
|
|
158
|
-
``CPU`` ``GPU``
|
|
159
|
-
|
|
160
|
-
Examples:
|
|
161
|
-
Solve the lower triangular system :math:`a x = b`, where::
|
|
162
|
-
|
|
163
|
-
[3 0 0 0] [4]
|
|
164
|
-
a = [2 1 0 0] b = [2]
|
|
165
|
-
[1 0 1 0] [4]
|
|
166
|
-
[1 1 1 1] [2]
|
|
167
|
-
|
|
168
|
-
>>> import numpy as onp
|
|
169
|
-
>>> from mindspore.common import Tensor
|
|
170
|
-
>>> import mindspore.numpy as mnp
|
|
171
|
-
>>> from mindspore.scipy.linalg import solve_triangular
|
|
172
|
-
>>> a = Tensor(onp.array([[3, 0, 0, 0], [2, 1, 0, 0], [1, 0, 1, 0], [1, 1, 1, 1]], onp.float64))
|
|
173
|
-
>>> b = Tensor(onp.array([4, 2, 4, 2], onp.float64))
|
|
174
|
-
>>> x = solve_triangular(a, b, lower=True, unit_diagonal=False, trans='N')
|
|
175
|
-
>>> print(x)
|
|
176
|
-
[ 1.33333333 -0.66666667 2.66666667 -1.33333333]
|
|
177
|
-
>>> print(mnp.dot(a, x)) # Check the result
|
|
178
|
-
[4. 2. 4. 2.]
|
|
179
|
-
"""
|
|
180
|
-
func_name = 'solve_triangular'
|
|
181
|
-
_mstype_check(func_name, a, mstype.tensor_type, 'a')
|
|
182
|
-
_mstype_check(func_name, b, mstype.tensor_type, 'b')
|
|
183
|
-
_type_check(func_name, trans, (int, str), 'trans')
|
|
184
|
-
_type_check(func_name, lower, bool, 'lower')
|
|
185
|
-
_type_check(func_name, overwrite_b, bool, 'overwrite_b')
|
|
186
|
-
_type_check(func_name, check_finite, bool, 'check_finite')
|
|
187
|
-
_dtype_check(func_name, a, [mstype.int32, mstype.int64, mstype.float32, mstype.float64], 'a')
|
|
188
|
-
_dtype_check(func_name, b, [mstype.int32, mstype.int64, mstype.float32, mstype.float64], 'b')
|
|
189
|
-
_solve_check(func_name, a, b)
|
|
190
|
-
_value_check(func_name, debug, None, 'debug', op='is', fmt='todo')
|
|
191
|
-
_value_check(func_name, trans, (0, 1, 2, 'N', 'T', 'C'), "trans", "value")
|
|
192
|
-
|
|
193
|
-
if F.dtype(a) in (mstype.int32, mstype.int64):
|
|
194
|
-
a = F.cast(a, mstype.float64)
|
|
195
|
-
b = F.cast(b, mstype.float64)
|
|
196
|
-
if isinstance(trans, int):
|
|
197
|
-
trans_table = ['N', 'T', 'C']
|
|
198
|
-
trans = trans_table[trans]
|
|
199
|
-
solve = SolveTriangular(lower, unit_diagonal, trans)
|
|
200
|
-
return solve(a, b)
|
|
201
|
-
|
|
202
|
-
|
|
203
99
|
def inv(a, overwrite_a=False, check_finite=True):
|
|
204
100
|
"""
|
|
205
101
|
Compute the inverse of a matrix.
|
|
@@ -224,7 +120,7 @@ def inv(a, overwrite_a=False, check_finite=True):
|
|
|
224
120
|
ValueError: If :math:`a` is not square, or not 2D.
|
|
225
121
|
|
|
226
122
|
Supported Platforms:
|
|
227
|
-
``
|
|
123
|
+
``GPU`` ``CPU``
|
|
228
124
|
|
|
229
125
|
Examples:
|
|
230
126
|
>>> import numpy as onp
|
|
@@ -289,7 +185,7 @@ def cho_factor(a, lower=False, overwrite_a=False, check_finite=True):
|
|
|
289
185
|
ValueError: If input a tensor is not a square matrix or it's dims not equal to 2D.
|
|
290
186
|
|
|
291
187
|
Supported Platforms:
|
|
292
|
-
``
|
|
188
|
+
``GPU`` ``CPU``
|
|
293
189
|
|
|
294
190
|
Examples:
|
|
295
191
|
>>> import numpy as onp
|
|
@@ -350,7 +246,7 @@ def cholesky(a, lower=False, overwrite_a=False, check_finite=True):
|
|
|
350
246
|
ValueError: If input a tensor is not a square matrix or it's dims not equal to 2D.
|
|
351
247
|
|
|
352
248
|
Supported Platforms:
|
|
353
|
-
``
|
|
249
|
+
``GPU`` ``CPU``
|
|
354
250
|
|
|
355
251
|
Examples:
|
|
356
252
|
>>> import numpy as onp
|
|
@@ -402,7 +298,7 @@ def cho_solve(c_and_lower, b, overwrite_b=False, check_finite=True):
|
|
|
402
298
|
Tensor, the solution to the system a x = b
|
|
403
299
|
|
|
404
300
|
Supported Platforms:
|
|
405
|
-
``
|
|
301
|
+
``GPU`` ``CPU``
|
|
406
302
|
|
|
407
303
|
Examples:
|
|
408
304
|
>>> import numpy as onp
|
|
@@ -512,7 +408,7 @@ def eigh(a, b=None, lower=True, eigvals_only=False, overwrite_a=False,
|
|
|
512
408
|
ValueError: If `eigvals` is not None.
|
|
513
409
|
|
|
514
410
|
Supported Platforms:
|
|
515
|
-
``
|
|
411
|
+
``GPU`` ``CPU``
|
|
516
412
|
|
|
517
413
|
Examples:
|
|
518
414
|
>>> import numpy as onp
|
|
@@ -603,7 +499,7 @@ def lu_factor(a, overwrite_a=False, check_finite=True):
|
|
|
603
499
|
ValueError: If :math:`a` is not square.
|
|
604
500
|
|
|
605
501
|
Supported Platforms:
|
|
606
|
-
``
|
|
502
|
+
``GPU`` ``CPU``
|
|
607
503
|
|
|
608
504
|
Examples:
|
|
609
505
|
>>> import numpy as onp
|
|
@@ -672,7 +568,7 @@ def lu(a, permute_l=False, overwrite_a=False, check_finite=True):
|
|
|
672
568
|
- Tensor, :math:`(K, N)` upper triangular or trapezoidal matrix.
|
|
673
569
|
|
|
674
570
|
Supported Platforms:
|
|
675
|
-
``
|
|
571
|
+
``GPU`` ``CPU``
|
|
676
572
|
|
|
677
573
|
Examples:
|
|
678
574
|
>>> import numpy as onp
|
|
@@ -750,7 +646,7 @@ def lu_solve(lu_and_piv, b, trans=0, overwrite_b=False, check_finite=True):
|
|
|
750
646
|
Tensor, solution to the system
|
|
751
647
|
|
|
752
648
|
Supported Platforms:
|
|
753
|
-
``
|
|
649
|
+
``GPU`` ``CPU``
|
|
754
650
|
|
|
755
651
|
Examples:
|
|
756
652
|
>>> import numpy as onp
|
mindspore/scipy/ops.py
CHANGED
|
@@ -13,8 +13,8 @@
|
|
|
13
13
|
# limitations under the License.
|
|
14
14
|
# ============================================================================
|
|
15
15
|
"""Operators for scipy submodule"""
|
|
16
|
+
from mindspore import _checkparam as validator
|
|
16
17
|
from ..ops import PrimitiveWithInfer, prim_attr_register, Primitive
|
|
17
|
-
from .._checkparam import Validator as validator
|
|
18
18
|
from ..common import dtype as mstype
|
|
19
19
|
|
|
20
20
|
|
|
@@ -50,7 +50,7 @@ class SolveTriangular(Primitive):
|
|
|
50
50
|
LinAlgError: If :math:`a` is singular
|
|
51
51
|
|
|
52
52
|
Supported Platforms:
|
|
53
|
-
``
|
|
53
|
+
``GPU`` ``CPU``
|
|
54
54
|
|
|
55
55
|
Examples:
|
|
56
56
|
Solve the lower triangular system :math:`a x = b`, where:
|
mindspore/scipy/ops_wrapper.py
CHANGED