From e0f6dde402d3d0ac998c65178c047963ce2834ed Mon Sep 17 00:00:00 2001
From: Liwei Ji <jiliwei.phys@gmail.com>
Date: Mon, 19 Aug 2024 12:07:40 -0400
Subject: [PATCH] PunctureTracker: assert omp_in_parallel only when not using
 AMREX_USE_GPU

---
 PunctureTracker/src/puncture_tracker.cxx | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/PunctureTracker/src/puncture_tracker.cxx b/PunctureTracker/src/puncture_tracker.cxx
index c87c35a0..6825579c 100644
--- a/PunctureTracker/src/puncture_tracker.cxx
+++ b/PunctureTracker/src/puncture_tracker.cxx
@@ -5,8 +5,6 @@
 #include <cctk_Parameters.h>
 #include <util_Table.h>
 
-#include <omp.h>
-
 #include <array>
 #include <cassert>
 #include <cmath>
@@ -103,7 +101,10 @@ extern "C" void PunctureTracker_Track(CCTK_ARGUMENTS) {
   DECLARE_CCTK_ARGUMENTS_PunctureTracker_Track;
   DECLARE_CCTK_PARAMETERS;
 
+#ifndef AMREX_USE_GPU
   assert(!omp_in_parallel());
+#endif
+
   // we can remove this segment when global mode works
   if (cctk_iteration == previous_iteration) {
     return;