Skip to content

Commit a4c3f7a

Browse files
committed
make oversubscribe available to the user
1 parent ade1e87 commit a4c3f7a

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

pylammpsmpi/lammps_wrapper.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -23,17 +23,19 @@ class LammpsLibrary:
2323
"""
2424

2525
def __init__(
26-
self, cores=1, working_directory=".", client=None, mode="local", cmdargs=None
26+
self, cores=1, oversubscribe=False, working_directory=".", client=None, mode="local", cmdargs=None
2727
):
2828
self.cores = cores
2929
self.working_directory = working_directory
30+
self.oversubscribe = oversubscribe
3031
self.client = client
3132
self.mode = mode
3233

3334
if self.mode == "dask":
3435
fut = self.client.submit(
3536
LammpsBase,
3637
cores=self.cores,
38+
oversubscribe=self.oversubscribe,
3739
working_directory=self.working_directory,
3840
cmdargs=cmdargs,
3941
actor=True,
@@ -46,6 +48,7 @@ def __init__(
4648
elif self.mode == "local":
4749
self.lmp = LammpsBase(
4850
cores=self.cores,
51+
oversubscribe=self.oversubscribe,
4952
working_directory=self.working_directory,
5053
cmdargs=cmdargs,
5154
)

0 commit comments

Comments
 (0)