d19d3fcccc2c30dc3dbc6153a5875ba81cfbff37,nn_dataflow/Scheduling.py,Scheduling,schedule_search_per_node,#Scheduling#Any#Any#Any#,186

Before Change

                for t in r:
                    yield t

        if options.nprocesses > 1:
            pool = Pool(processes=options.nprocesses)
            apply_func = pool.apply_async
            retrieve_func = retrieve_result()
        else:
            pool = None
            apply_func = apply
            retrieve_func = retrieve_result_st()

        // Partitioned layer.
        p_layer, p_batch_size, p_occ = part.part_layer(self.layer,
                                                       self.batch_size)

        // Mapping strategy.

After Change

            for lbs in LoopBlocking.gen_loopblocking(
                    nested_loop_desc, resource, self.cost, p_occ, options):

                if lbs.is_valid():
                    top_lbs_list.append(lbs)

        self.pernode_sched_cache[cache_key] = top_lbs_list

        return top_lbs_list

Instances