5a4a05d776
Summary: [libomptarget][nfc] Move some source into common from nvptx Moves some source that compiles cleanly under amdgcn into a common subdirectory Includes some non-trivial files and some headers. Keeps the cuda file extension. The build systems for different architectures seem unlikely to have much in common. The idea is therefore to set include paths such that files under common/src compile as if they were under arch/src as the mechanism for sharing. In particular, files under common/src need to be able to include target_impl.h. The corresponding -Icommon is left out in favour of explicit includes on the basis that the it makes it clearer which files under common are used by a given architecture. Reviewers: jdoerfert, ABataev, grokos Reviewed By: ABataev Subscribers: jfb, mgorny, openmp-commits Tags: #openmp Differential Revision: https://reviews.llvm.org/D70328
52 lines
1.6 KiB
C++
52 lines
1.6 KiB
C++
//===--------- statequeue.h - NVPTX OpenMP GPU State Queue ------- CUDA -*-===//
|
|
//
|
|
// 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
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
//
|
|
// This file contains a queue to hand out OpenMP state objects to teams of
|
|
// one or more kernels.
|
|
//
|
|
// Reference:
|
|
// Thomas R.W. Scogland and Wu-chun Feng. 2015.
|
|
// Design and Evaluation of Scalable Concurrent Queues for Many-Core
|
|
// Architectures. International Conference on Performance Engineering.
|
|
//
|
|
//===----------------------------------------------------------------------===//
|
|
|
|
#ifndef __STATE_QUEUE_H
|
|
#define __STATE_QUEUE_H
|
|
|
|
#include <stdint.h>
|
|
|
|
#include "target_impl.h"
|
|
|
|
template <typename ElementType, uint32_t SIZE> class omptarget_nvptx_Queue {
|
|
private:
|
|
ElementType elements[SIZE];
|
|
volatile ElementType *elementQueue[SIZE];
|
|
volatile uint32_t head;
|
|
volatile uint32_t ids[SIZE];
|
|
volatile uint32_t tail;
|
|
|
|
static const uint32_t MAX_ID = (1u << 31) / SIZE / 2;
|
|
INLINE uint32_t ENQUEUE_TICKET();
|
|
INLINE uint32_t DEQUEUE_TICKET();
|
|
INLINE static uint32_t ID(uint32_t ticket);
|
|
INLINE bool IsServing(uint32_t slot, uint32_t id);
|
|
INLINE void PushElement(uint32_t slot, ElementType *element);
|
|
INLINE ElementType *PopElement(uint32_t slot);
|
|
INLINE void DoneServing(uint32_t slot, uint32_t id);
|
|
|
|
public:
|
|
INLINE omptarget_nvptx_Queue() {}
|
|
INLINE void Enqueue(ElementType *element);
|
|
INLINE ElementType *Dequeue();
|
|
};
|
|
|
|
#include "state-queuei.h"
|
|
|
|
#endif
|