diff --git a/src/atrip/Atrip.cxx b/src/atrip/Atrip.cxx index 8265c0e..6c3045f 100644 --- a/src/atrip/Atrip.cxx +++ b/src/atrip/Atrip.cxx @@ -202,7 +202,7 @@ Atrip::Output Atrip::run(Atrip::Input const& in) { _CHECK_CUDA_SUCCESS("Zijk", cuMemAlloc(&Zijk, sizeof(F) * No * No * No)); #else - std::vector &Tai = _Tai, &epsi = _epsi, &epsa = _epsa; + DataPtr Tai = _Tai.data(), epsi = _epsi.data(), epsa = _epsa.data(); Zijk = (DataFieldType*)malloc(No*No*No * sizeof(DataFieldType)); Tijk = (DataFieldType*)malloc(No*No*No * sizeof(DataFieldType)); #endif @@ -693,7 +693,7 @@ Atrip::Output Atrip::run(Atrip::Input const& in) { (DataFieldType*)Tai, #else singlesContribution(No, Nv, abc[0], abc[1], abc[2], - Tai.data(), + Tai, #endif (DataFieldType*)abhh.unwrapSlice(Slice::AB, abc), @@ -733,18 +733,30 @@ Atrip::Output Atrip::run(Atrip::Input const& in) { 1, 1, // for cuda _epsabc, No, +#if defined(HAVE_CUDA) (DataFieldType*)epsi, (DataFieldType*)Tijk, (DataFieldType*)Zijk, +#else + epsi, + Tijk, + Zijk, +#endif tupleEnergy); } else { ACC_FUNCALL(getEnergySame>, 1, 1, // for cuda _epsabc, No, +#if defined(HAVE_CUDA) (DataFieldType*)epsi, (DataFieldType*)Tijk, (DataFieldType*)Zijk, +#else + epsi, + Tijk, + Zijk, +#endif tupleEnergy); })