utils/pyscop/pyscop.py

   1 import json
   2 from isl import *
   3
   4 class Scop:
   5   def __init__(self, filename):
   6     f = open(filename, 'r')
   7     self.json = json.load(f)
   8     return
   9
  10   def __str__(self):
  11     return json.dumps(self.json, indent=2)
  12
  13   def __repr__(self):
  14     return str(self)
  15
  16   @property
  17   def statements(self):
  18     return self.json['statements']
  19
  20 class Transforms:
  21   """
  22   Create a map that interchanges two dimensions 'A' and 'B'
  23
  24   numberDimensions: The overall number of dimensions
  25   dimensionA: The dimension of dimension 'A'
  26   dimensionB: The dimension of dimension 'B'
  27
  28   getInterchange(2, 0, 1):
  29   {[d0, d1] -> [d1, d0]}
  30   """
  31   @staticmethod
  32   def getInterchange(numberDimensions, dimensionA, dimensionB):
  33
  34     dims = ['d' + str(i) for i in range(numberDimensions)]
  35     dimString = ",".join(dims)
  36
  37     changedDims = dims
  38     first = dims[dimensionA]
  39     second = dims[dimensionB]
  40     changedDims[dimensionA] = second
  41     changedDims[dimensionB] = first
  42     changedDimString = ",".join(changedDims)
  43
  44     return Map("{[%s] -> [%s]}" % (dimString, changedDimString))
  45
  46   """
  47   Create a map that strip mines one dimension
  48
  49   numberDimensions: The overall number of dimensions
  50   stripMineDim: The dimension to strip mine
  51   factor: The strip mining factor
  52
  53   getStripMine(2, 1, 64):
  54   {[d0, d1] -> [d0, o, d1] : o % 64 = 0 and o <= d1 <= d1 + 63}
  55   """
  56   @staticmethod
  57   def getStripMine(numberDimensions, stripMineDim, factor):
  58
  59     dims = ['d' + str(i) for i in range(numberDimensions)]
  60     dimString = ",".join(dims)
  61
  62     changedDims = dims
  63     smd = dims[stripMineDim]
  64     changedDims[stripMineDim] = "o,%s" % smd
  65     changedDimString = ",".join(changedDims)
  66     string = "{[%s] -> [%s]: o %% %i = 0 and o <= %s <= o + %i}" % \
  67           (dimString, changedDimString, factor, smd, factor - 1)
  68     return Map(string)