//===----------------------------------------------------------------------===// // // 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) 2024 NVIDIA CORPORATION & AFFILIATES. // //===----------------------------------------------------------------------===// // Modifications Copyright (c) 2025 Advanced Micro Devices, Inc. // Permission is hereby granted, free of charge, to any person obtaining a copy // of this software and associated documentation files (the "Software"), to deal // in the Software without restriction, including without limitation the rights // to use, copy, modify, merge, publish, distribute, sublicense, and/or sell // copies of the Software, and to permit persons to whom the Software is // furnished to do so, subject to the following conditions: // The above copyright notice and this permission notice shall be included in // all copies or substantial portions of the Software. // THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR // IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY, // FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE // AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER // LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM, // OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN // THE SOFTWARE. #ifndef __CCCL_PTX_ISA_H_ #define __CCCL_PTX_ISA_H_ #include #include #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 #include // __CUDA_MINIMUM_ARCH__ and friends /* * Targeting macros * * Information from: * https://docs.nvidia.com/cuda/parallel-thread-execution/index.html#release-notes */ // PTX ISA 8.7 is available from CUDA 12.8 // The first define is for future major versions of CUDACC. // We make sure that these get the highest known PTX ISA version. // For clang cuda check https://github.com/llvm/llvm-project/blob/release/XX.x/clang/lib/Driver/ToolChains/Cuda.cpp // getNVPTXTargetFeatures #if _CCCL_CUDACC_AT_LEAST(13, 0) # define __cccl_ptx_isa 870ULL // PTX ISA 8.7 is available from CUDA 12.8, driver r570 #elif _CCCL_CUDACC_AT_LEAST(12, 8) && !_CCCL_CUDA_COMPILER(CLANG, <, 20) # define __cccl_ptx_isa 870ULL // PTX ISA 8.5 is available from CUDA 12.5, driver r555 #elif _CCCL_CUDACC_AT_LEAST(12, 5) && !_CCCL_CUDA_COMPILER(CLANG, <, 19) # define __cccl_ptx_isa 850ULL // PTX ISA 8.4 is available from CUDA 12.4, driver r550 #elif _CCCL_CUDACC_AT_LEAST(12, 4) && !_CCCL_CUDA_COMPILER(CLANG, <, 19) # define __cccl_ptx_isa 840ULL // PTX ISA 8.3 is available from CUDA 12.3, driver r545 #elif _CCCL_CUDACC_AT_LEAST(12, 3) && !_CCCL_CUDA_COMPILER(CLANG, <, 18) # define __cccl_ptx_isa 830ULL // PTX ISA 8.2 is available from CUDA 12.2, driver r535 #elif _CCCL_CUDACC_AT_LEAST(12, 2) && !_CCCL_CUDA_COMPILER(CLANG, <, 18) # define __cccl_ptx_isa 820ULL // PTX ISA 8.1 is available from CUDA 12.1, driver r530 #elif _CCCL_CUDACC_AT_LEAST(12, 1) && !_CCCL_CUDA_COMPILER(CLANG, <, 17) # define __cccl_ptx_isa 810ULL // PTX ISA 8.0 is available from CUDA 12.0, driver r525 #elif _CCCL_CUDACC_AT_LEAST(12, 0) && !_CCCL_CUDA_COMPILER(CLANG, <, 17) # define __cccl_ptx_isa 800ULL // PTX ISA 7.8 is available from CUDA 11.8, driver r520 #elif _CCCL_CUDACC_AT_LEAST(11, 8) && !_CCCL_CUDA_COMPILER(CLANG, <, 16) # define __cccl_ptx_isa 780ULL // PTX ISA 7.7 is available from CUDA 11.7, driver r515 #elif _CCCL_CUDACC_AT_LEAST(11, 7) && !_CCCL_CUDA_COMPILER(CLANG, <, 16) # define __cccl_ptx_isa 770ULL // PTX ISA 7.6 is available from CUDA 11.6, driver r510 #elif _CCCL_CUDACC_AT_LEAST(11, 6) && !_CCCL_CUDA_COMPILER(CLANG, <, 16) # define __cccl_ptx_isa 760ULL // PTX ISA 7.5 is available from CUDA 11.5, driver r495 #elif _CCCL_CUDACC_AT_LEAST(11, 5) && !_CCCL_CUDA_COMPILER(CLANG, <, 14) # define __cccl_ptx_isa 750ULL // PTX ISA 7.4 is available from CUDA 11.4, driver r470 #elif _CCCL_CUDACC_AT_LEAST(11, 4) && !_CCCL_CUDA_COMPILER(CLANG, <, 14) # define __cccl_ptx_isa 740ULL // PTX ISA 7.3 is available from CUDA 11.3, driver r465 #elif _CCCL_CUDACC_AT_LEAST(11, 3) && !_CCCL_CUDA_COMPILER(CLANG, <, 14) # define __cccl_ptx_isa 730ULL // PTX ISA 7.2 is available from CUDA 11.2, driver r460 #elif _CCCL_CUDACC_AT_LEAST(11, 2) && !_CCCL_CUDA_COMPILER(CLANG, <, 13) # define __cccl_ptx_isa 720ULL // PTX ISA 7.1 is available from CUDA 11.1, driver r455 #elif _CCCL_CUDACC_AT_LEAST(11, 1) && !_CCCL_CUDA_COMPILER(CLANG, <, 13) # define __cccl_ptx_isa 710ULL // PTX ISA 7.0 is available from CUDA 11.0, driver r445 #elif _CCCL_CUDACC_AT_LEAST(11, 0) && !_CCCL_CUDA_COMPILER(CLANG, <, 12) # define __cccl_ptx_isa 700ULL // Fallback case. Define the ISA version to be zero. This ensures that the macro is always defined. #else # define __cccl_ptx_isa 0ULL #endif // We define certain feature test macros depending on availability. When // __CUDA_MINIMUM_ARCH__ is not available, we define the following features // depending on PTX ISA. This permits checking for the feature in host code. // When __CUDA_MINIMUM_ARCH__ is available, we only enable the feature when the // hardware supports it. #if __cccl_ptx_isa >= 800 # if (!defined(__CUDA_MINIMUM_ARCH__)) || (defined(__CUDA_MINIMUM_ARCH__) && 900 <= __CUDA_MINIMUM_ARCH__) # define __cccl_lib_local_barrier_arrive_tx # define __cccl_lib_experimental_ctk12_cp_async_exposure # endif #endif // __cccl_ptx_isa >= 800 // NVRTC uses its own header, so we need to manually tell it when we expect SM90a to be available #if _CCCL_COMPILER(NVRTC) && !defined(NV_HAS_FEATURE_SM_90a) # if (defined(__CUDA_ARCH__) && (__CUDA_ARCH__ >= 900) && defined(__CUDA_ARCH_FEAT_SM90_ALL)) # define NV_HAS_FEATURE_SM_90a NV_PROVIDES_SM_90 # else // ^^^ SM90a ^^^ / vvv !SM90a vvv # define NV_HAS_FEATURE_SM_90a NV_NO_TARGET_LIBHIPCXX # endif // #endif // _CCCL_COMPILER(NVRTC) && !NV_HAS_FEATURE_SM_90a #endif // __CCCL_PTX_ISA_H_