1 files changed, 6 insertions, 3 deletions

diff --git a/preemption_logger.cu b/preemption_logger.cu
index d93600e..89b348a 100644
--- a/preemption_logger.cu
+++ b/preemption_logger.cu
@@ -1,4 +1,4 @@
-/* Copyright 2023 Joshua Bakita
+/* Copyright 2024 Joshua Bakita
  * Tool that tracks preemptions as discontinuities in GPU time and logs them.
  *
  * The logged intervals are output to standard out as a CSV, with each row
@@ -92,8 +92,11 @@ int main(int argc, char **argv) {
         if (!skip_conversion) {
                 // Note that this appears to almost always be 1.0 on recent GPUs
                 d2h_scale = GetGPUTimerScale(0);
-                // The following uses a subprocess for clock synchronization. Sketch?
-                GetHostDeviceTimeOffset(0, &host_s, &dev_ns);
+                InternalReadGPUNanoseconds(0, &host_s, &dev_ns);
+                if (host_s == 0 && !dev_ns) {
+                        fprintf(stderr, "Unable to synchronize time with the GPU. Aborting...\n");
+                        return 1;
+                }
         }
 
         // Initialize GPU and CPU memory to store the intervals of execution