1 // Copyright (c) 2017-2026, Lawrence Livermore National Security, LLC and other CEED contributors.
2 // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
4 // SPDX-License-Identifier: BSD-2-Clause
6 // This file is part of CEED:  http://github.com/ceed
8 #include "ceed-hip-common.h"
10 #include <ceed.h>
11 #include <ceed/backend.h>
15 //------------------------------------------------------------------------------
17 //------------------------------------------------------------------------------
18 int CeedInit_Hip(Ceed ceed, const char *resource) {  in CeedInit_Hip()  argument
21   const int   device_id   = (device_spec) ? atoi(device_spec + 11) : -1;  in CeedInit_Hip()
25   CeedCallHip(ceed, hipGetDevice(&current_device_id));  in CeedInit_Hip()
27     CeedCallHip(ceed, hipSetDevice(device_id));  in CeedInit_Hip()
31   CeedCallBackend(CeedGetData(ceed, &data));  in CeedInit_Hip()
32   data->device_id = current_device_id;  in CeedInit_Hip()
33   CeedCallHip(ceed, hipGetDeviceProperties(&data->device_prop, current_device_id));  in CeedInit_Hip()
36   data->has_unified_addressing = xnack_value > 0 ? data->device_prop.unifiedAddressing : 0;  in CeedInit_Hip()
37   if (data->has_unified_addressing) {  in CeedInit_Hip()
38     CeedDebug(ceed, "Using unified memory addressing");  in CeedInit_Hip()
40   data->opt_block_size = 256;  in CeedInit_Hip()
44 //------------------------------------------------------------------------------
46 //------------------------------------------------------------------------------
47 int CeedDestroy_Hip(Ceed ceed) {  in CeedDestroy_Hip()  argument
50   CeedCallBackend(CeedGetData(ceed, &data));  in CeedDestroy_Hip()
51   if (data->hipblas_handle) CeedCallHipblas(ceed, hipblasDestroy(data->hipblas_handle));  in CeedDestroy_Hip()
56 //------------------------------------------------------------------------------
58 //------------------------------------------------------------------------------
59 static inline int CeedSetDeviceGenericArray_Hip(Ceed ceed, const void *source_array, CeedCopyMode c…  in CeedSetDeviceGenericArray_Hip()  argument
67 …if (!*(void **)target_array_owned) CeedCallHip(ceed, hipMalloc(target_array_owned, size_unit * num…  in CeedSetDeviceGenericArray_Hip()
71 …if (source_array) CeedCallHip(ceed, hipMemcpy(*(void **)target_array, source_array, size_unit * nu…  in CeedSetDeviceGenericArray_Hip()
74       CeedCallHip(ceed, hipFree(*(void **)target_array_owned));  in CeedSetDeviceGenericArray_Hip()
80       CeedCallHip(ceed, hipFree(*(void **)target_array_owned));  in CeedSetDeviceGenericArray_Hip()
88 int CeedSetDeviceBoolArray_Hip(Ceed ceed, const bool *source_array, CeedCopyMode copy_mode, CeedSiz…  in CeedSetDeviceBoolArray_Hip()  argument
90 …CeedCallBackend(CeedSetDeviceGenericArray_Hip(ceed, source_array, copy_mode, sizeof(bool), num_val…  in CeedSetDeviceBoolArray_Hip()
95 int CeedSetDeviceCeedInt8Array_Hip(Ceed ceed, const CeedInt8 *source_array, CeedCopyMode copy_mode,…  in CeedSetDeviceCeedInt8Array_Hip()  argument
97 …CeedCallBackend(CeedSetDeviceGenericArray_Hip(ceed, source_array, copy_mode, sizeof(CeedInt8), num…  in CeedSetDeviceCeedInt8Array_Hip()
102 int CeedSetDeviceCeedIntArray_Hip(Ceed ceed, const CeedInt *source_array, CeedCopyMode copy_mode, C…  in CeedSetDeviceCeedIntArray_Hip()  argument
104 …CeedCallBackend(CeedSetDeviceGenericArray_Hip(ceed, source_array, copy_mode, sizeof(CeedInt), num_…  in CeedSetDeviceCeedIntArray_Hip()
109 int CeedSetDeviceCeedScalarArray_Hip(Ceed ceed, const CeedScalar *source_array, CeedCopyMode copy_m…  in CeedSetDeviceCeedScalarArray_Hip()  argument
112 …CeedCallBackend(CeedSetDeviceGenericArray_Hip(ceed, source_array, copy_mode, sizeof(CeedScalar), n…  in CeedSetDeviceCeedScalarArray_Hip()
117 //------------------------------------------------------------------------------