Skip to content

Commit

Permalink
small debug
Browse files Browse the repository at this point in the history
  • Loading branch information
Olender committed Jun 28, 2024
1 parent 30e14a4 commit ba3da8d
Show file tree
Hide file tree
Showing 2 changed files with 2 additions and 2 deletions.
2 changes: 1 addition & 1 deletion spyro/io/basicio.py
Original file line number Diff line number Diff line change
Expand Up @@ -91,7 +91,7 @@ def wrapper(*args, **kwargs):
_comm = args[0].comm
for id_shots, shots_in_core in enumerate(shots_per_core_list):
if is_owner(_comm, id_shots):
u, u_r = func(*args, **dict(kwargs, source_num=shots_in_core))
u, u_r = func(*args, **dict(kwargs, source_nums=shots_in_core))
return u, u_r

return wrapper
Expand Down
2 changes: 1 addition & 1 deletion temp_forward_shot.py
Original file line number Diff line number Diff line change
Expand Up @@ -24,7 +24,7 @@ def run_forward(dt):
# spyro however supports both spatial parallelism and "shot" parallelism.
dictionary["parallelism"] = {
"type": "custom", # options: automatic (same number of cores for evey processor) or spatial
"seperate_shots" : True,
"seperate_shots": True,
"shots_per_core": [[0, 1]],
}

Expand Down

0 comments on commit ba3da8d

Please sign in to comment.