forked from OSSInnovation/mindspore
optimize summary support cpu
This commit is contained in:
parent
ed77c761ec
commit
0090682ae7
|
@ -14,6 +14,7 @@
|
|||
# ============================================================================
|
||||
"""Summary cpu st."""
|
||||
import os
|
||||
import platform
|
||||
import tempfile
|
||||
|
||||
import numpy as np
|
||||
|
@ -58,22 +59,23 @@ def train_summary_record(test_writer, steps):
|
|||
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…
Reference in New Issue