Skip to content
Snippets Groups Projects
utils.py 859 B
Newer Older
xuanyoya's avatar
xuanyoya committed
import loop_enum as le
import buffer_enum as be

def print_loop_nest(point):
    loop_orders = zip(*point.loop_orders)
    loop_blockings = zip(*point.loop_blockings)
    loop_partitionings = zip(*point.loop_partitionings)
    para_dims = point.para_loop_dim
    num_level = len(loop_orders)
    order_lists = []
    for level in xrange(num_level):
        order_list = [None] * le.NUM
        for order in xrange(le.NUM):
            if loop_blockings[level][order] != 1 or loop_partitionings[level][order] != 1 :
                order_list[loop_orders[level][order]] = (le.table[order], 
                                                        loop_blockings[level][order],
                                                        loop_partitionings[level][order]) 
                
        order_lists.append(order_list) 

    print order_lists, para_dims