optimize summary support cpu

tests/st/summary/test_cpu_summary.py

@@ -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:
+@pytest.mark.level0
-    """Test cpu summary."""
+@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
+    with tempfile.TemporaryDirectory() as tmp_dir:
-    @pytest.mark.platform_x86_cpu_training
+        steps = 2
-    @pytest.mark.env_onecard
+        with SummaryRecord(tmp_dir) as test_writer:
-    def test_summary_step2_summary_record1(self):
+            train_summary_record(test_writer, steps=steps)
-        """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)
 
-                file_name = os.path.realpath(test_writer.full_file_name)
+            file_name = os.path.realpath(test_writer.full_file_name)
-            with SummaryReader(file_name) as summary_writer:
+        with SummaryReader(file_name) as summary_writer:
-                for _ in range(steps):
+            for _ in range(steps):
-                    event = summary_writer.read_event()
+                event = summary_writer.read_event()
-                    tags = set(value.tag for value in event.summary.value)
+                tags = set(value.tag for value in event.summary.value)
-                    assert tags == {'tensor', 'histogram', 'scalar', 'image'}
+                assert tags == {'tensor', 'histogram', 'scalar', 'image'}