happyz
/
propagation_gan
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

allow irregular number of cpus

This commit is contained in:
HappyZ 2019-05-23 20:47:00 -05:00
parent 3c42d1f96d
commit 1d3eed389b
1 changed files with 3 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
syngen_floorplan_multipath.py
View File

@ -114,7 +114,8 @@ def generateData(floormap, tx_locs, args):
for j in range(0, rx_locs.shape[0], args.procnum):
procs = []
results = Manager().dict()
for kk in range(args.procnum):
actual_num = min(args.procnum, rx_locs.shape[0] - j)
for kk in range(actual_num):
rx_loc = SpaceBlock(rx_locs[j + kk, 0], rx_locs[j + kk, 1])
proc = Process(
target=log_gamma_floorplan_multi,
@ -124,7 +125,7 @@ def generateData(floormap, tx_locs, args):
procs.append(proc)
for proc in procs:
proc.join()
rss_vec.extend([results[kk] for kk in range(args.procnum)])
rss_vec.extend([results[kk] for kk in range(actual_num)])
# rx_loc = SpaceBlock(rx_locs[j, 0], rx_locs[j, 1])
# rss, multipaths = log_gamma_floorplan(