[libc-commits] [libc] [libc] Partially implement 'rand' for the GPU (PR #66167)

Joseph Huber via libc-commits libc-commits at lists.llvm.org
Mon Sep 25 09:03:51 PDT 2023


================
@@ -9,11 +9,33 @@
 #ifndef LLVM_LIBC_SRC_STDLIB_RAND_UTIL_H
 #define LLVM_LIBC_SRC_STDLIB_RAND_UTIL_H
 
+#include "src/__support/GPU/utils.h"
 #include "src/__support/macros/attributes.h"
 
 namespace __llvm_libc {
 
+#ifdef LIBC_TARGET_ARCH_IS_GPU
+// Implement thread local storage on the GPU using local memory. Each thread
+// gets its slot in the local memory array and is private to the group.
+// TODO: We need to implement the 'thread_local' keyword on the GPU. This is an
+// inefficient and incomplete stand-in until that is done.
+template <typename T> class ThreadLocal {
+private:
+  static constexpr long MAX_THREADS = 1024;
+  [[clang::loader_uninitialized]] static inline gpu::Local<T>
----------------
jhuber6 wrote:

Why should that cause pushback? We should be able to optimize any of this away in the general case. If we use this to handle `thread_local` semantics then it's a pay-for-what-you-use type situation I'd think.

https://github.com/llvm/llvm-project/pull/66167


More information about the libc-commits mailing list