Python rotate Examples

Example #1

 def _rotate_tensor_block(buf):
     ntasks = mpi.pool.size
     tasks = list(range(ntasks))
     tasks = tasks[rank:] + tasks[:rank]
     for task in tasks:
         if task != rank:
             buf = mpi.rotate(buf)
         yield task, buf

Example #2

File: ccsd.py Project: sunqm/mpi4pyscf

 def _rotate_tensor_block(buf):
     ntasks = mpi.pool.size
     tasks = list(range(ntasks))
     tasks = tasks[rank:] + tasks[:rank]
     for task in tasks:
         if task != rank:
             buf = mpi.rotate(buf)
         yield task, buf

Example #3

 def rotate():
     buf_prefetch[0] = mpi.rotate(buf, blocking=False)

Example #4

File: ccsd.py Project: sunqm/mpi4pyscf

 def rotate():
     buf_prefetch[0] = mpi.rotate(buf, blocking=False)