Browse Source

!1923 optimize summary support cpu code

Merge pull request !1923 from wenkai/wk1_cpu_summary_opt_0609
tags/v0.5.0-beta
mindspore-ci-bot Gitee 5 years ago
parent
commit
17c3148ff8
1 changed files with 19 additions and 17 deletions
  1. +19
    -17
      tests/st/summary/test_cpu_summary.py

+ 19
- 17
tests/st/summary/test_cpu_summary.py View File

@@ -14,6 +14,7 @@
# ============================================================================ # ============================================================================
"""Summary cpu st.""" """Summary cpu st."""
import os import os
import platform
import tempfile import tempfile


import numpy as np import numpy as np
@@ -58,22 +59,23 @@ def train_summary_record(test_writer, steps):
return out_me_dict return out_me_dict




class TestCpuSummary:
"""Test cpu summary."""
@pytest.mark.level0
@pytest.mark.platform_x86_cpu
@pytest.mark.env_onecard
def test_summary_step2_summary_record1():
"""Test record 10 step summary."""
if platform.system() == "Windows":
# Summary does not support windows currently.
return


@pytest.mark.level0
@pytest.mark.platform_x86_cpu_training
@pytest.mark.env_onecard
def test_summary_step2_summary_record1(self):
"""Test record 10 step summary."""
with tempfile.TemporaryDirectory() as tmp_dir:
steps = 2
with SummaryRecord(tmp_dir) as test_writer:
train_summary_record(test_writer, steps=steps)
with tempfile.TemporaryDirectory() as tmp_dir:
steps = 2
with SummaryRecord(tmp_dir) as test_writer:
train_summary_record(test_writer, steps=steps)


file_name = os.path.realpath(test_writer.full_file_name)
with SummaryReader(file_name) as summary_writer:
for _ in range(steps):
event = summary_writer.read_event()
tags = set(value.tag for value in event.summary.value)
assert tags == {'tensor', 'histogram', 'scalar', 'image'}
file_name = os.path.realpath(test_writer.full_file_name)
with SummaryReader(file_name) as summary_writer:
for _ in range(steps):
event = summary_writer.read_event()
tags = set(value.tag for value in event.summary.value)
assert tags == {'tensor', 'histogram', 'scalar', 'image'}

Loading…
Cancel
Save