Clone the module on TorchForceImpl initialization to avoid several

contexts sharing the same model
openmm · Sep 4, 2023 · afb1d8c · afb1d8c
1 parent e9f2ae2
commit afb1d8c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/openmmapi/src/TorchForceImpl.cpp b/openmmapi/src/TorchForceImpl.cpp
@@ -46,7 +46,7 @@ TorchForceImpl::~TorchForceImpl() {
 }
 
 void TorchForceImpl::initialize(ContextImpl& context) {
- auto module = owner.getModule();
+ auto module = owner.getModule().clone();
  // Create the kernel.
  kernel = context.getPlatform().createKernel(CalcTorchForceKernel::Name(), context);
  kernel.getAs<CalcTorchForceKernel>().initialize(context.getSystem(), owner, module);