From 9d6d34c809d8eb60c639a645bb23a6dbb2bb6a6c Mon Sep 17 00:00:00 2001
From: RaulPPealez <raulppelaez@gmail.com>
Date: Wed, 15 May 2024 14:39:45 +0200
Subject: [PATCH] Fix return

---
 platforms/cuda/src/CudaTorchKernels.cpp | 4 +---
 1 file changed, 1 insertion(+), 3 deletions(-)

diff --git a/platforms/cuda/src/CudaTorchKernels.cpp b/platforms/cuda/src/CudaTorchKernels.cpp
index d59e916..56559f6 100644
--- 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) {