Fix return

openmm · May 15, 2024 · 9d6d34c · 9d6d34c
1 parent 277c90d
commit 9d6d34c
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/platforms/cuda/src/CudaTorchKernels.cpp b/platforms/cuda/src/CudaTorchKernels.cpp
@@ -52,9 +52,7 @@ using namespace std;
  }
 
 static map<string, double>& extractEnergyParameterDerivatives(CudaContext& context) {
- //CudaPlatform::PlatformData* data = reinterpret_cast<CudaPlatform::PlatformData*>(context.getPlatformData());
- //return *data->energyParameterDerivatives;
- context.getEnergyParamDerivWorkspace();
+ return context.getEnergyParamDerivWorkspace();
 }
 
 CudaCalcTorchForceKernel::CudaCalcTorchForceKernel(string name, const Platform& platform, CudaContext& cu) : CalcTorchForceKernel(name, platform), hasInitializedKernel(false), cu(cu) {