diff --git a/src/libnptm/tra_subs.cpp b/src/libnptm/tra_subs.cpp
index 724c8211cb0ebb59d719efd452e431ff68e779fb..8d04f4479c86d72407a7f3b836864425e3d0abdd 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++) {