Browse Source

Modify the minddata module logic to make it more reasonable

tags/v1.1.0
zhangyihui 5 years ago
parent
commit
799385a735
2 changed files with 5 additions and 4 deletions
  1. +4
    -3
      mindinsight/profiler/analyser/minddata_analyser.py
  2. +1
    -1
      tests/st/func/profiler/test_minddata_analyse.py

+ 4
- 3
mindinsight/profiler/analyser/minddata_analyser.py View File

@@ -198,7 +198,7 @@ class MinddataAnalyser(BaseAnalyser):
if get_next_queue_info and device_queue_info:
result = {"data_process": {"status": "normal"},
"device_queue_op": {"status": "normal"},
"tdt": {"status": "normal"},
"data_transmission": {"status": "normal"},
"get_next": {"status": "normal"}}
get_next_queue_empty_count = get_next_queue_info.get(
@@ -220,13 +220,14 @@ class MinddataAnalyser(BaseAnalyser):
"full_batch_count": device_queue_full_count,
"total_batch": device_queue_info.get("size")}}
if get_next_queue_empty_count:
# Adapt to the case that the first step data in the GPU is always empty
if get_next_queue_empty_count > 1:
if device_queue_empty_count > device_queue_info.get("size", 0)*\
MinddataAnalyser.DEVICE_QUEUE_EMPTY_WARNING_THRESHOLD:
result["data_process"]["status"] = "warning"
elif device_queue_empty_count < device_queue_info.get("size", 0)*\
MinddataAnalyser.DEVICE_QUEUE_NOT_EMPTY_THRESHOLD:
result["tdt"]["status"] = "warning"
result["data_transmission"]["status"] = "warning"
result["device_queue_op"]["status"] = "warning"
elif device_queue_info and not get_next_queue_info:


+ 1
- 1
tests/st/func/profiler/test_minddata_analyse.py View File

@@ -55,7 +55,7 @@ class TestMinddataAnalyser:
"device_queue_op": {"status": "normal"},
"get_next": {"status": "normal"},
"get_next_queue_info": {"summary": {"empty_batch_count": 0, "total_batch": 3}},
"tdt": {"status": "normal"}
"data_transmission": {"status": "normal"}
}

get_next_queue_info, _ = self._analyser.analyse_get_next_info(info_type="queue")


Loading…
Cancel
Save