Implement omp_get_mapped_ptr

Author: alkino

coreneuron/utils/offload.hpp

@@ -25,12 +25,24 @@
 #include <cstddef>
 
 namespace coreneuron {
+#define cnrn_target_deviceptr(h_ptr)  [&]() { std::cout << #h_ptr << std::endl; return cnrn_target_deviceptr2(h_ptr);}()
+
 template <typename T>
-T* cnrn_target_deviceptr(const T* h_ptr) {
+T* cnrn_target_deviceptr2(T* h_ptr) {
 #if defined(CORENEURON_ENABLE_GPU) && !defined(CORENEURON_PREFER_OPENMP_OFFLOAD) && defined(_OPENACC)
-    return static_cast<T*>(acc_deviceptr(const_cast<T*>(h_ptr)));
+    return static_cast<T*>(acc_deviceptr(h_ptr));
 #elif defined(CORENEURON_ENABLE_GPU) && defined(CORENEURON_PREFER_OPENMP_OFFLOAD) && defined(_OPENMP)
-    return static_cast<T*>(omp_get_mapped_ptr(const_cast<T*>(h_ptr), omp_get_default_device()));
+    if (omp_get_mapped_ptr(h_ptr, omp_get_default_device()) == nullptr) {
+        throw std::runtime_error("here");
+    }
+    T *d_ptr = nullptr;
+
+    nrn_pragma_omp(target data use_device_ptr(h_ptr))
+    {
+        d_ptr = h_ptr;
+    }
+
+    return d_ptr;
 #else
     throw std::runtime_error("cnrn_target_deviceptr() not implemented without OpenACC/OpenMP and gpu build");
 #endif