-
Notifications
You must be signed in to change notification settings - Fork 0
/
server.py
69 lines (56 loc) · 1.25 KB
/
server.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
import argparse
from eden.block import Block
from eden.hosting import host_block
parser = argparse.ArgumentParser()
parser.add_argument(
"-ug",
"--use-gpu",
help="set to True if you want to check if eden is on a GPU runtime",
required=False,
default=False,
type=bool,
)
parser.add_argument(
"-n",
"--num-workers",
help="maximum number of workers to be run in parallel",
required=False,
default=1,
type=int,
)
parser.add_argument(
"-l",
"--logfile",
help="filename of log file",
required=False,
type=str,
default=None,
)
parser.add_argument(
"-p", "--port", help="localhost port", required=False, type=int, default=5656
)
args = parser.parse_args()
eden_block = Block()
my_args = {
"name": "abraham",
"number": 12345,
}
@eden_block.run(args=my_args)
def do_something(config):
device = "cpu"
if args.use_gpu:
device = config.gpu
print(f"on device:{device}")
## your stuff goes here
return {
"message": f"hello {config['name']}",
"number": config["number"],
"device": device,
}
host_block(
block=eden_block,
port=args.port,
logfile=args.logfile,
requires_gpu=args.use_gpu,
max_num_workers=args.num_workers,
)