c3826b3955bd3356cb2531821210286fbe3d4f92,nn_dataflow/tests/unit_test/test_nn_dataflow_scheme.py,TestNNDataflowScheme,test_segment_time_list,#TestNNDataflowScheme#,274

Before Change

        dtfl = NNDataflowScheme(self.network, self.input_layout)
        dtfl["c1"] = self.c1res
        dtfl["p1"] = self.p1res
        dtfl["p2"] = SchedulingResult(
            dict_loop=self.p2res.dict_loop, dict_part=self.p2res.dict_part,
            ofmap_layout=self.p2res.ofmap_layout, sched_seq=(1, 0, 0))
        self.assertListEqual(dtfl.segment_time_list(), [205, 5])

    def test_stats_active_node_pes(self):
        """ Per-layer stats: active node PEs. """

After Change

        dtfl = NNDataflowScheme(self.network, self.input_layout)
        dtfl["c1"] = self.c1res
        dtfl["p1"] = self.p1res
        dtfl["p2"] = self.p2res._replace(sched_seq=(1, 0, 0))
        self.assertListEqual(dtfl.segment_time_list(), [205, 5])

    def test_stats_active_node_pes(self):
        """ Per-layer stats: active node PEs. """

Instances