From 07136b8b3aa8b08bf832790cda9b480aac81efec Mon Sep 17 00:00:00 2001
From: Giovanni La Mura <giovanni.lamura@inaf.it>
Date: Thu, 26 Sep 2024 15:20:08 +0200
Subject: [PATCH] Add compilation options to use OMP and offload

---
 src/libnptm/tra_subs.cpp | 12 ++++++++++++
 1 file changed, 12 insertions(+)

diff --git a/src/libnptm/tra_subs.cpp b/src/libnptm/tra_subs.cpp
index 724c8211..8d04f447 100644
--- a/src/libnptm/tra_subs.cpp
+++ b/src/libnptm/tra_subs.cpp
@@ -45,6 +45,18 @@
 #include "../include/tra_subs.h"
 #endif
 
+#ifdef USE_NVTX
+#include <nvtx3/nvToolsExt.h>
+#endif
+
+#ifdef _OPENMP
+#include <omp.h>
+#endif
+
+#ifdef USE_TARGET_OFFLOAD
+#pragma omp requires unified_shared_memory
+#endif
+
 void camp(dcomplex *ac, dcomplex **am0m, dcomplex *ws, CIL *cil) {
   for (int j = 0; j < cil->nlemt; j++) {
     for (int i = 0; i < cil->nlemt; i++) {
-- 
GitLab