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.
3 //
4 // SPDX-License-Identifier: BSD-2-Clause
5 //
6 // This file is part of CEED: http://github.com/ceed
7
8 #include "ceed-hip-ref.h"
9
10 #include <ceed.h>
11 #include <ceed/backend.h>
12 #include <stdbool.h>
13 #include <string.h>
14
15 #include "../hip/ceed-hip-common.h"
16
17 //------------------------------------------------------------------------------
18 // HIP preferred MemType
19 //------------------------------------------------------------------------------
CeedGetPreferredMemType_Hip(CeedMemType * mem_type)20 static int CeedGetPreferredMemType_Hip(CeedMemType *mem_type) {
21 *mem_type = CEED_MEM_DEVICE;
22 return CEED_ERROR_SUCCESS;
23 }
24
25 //------------------------------------------------------------------------------
26 // Get hipBLAS handle
27 //------------------------------------------------------------------------------
CeedGetHipblasHandle_Hip(Ceed ceed,hipblasHandle_t * handle)28 int CeedGetHipblasHandle_Hip(Ceed ceed, hipblasHandle_t *handle) {
29 Ceed_Hip *data;
30
31 CeedCallBackend(CeedGetData(ceed, &data));
32 if (!data->hipblas_handle) {
33 CeedCallHipblas(ceed, hipblasCreate(&data->hipblas_handle));
34 CeedCallHipblas(ceed, hipblasSetPointerMode(data->hipblas_handle, HIPBLAS_POINTER_MODE_HOST));
35 }
36 *handle = data->hipblas_handle;
37 return CEED_ERROR_SUCCESS;
38 }
39
40 //------------------------------------------------------------------------------
41 // Backend Init
42 //------------------------------------------------------------------------------
CeedInit_Hip_ref(const char * resource,Ceed ceed)43 static int CeedInit_Hip_ref(const char *resource, Ceed ceed) {
44 Ceed_Hip *data;
45 char *resource_root;
46
47 CeedCallBackend(CeedGetResourceRoot(ceed, resource, ":", &resource_root));
48 CeedCheck(!strcmp(resource_root, "/gpu/hip/ref"), ceed, CEED_ERROR_BACKEND, "Hip backend cannot use resource: %s", resource);
49 CeedCallBackend(CeedFree(&resource_root));
50 CeedCallBackend(CeedSetDeterministic(ceed, true));
51
52 CeedCallBackend(CeedCalloc(1, &data));
53 CeedCallBackend(CeedSetData(ceed, data));
54 CeedCallBackend(CeedInit_Hip(ceed, resource));
55
56 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "GetPreferredMemType", CeedGetPreferredMemType_Hip));
57 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "VectorCreate", CeedVectorCreate_Hip));
58 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateTensorH1", CeedBasisCreateTensorH1_Hip));
59 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateH1", CeedBasisCreateH1_Hip));
60 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateHdiv", CeedBasisCreateHdiv_Hip));
61 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateHcurl", CeedBasisCreateHcurl_Hip));
62 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "ElemRestrictionCreate", CeedElemRestrictionCreate_Hip));
63 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "ElemRestrictionCreateAtPoints", CeedElemRestrictionCreate_Hip));
64 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "QFunctionCreate", CeedQFunctionCreate_Hip));
65 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "QFunctionContextCreate", CeedQFunctionContextCreate_Hip));
66 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "OperatorCreate", CeedOperatorCreate_Hip));
67 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "OperatorCreateAtPoints", CeedOperatorCreateAtPoints_Hip));
68 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "Destroy", CeedDestroy_Hip));
69 return CEED_ERROR_SUCCESS;
70 }
71
72 //------------------------------------------------------------------------------
73 // Backend Register
74 //------------------------------------------------------------------------------
CeedRegister_Hip(void)75 CEED_INTERN int CeedRegister_Hip(void) { return CeedRegister("/gpu/hip/ref", CeedInit_Hip_ref, 40); }
76
77 //------------------------------------------------------------------------------
78