Python stencil Examples

Example #1

File: test_convnet.py Project: rossparks/spartan

def test_convnet(ctx):
    hint = util.divup(64, sqrt(ctx.num_workers))

    images = expr.eager(
        expr.ones((N_IMGS, ) + IMG_SIZE,
                  tile_hint=(N_IMGS, N_COLORS, hint, hint)))

    w1 = expr.eager(
        expr.ones((N_FILTERS, N_COLORS) + FILTER_SIZE, tile_hint=ONE_TILE))

    conv1 = stencil.stencil(images, w1, 2)
    pool1 = stencil.maxpool(conv1)

    w2 = expr.eager(
        expr.ones((N_FILTERS, N_FILTERS) + FILTER_SIZE, tile_hint=ONE_TILE))

    conv2 = stencil.stencil(pool1, w2, 2)
    pool2 = stencil.maxpool(conv2)

    w3 = expr.eager(
        expr.ones((N_FILTERS, N_FILTERS) + FILTER_SIZE, tile_hint=ONE_TILE))
    conv3 = stencil.stencil(pool2, w3, 2)
    pool3 = stencil.maxpool(conv3)

    util.log_info(pool3.shape)

Example #2

File: benchmark_convnet.py Project: EasonLiao/spartan

 def _():
   conv1 = stencil.stencil(images, w1, 2)
   pool1 = stencil.maxpool(conv1)
  
   conv2 = stencil.stencil(pool1, w2, 2)
   pool2 = stencil.maxpool(conv2)
   
   conv3 = stencil.stencil(pool2, w3, 2)
   pool3 = stencil.maxpool(conv3)
   
   expr.force(pool3)

Example #3

  def _():
    conv1 = stencil.stencil(images, w1, 2)
    pool1 = stencil.maxpool(conv1)

    conv2 = stencil.stencil(pool1, w2, 2)
    pool2 = stencil.maxpool(conv2)

    conv3 = stencil.stencil(pool2, w3, 2)
    pool3 = stencil.maxpool(conv3)

    pool3.evaluate()

Example #4

File: benchmark_convnet.py Project: rossparks/spartan

    def _():
        conv1 = stencil.stencil(images, w1, 2)
        pool1 = stencil.maxpool(conv1)

        conv2 = stencil.stencil(pool1, w2, 2)
        pool2 = stencil.maxpool(conv2)

        conv3 = stencil.stencil(pool2, w3, 2)
        pool3 = stencil.maxpool(conv3)

        expr.force(pool3)

Example #5

File: test_convnet.py Project: EasonLiao/spartan

def test_convnet(ctx):
  hint = util.divup(64, sqrt(ctx.num_workers))
   
  images = expr.eager(expr.ones((N_IMGS,) + IMG_SIZE,
                                tile_hint=(N_IMGS, N_COLORS, hint, hint)))
  
  w1 = expr.eager(expr.ones((N_FILTERS, N_COLORS) + FILTER_SIZE,
                            tile_hint=ONE_TILE))
  
  conv1 = stencil.stencil(images, w1, 2)
  pool1 = stencil.maxpool(conv1)
 
  w2 = expr.eager(expr.ones((N_FILTERS, N_FILTERS) + FILTER_SIZE,
                            tile_hint=ONE_TILE))
  
  conv2 = stencil.stencil(pool1, w2, 2)
  pool2 = stencil.maxpool(conv2)
  
  w3 = expr.eager(expr.ones((N_FILTERS, N_FILTERS) + FILTER_SIZE,
                            tile_hint=ONE_TILE))
  conv3 = stencil.stencil(pool2, w3, 2)
  pool3 = stencil.maxpool(conv3)
  
  util.log_info(pool3.shape)

Example #6

File: test_stencil.py Project: rossparks/spartan

def test_stencil(ctx):
  st = time.time()

  IMG_SIZE = int(8 * math.sqrt(ctx.num_workers))
  FILT_SIZE = 8
  N = 8
  F = 32
  
  tile_size = util.divup(IMG_SIZE, math.sqrt(ctx.num_workers))
  
  images = expr.ones((N, 3, IMG_SIZE, IMG_SIZE), 
                     dtype=np.float, 
                     tile_hint=(N, 3, tile_size, tile_size))
  
  filters = expr.ones((F, 3, FILT_SIZE, FILT_SIZE), 
                      dtype=np.float, 
                      tile_hint=ONE_TILE)
  
  result = stencil.stencil(images, filters, 1)
  ed = time.time()
  print ed - st