43 lines
1.6 KiB
Plaintext
43 lines
1.6 KiB
Plaintext
|
//===----------------------------------------------------------------------===//
|
||
|
|
//
|
||
|
|
// Part of libcu++, the C++ Standard Library for your entire system,
|
||
|
|
// 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
|
||
|
|
// SPDX-FileCopyrightText: Copyright (c) 2023 NVIDIA CORPORATION & AFFILIATES.
|
||
|
|
//
|
||
|
|
//===----------------------------------------------------------------------===//
|
||
|
|
|
||
|
|
#ifndef _CUDA_STD_BARRIER
|
||
|
|
#define _CUDA_STD_BARRIER
|
||
|
|
|
||
|
|
#include <cuda/std/detail/__config>
|
||
|
|
|
||
|
|
#if defined(_CCCL_IMPLICIT_SYSTEM_HEADER_GCC)
|
||
|
|
# pragma GCC system_header
|
||
|
|
#elif defined(_CCCL_IMPLICIT_SYSTEM_HEADER_CLANG)
|
||
|
|
# pragma clang system_header
|
||
|
|
#elif defined(_CCCL_IMPLICIT_SYSTEM_HEADER_MSVC)
|
||
|
|
# pragma system_header
|
||
|
|
#endif // no system header
|
||
|
|
|
||
|
|
#if _CCCL_DEVICE_COMPILATION() && _CCCL_PTX_ARCH() < 700 && !_CCCL_CUDA_COMPILER(NVHPC)
|
||
|
|
# error "CUDA synchronization primitives are only supported for sm_70 and up."
|
||
|
|
#endif // _CCCL_DEVICE_COMPILATION() && _CCCL_PTX_ARCH() < 700 && !_CCCL_CUDA_COMPILER(NVHPC)
|
||
|
|
|
||
|
|
#include <cuda/std/__barrier/barrier.h>
|
||
|
|
#include <cuda/std/__barrier/empty_completion.h>
|
||
|
|
#include <cuda/std/__barrier/poll_tester.h>
|
||
|
|
|
||
|
|
//! TODO: Drop cuda only features
|
||
|
|
#include <cuda/__barrier/barrier.h>
|
||
|
|
#include <cuda/__barrier/barrier_arrive_tx.h>
|
||
|
|
#include <cuda/__barrier/barrier_block_scope.h>
|
||
|
|
#include <cuda/__barrier/barrier_expect_tx.h>
|
||
|
|
#include <cuda/__barrier/barrier_thread_scope.h>
|
||
|
|
#include <cuda/__fwd/pipeline.h>
|
||
|
|
#include <cuda/__memcpy_async/memcpy_async.h>
|
||
|
|
#include <cuda/__memcpy_async/memcpy_async_tx.h>
|
||
|
|
|
||
|
|
#endif // _CUDA_STD_BARRIER
|