Ejemplo n.º 1
0
def benchmark_jacobi(ctx, timer):
  global base, ITERATION
  util.log_warn('util.log_warn: %s', ctx.num_workers)

  A, b = jacobi.jacobi_init(base * ctx.num_workers)
  A, b = A.evaluate(), b.evaluate()

  start = time.time()

  result = jacobi.jacobi_method(A, b, ITERATION).glom()

  cost = time.time() - start

  util.log_info('\nresult =\n%s', result)
  util.log_warn('time cost: %s s', cost)
  util.log_warn('cost per iteration: %s s\n', cost / ITERATION)
Ejemplo n.º 2
0
 def test_jacobi(self):
   global base
   A, b = jacobi.jacobi_init(base * blob_ctx.get().num_workers)
   jacobi.jacobi_method(A, b, 10).glom()