# Copyright 2021 Huawei Technologies Co., Ltd
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
# ============================================================================
"""
mindspore.run_check
The goal is to provide a convenient API to check if the installation is successful or failed.
"""
from importlib import import_module
import numpy as np
try:
ms = import_module("mindspore")
except ModuleNotFoundError:
ms = None
def _check_mul():
"""
Define the mul method.
"""
input_x = ms.Tensor(np.array([1.0, 2.0, 3.0]), ms.float32)
input_y = ms.Tensor(np.array([4.0, 5.0, 6.0]), ms.float32)
mul = ms.ops.Mul()
mul(input_x, input_y)
print(f"The result of multiplication calculation is correct, MindSpore has been installed successfully!")
def _check_install():
"""
Define the check install method.
Print MindSpore version.
"""
print(f"MindSpore version:", ms.__version__)
[docs]def run_check():
"""
Provide a convenient API to check if the installation is successful or failed.
Examples:
>>> import mindspore
>>> mindspore.run_check()
MindSpore version: xxx
The result of multiplication calculation is correct, MindSpore has been installed successfully!
"""
try:
_check_install()
_check_mul()
# pylint: disable=broad-except
except Exception as e:
print("MindSpore running check failed.")
print(str(e))