hipcub/device/device_for.hpp Source File

hipcub/device/device_for.hpp Source File#

hipCUB: hipcub/device/device_for.hpp Source File
device_for.hpp
1 /******************************************************************************
2  * Copyright (c) 2024, NVIDIA CORPORATION. All rights reserved.
3  * Modifications Copyright (c) 2024-2025, Advanced Micro Devices, Inc. All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions are met:
7  * * Redistributions of source code must retain the above copyright
8  * notice, this list of conditions and the following disclaimer.
9  * * Redistributions in binary form must reproduce the above copyright
10  * notice, this list of conditions and the following disclaimer in the
11  * documentation and/or other materials provided with the distribution.
12  * * Neither the name of the NVIDIA CORPORATION nor the
13  * names of its contributors may be used to endorse or promote products
14  * derived from this software without specific prior written permission.
15  *
16  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
17  * AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
18  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
19  * ARE DISCLAIMED. IN NO EVENT SHALL NVIDIA CORPORATION BE LIABLE FOR ANY
20  * DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES
21  * (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
22  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND
23  * ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
25  * SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26  *
27  ******************************************************************************/
28 
29 #ifndef HIPCUB_DEVICE_DEVICE_FOR_HPP_
30 #define HIPCUB_DEVICE_DEVICE_FOR_HPP_
31 
32 #include <hip/hip_common.h>
33 
34 #ifdef __HIP_PLATFORM_AMD__
35  #include "../backend/rocprim/device/device_for.hpp" // IWYU pragma: export
36 #elif defined(__HIP_PLATFORM_NVIDIA__)
37  #include "../backend/cub/device/device_for.hpp" // IWYU pragma: export
38 #endif
39 
40 #endif // HIPCUB_DEVICE_DEVICE_FOR_HPP_