nvtx3/nvtxDetail/nvtxExtImplMemCudaRt_v1.h

0001 /*
0002  * SPDX-FileCopyrightText: Copyright (c) 2009-2025 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
0003  * SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
0004  *
0005  * Licensed under the Apache License, Version 2.0 (the "License");
0006  * you may not use this file except in compliance with the License.
0007  * You may obtain a copy of the License at
0008  *
0009  *     http://www.apache.org/licenses/LICENSE-2.0
0010  *
0011  * Unless required by applicable law or agreed to in writing, software
0012  * distributed under the License is distributed on an "AS IS" BASIS,
0013  * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
0014  * See the License for the specific language governing permissions and
0015  * limitations under the License.
0016  *
0017  * Licensed under the Apache License v2.0 with LLVM Exceptions.
0018  * See https://nvidia.github.io/NVTX/LICENSE.txt for license information.
0019  */
0020
0021 #ifndef NVTX_EXT_IMPL_MEM_CUDART_GUARD
0022 #error Never include this file directly -- it is automatically included by nvToolsExtMemCudaRt.h (except when NVTX_NO_IMPL is defined).
0023 #endif
0024
0025 #if defined(NVTX_AS_SYSTEM_HEADER)
0026 #if defined(__clang__)
0027 #pragma clang system_header
0028 #elif defined(__GNUC__) || defined(__NVCOMPILER)
0029 #pragma GCC system_header
0030 #elif defined(_MSC_VER)
0031 #pragma system_header
0032 #endif
0033 #endif
0034
0035 #ifndef NVTX_EXT_IMPL_MEM_CUDART_V1
0036 #define NVTX_EXT_IMPL_MEM_CUDART_V1
0037
0038 #ifdef __cplusplus
0039 extern "C" {
0040 #endif /* __cplusplus */
0041
0042 /* NVTX_EXT_MEM_IMPL_FN_V1 defined in nvtxExtImplMem_v1.h */
0043
0044 /* Non-void functions. */
0045 #define NVTX_EXT_FN_RETURN return
0046 #define NVTX_EXT_FN_RETURN_INVALID(rtype) return NVTX_NULLPTR;
0047
0048 NVTX_EXT_MEM_IMPL_FN_V1(nvtxMemPermissionsHandle_t, nvtxMemCudaGetProcessWidePermissions, (nvtxDomainHandle_t domain), (domain))
0049
0050 NVTX_EXT_MEM_IMPL_FN_V1(nvtxMemPermissionsHandle_t, nvtxMemCudaGetDeviceWidePermissions, (nvtxDomainHandle_t domain, int device), (domain, device))
0051
0052 #undef NVTX_EXT_FN_RETURN
0053 #undef NVTX_EXT_FN_RETURN_INVALID
0054 /* END: Non-void functions. */
0055
0056 /* void functions. */
0057 #define NVTX_EXT_FN_RETURN
0058 #define NVTX_EXT_FN_RETURN_INVALID(rtype)
0059
0060 NVTX_EXT_MEM_IMPL_FN_V1(void, nvtxMemCudaSetPeerAccess, (nvtxDomainHandle_t domain, nvtxMemPermissionsHandle_t permissions, int devicePeer, uint32_t flags), (domain, permissions, devicePeer, flags))
0061
0062 NVTX_EXT_MEM_IMPL_FN_V1(void, nvtxMemCudaMarkInitialized, (nvtxDomainHandle_t domain, cudaStream_t stream, uint8_t isPerThreadStream, nvtxMemMarkInitializedBatch_t const* desc), (domain, stream, isPerThreadStream, desc))
0063
0064 #undef NVTX_EXT_FN_RETURN
0065 #undef NVTX_EXT_FN_RETURN_INVALID
0066 /* END: void functions. */
0067
0068 #ifdef __cplusplus
0069 } /* extern "C" */
0070 #endif /* __cplusplus */
0071
0072 #endif /* NVTX_EXT_IMPL_MEM_CUDART_V1 */