Huawei_Technology
/
mindspore-mindinsight
Not watched
1
0
Fork 0
Code
Releases 11 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
Tree: 7a1d5fb0fd
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '7a1d5fb0fd'
${ noResults }
mindspore-mindinsight/tests/ut/debugger/configurations.py

96 lines
2.8 kB
Raw Blame History 

  1. # Copyright 2020 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. """Common configurations for debugger unit testing."""

  16. import json

  17. import os

  18. 

  19. from google.protobuf import json_format

  20. 

  21. from mindinsight.debugger.proto import ms_graph_pb2

  22. from mindinsight.debugger.stream_handler.graph_handler import GraphHandler

  23. 

  24. graph_proto_file = os.path.join(

  25.     os.path.dirname(__file__), '../../utils/resource/graph_pb/lenet.pb'

  26. )

  27. 

  28. 

  29. def get_graph_proto():

  30.     """Get graph proto."""

  31.     with open(graph_proto_file, 'rb') as f:

  32.         content = f.read()

  33. 

  34.     graph = ms_graph_pb2.GraphProto()

  35.     graph.ParseFromString(content)

  36. 

  37.     return graph

  38. 

  39. 

  40. def init_graph_handler():

  41.     """Init graph proto."""

  42.     graph = get_graph_proto()

  43.     graph_handler = GraphHandler()

  44.     graph_handler.put(graph)

  45. 

  46.     return graph_handler

  47. 

  48. 

  49. def mock_tensor_proto():

  50.     """Mock tensor proto."""

  51.     tensor_dict = {

  52.         "node_name":

  53.             "Default/network-WithLossCell/_backbone-LeNet5/relu-ReLU/gradReLU/ReluGradV2-op92",

  54.         "slot": "0"

  55.     }

  56.     tensor_proto = json_format.Parse(json.dumps(tensor_dict), ms_graph_pb2.TensorProto())

  57. 

  58.     return tensor_proto

  59. 

  60. 

  61. def mock_tensor_history():

  62.     """Mock tensor history."""

  63.     tensor_history = {

  64.         "tensor_history": [

  65.             {"name": "Default/TransData-op99:0",

  66.              "full_name": "Default/TransData-op99:0",

  67.              "node_type": "TransData",

  68.              "type": "output",

  69.              "step": 0,

  70.              "dtype": "DT_FLOAT32",

  71.              "shape": [2, 3],

  72.              "has_prev_step": False,

  73.              "value": "click to view"},

  74.             {"name": "Default/args0:0",

  75.              "full_name": "Default/args0:0",

  76.              "node_type": "Parameter",

  77.              "type": "input",

  78.              "step": 0,

  79.              "dtype": "DT_FLOAT32",

  80.              "shape": [2, 3],

  81.              "has_prev_step": False,

  82.              "value": "click to view"}

  83.         ],

  84.         "metadata": {

  85.             "state": "waiting",

  86.             "step": 0,

  87.             "device_name": "0",

  88.             "pos": "0",

  89.             "ip": "127.0.0.1:57492",

  90.             "node_name": "",

  91.             "backend": "Ascend"

  92.         }

  93.     }

  94. 

  95.     return tensor_history