Support gpu

Author: petrhosek

libc/src/time/gpu/CMakeLists.txt

@@ -44,3 +44,15 @@ add_entrypoint_object(
     libc.hdr.types.struct_timespec
     .time_utils
 )
+
+add_entrypoint_object(
+  timespec_get
+  SRCS
+    timespec_get.cpp
+  HDRS
+    ../timespec_get.h
+  DEPENDS
+    libc.hdr.time_macros
+    libc.hdr.types.struct_timespec
+    .time_utils
+)

libc/src/time/gpu/timespec_get.cpp

@@ -0,0 +1,29 @@
+//===-- Implementation of timespec_get for gpu ----------------------------===//
+//
+// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
+// See https://llvm.org/LICENSE.txt for license information.
+// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
+//
+//===----------------------------------------------------------------------===//
+
+#include "src/time/timespec_get.h"
+#include "hdr/time_macros.h"
+#include "src/__support/common.h"
+#include "src/__support/macros/config.h"
+
+namespace LIBC_NAMESPACE_DECL {
+
+LLVM_LIBC_FUNCTION(int, timespec_get, (struct timespec * ts, int base)) {
+  if (base != TIME_MONOTONIC || !ts)
+    return 0;
+
+  uint64_t ns_per_tick = TICKS_PER_SEC / GPU_CLOCKS_PER_SEC;
+  uint64_t ticks = gpu::fixed_frequency_clock();
+
+  ts->tv_nsec = (ticks * ns_per_tick) % TICKS_PER_SEC;
+  ts->tv_sec = (ticks * ns_per_tick) / TICKS_PER_SEC;
+
+  return base;
+}
+
+} // namespace LIBC_NAMESPACE_DECL

libc/test/src/time/timespec_get_test.cpp

@@ -13,17 +13,18 @@
 #include "test/UnitTest/Test.h"
 
 TEST(LlvmLibcTimespecGet, Utc) {
-#ifndef LIBC_TARGET_ARCH_IS_GPU
   timespec ts;
   int result;
   result = LIBC_NAMESPACE::timespec_get(&ts, TIME_UTC);
+#ifdef LIBC_TARGET_ARCH_IS_GPU
+  ASSERT_EQ(result, 0);
+#else
   ASSERT_EQ(result, TIME_UTC);
   ASSERT_GT(ts.tv_sec, time_t(0));
 #endif
 }
 
 TEST(LlvmLibcTimespecGet, Monotonic) {
-#ifndef LIBC_TARGET_ARCH_IS_GPU
   timespec ts1, ts2;
   int result;
   result = LIBC_NAMESPACE::timespec_get(&ts1, TIME_MONOTONIC);
@@ -35,14 +36,11 @@ TEST(LlvmLibcTimespecGet, Monotonic) {
   if (ts2.tv_sec == ts1.tv_sec) {
     ASSERT_GE(ts2.tv_nsec, ts1.tv_nsec);
   }
-#endif
 }
 
 TEST(LlvmLibcTimespecGet, Unknown) {
-#ifndef LIBC_TARGET_ARCH_IS_GPU
   timespec ts;
   int result;
   result = LIBC_NAMESPACE::timespec_get(&ts, 0);
   ASSERT_EQ(result, 0);
-#endif
 }