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-magma.h"
9
10 #include <ceed.h>
11 #include <ceed/backend.h>
12 #include <stdlib.h>
13 #include <string.h>
14
15 #include "ceed-magma-common.h"
16
17 //------------------------------------------------------------------------------
18 // Backend Init
19 //------------------------------------------------------------------------------
CeedInit_Magma(const char * resource,Ceed ceed)20 static int CeedInit_Magma(const char *resource, Ceed ceed) {
21 Ceed ceed_ref;
22 Ceed_Magma *data;
23 const int nrc = 14; // number of characters in resource
24
25 CeedCheck(!strncmp(resource, "/gpu/cuda/magma", nrc) || !strncmp(resource, "/gpu/hip/magma", nrc), ceed, CEED_ERROR_BACKEND,
26 "Magma backend cannot use resource: %s", resource);
27
28 CeedCallBackend(CeedCalloc(1, &data));
29 CeedCallBackend(CeedSetData(ceed, data));
30 CeedCallBackend(CeedInit_Magma_common(ceed, resource));
31
32 // Create reference Ceed that implementation will be dispatched through unless overridden
33 #ifdef CEED_MAGMA_USE_HIP
34 CeedCallBackend(CeedInit("/gpu/hip/ref", &ceed_ref));
35 #else
36 CeedCallBackend(CeedInit("/gpu/cuda/ref", &ceed_ref));
37 #endif
38 CeedCallBackend(CeedSetDelegate(ceed, ceed_ref));
39 CeedCallBackend(CeedDestroy(&ceed_ref));
40
41 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateTensorH1", CeedBasisCreateTensorH1_Magma));
42 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateH1", CeedBasisCreateH1_Magma));
43 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateHdiv", CeedBasisCreateHdiv_Magma));
44 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateHcurl", CeedBasisCreateHcurl_Magma));
45 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "Destroy", CeedDestroy_Magma));
46 return CEED_ERROR_SUCCESS;
47 }
48
49 //------------------------------------------------------------------------------
50 // Backend Register
51 //------------------------------------------------------------------------------
CeedRegister_Magma(void)52 CEED_INTERN int CeedRegister_Magma(void) {
53 #ifdef CEED_MAGMA_USE_HIP
54 return CeedRegister("/gpu/hip/magma", CeedInit_Magma, 120);
55 #else
56 return CeedRegister("/gpu/cuda/magma", CeedInit_Magma, 120);
57 #endif
58 }
59
60 //------------------------------------------------------------------------------
61