23 lines
437 B
Python
Executable File
23 lines
437 B
Python
Executable File
#!/usr/bin/python
|
|
"""
|
|
demo python script of rabit
|
|
"""
|
|
import os
|
|
import sys
|
|
# add path to wrapper
|
|
sys.path.append(os.path.dirname(__file__) + '/../wrapper')
|
|
import rabit
|
|
|
|
rabit.init()
|
|
n = 3
|
|
rank = rabit.get_rank()
|
|
s = None
|
|
if rank == 0:
|
|
s = {'hello world':100, 2:3}
|
|
print '@node[%d] before-broadcast: s=\"%s\"' % (rank, str(s))
|
|
s = rabit.broadcast(s, 0)
|
|
|
|
print '@node[%d] after-broadcast: s=\"%s\"' % (rank, str(s))
|
|
rabit.finalize()
|
|
|