bug fix

Nek5000 · Sep 7, 2024 · 3fb824e · 3fb824e
1 parent d2486f0
commit 3fb824e
Showing 1 changed file with 5 additions and 5 deletions.
diff --git a/src/nrs/plugins/RANSktauBuo.cpp b/src/nrs/plugins/RANSktauBuo.cpp
@@ -136,32 +136,32 @@ void RANSbuo::updateSourceTerms()
   mesh_t *mesh = nrs->meshV;
   cds_t *cds = nrs->cds;
 
-  occa::memory o_Tgrad = platform->o_memPool.reserve<dfloat>(nrs->cds->fieldOffset[kFieldIndex]);
+  occa::memory o_Tgrad = platform->o_memPool.reserve<dfloat>(nrs->NVfields * nrs->fieldOffset);
 
   nrs->gradientVolumeKernel(mesh->Nelements,
                             mesh->o_vgeo,
                             mesh->o_D,
-                            nrs->cds->fieldOffset[kFieldIndex],
+                            nrs->fieldOffset,
                             o_T,
                             o_Tgrad);
 
   oogs::startFinish(o_Tgrad,
                     nrs->NVfields,
-                    nrs->cds->fieldOffset[kFieldIndex],
+                    nrs->fieldOffset,
                     ogsDfloat,
                     ogsAdd,
                     nrs->gsh);
 
   platform->linAlg->axmyVector(mesh->Nlocal,
-			       nrs->cds->fieldOffset[kFieldIndex],
+			       nrs->fieldOffset,
 			       0,
 			       1.0,
 			       mesh->o_invLMM,
 			       o_Tgrad);
 
 
   computeKernel(mesh->Nelements * mesh->Np,
-                nrs->cds->fieldOffset[kFieldIndex],
+                nrs->fieldOffset,
                 rho,
 								Ri,
 								o_gvec,