/home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hipcub/checkouts/docs-5.0.2/hipcub/include/hipcub/config.hpp Source File# hipCUB: /home/docs/checkouts/readthedocs.org/user_builds/advanced-micro-devices-hipcub/checkouts/docs-5.0.2/hipcub/include/hipcub/config.hpp Source File homedocscheckoutsreadthedocs.orguser_buildsadvanced-micro-devices-hipcubcheckoutsdocs-5.0.2hipcubincludehipcub config.hpp 1 /****************************************************************************** 2 * Copyright (c) 2010-2011, Duane Merrill. All rights reserved. 3 * Copyright (c) 2011-2018, NVIDIA CORPORATION. All rights reserved. 4 * Modifications Copyright (c) 2019-2020, Advanced Micro Devices, Inc. All rights reserved. 5 * 6 * Redistribution and use in source and binary forms, with or without 7 * modification, are permitted provided that the following conditions are met: 8 * * Redistributions of source code must retain the above copyright 9 * notice, this list of conditions and the following disclaimer. 10 * * Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * * Neither the name of the NVIDIA CORPORATION nor the 14 * names of its contributors may be used to endorse or promote products 15 * derived from this software without specific prior written permission. 16 * 17 * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS" AND 18 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED 19 * WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE 20 * DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE FOR ANY 21 * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES 22 * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; 23 * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND 24 * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 25 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS 26 * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 27 * 28 ******************************************************************************/ 29 30 #ifndef HIPCUB_CONFIG_HPP_ 31 #define HIPCUB_CONFIG_HPP_ 32 33 #include <hip/hip_runtime.h> 34 35 #define HIPCUB_NAMESPACE hipcub 36 37 #define BEGIN_HIPCUB_NAMESPACE \ 38 namespace hipcub { 39 40 #define END_HIPCUB_NAMESPACE \ 41 } /* hipcub */ 42 43 #ifdef __HIP_PLATFORM_AMD__ 44 #define HIPCUB_ROCPRIM_API 1 45 #define HIPCUB_RUNTIME_FUNCTION __host__ 46 #elif defined(__HIP_PLATFORM_NVIDIA__) 47 #define HIPCUB_CUB_API 1 48 #define HIPCUB_RUNTIME_FUNCTION CUB_RUNTIME_FUNCTION 49 50 #include <cub/util_arch.cuh> 51 #define HIPCUB_WARP_THREADS CUB_PTX_WARP_THREADS 52 #define HIPCUB_DEVICE_WARP_THREADS CUB_PTX_WARP_THREADS 53 #define HIPCUB_HOST_WARP_THREADS CUB_PTX_WARP_THREADS 54 #define HIPCUB_ARCH CUB_PTX_ARCH 55 BEGIN_HIPCUB_NAMESPACE 56 using namespace cub; 57 END_HIPCUB_NAMESPACE 58 #endif 59 61 #define HIPCUB_WARP_SIZE_32 32u 62 #define HIPCUB_WARP_SIZE_64 64u 63 #define HIPCUB_MAX_WARP_SIZE HIPCUB_WARP_SIZE_64 64 65 #define HIPCUB_HOST __host__ 66 #define HIPCUB_DEVICE __device__ 67 #define HIPCUB_HOST_DEVICE __host__ __device__ 68 #define HIPCUB_SHARED_MEMORY __shared__ 69 70 BEGIN_HIPCUB_NAMESPACE 71 73 #if (defined(DEBUG) || defined(_DEBUG)) && !defined(HIPCUB_STDERR) 74 #define HIPCUB_STDERR 75 #endif 76 77 inline 78 hipError_t Debug( 79 hipError_t error, 80 const char* filename, 81 int line) 82 { 83 (void)filename; 84 (void)line; 85 #ifdef HIPCUB_STDERR 86 if (error) 87 { 88 fprintf(stderr, "HIP error %d [%s, %d]: %s\n", error, filename, line, hipGetErrorString(error)); 89 fflush(stderr); 90 } 91 #endif 92 return error; 93 } 94 95 #ifndef HipcubDebug 96 #define HipcubDebug(e) hipcub::Debug((hipError_t) (e), __FILE__, __LINE__) 97 #endif 98 99 END_HIPCUB_NAMESPACE 100 101 #endif // HIPCUB_CONFIG_HPP_