xref: /libCEED/backends/hip-shared/ceed-hip-shared.c (revision 0c1bc3c213b7e9c1cad4a62aaf359c4e4c7d4e25)
1 // Copyright (c) 2017-2018, Lawrence Livermore National Security, LLC.
2 // Produced at the Lawrence Livermore National Laboratory. LLNL-CODE-734707.
3 // All Rights reserved. See files LICENSE and NOTICE for details.
4 //
5 // This file is part of CEED, a collection of benchmarks, miniapps, software
6 // libraries and APIs for efficient high-order finite element and spectral
7 // element discretizations for exascale applications. For more information and
8 // source code availability see http://github.com/ceed.
9 //
10 // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC,
11 // a collaborative effort of two U.S. Department of Energy organizations (Office
12 // of Science and the National Nuclear Security Administration) responsible for
13 // the planning and preparation of a capable exascale ecosystem, including
14 // software, applications, hardware, advanced system engineering and early
15 // testbed platforms, in support of the nation's exascale computing imperative.
16 
17 #include <ceed/ceed.h>
18 #include <ceed/backend.h>
19 #include <stdbool.h>
20 #include <string.h>
21 #include "ceed-hip-shared.h"
22 #include "../hip/ceed-hip.h"
23 
24 //------------------------------------------------------------------------------
25 // Backend init
26 //------------------------------------------------------------------------------
27 static int CeedInit_Hip_shared(const char *resource, Ceed ceed) {
28   int ierr;
29   const int nrc = 8; // number of characters in resource
30   if (strncmp(resource, "/gpu/hip/shared", nrc))
31     // LCOV_EXCL_START
32     return CeedError(ceed, CEED_ERROR_BACKEND,
33                      "Hip backend cannot use resource: %s", resource);
34   // LCOV_EXCL_STOP
35   ierr = CeedSetDeterministic(ceed, true); CeedChkBackend(ierr);
36 
37   Ceed_Hip *data;
38   ierr = CeedCalloc(1, &data); CeedChkBackend(ierr);
39   ierr = CeedSetData(ceed, data); CeedChkBackend(ierr);
40   ierr = CeedHipInit(ceed, resource, nrc); CeedChkBackend(ierr);
41 
42   Ceed ceedref;
43   CeedInit("/gpu/hip/ref", &ceedref);
44   ierr = CeedSetDelegate(ceed, ceedref); CeedChkBackend(ierr);
45 
46   ierr = CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateTensorH1",
47                                 CeedBasisCreateTensorH1_Hip_shared);
48   CeedChkBackend(ierr);
49   ierr = CeedSetBackendFunction(ceed, "Ceed", ceed, "Destroy",
50                                 CeedDestroy_Hip); CeedChkBackend(ierr);
51   CeedChkBackend(ierr);
52   return CEED_ERROR_SUCCESS;
53 }
54 
55 //------------------------------------------------------------------------------
56 // Register backend
57 //------------------------------------------------------------------------------
58 CEED_INTERN int CeedRegister_Hip_Shared(void) {
59   return CeedRegister("/gpu/hip/shared", CeedInit_Hip_shared, 25);
60 }
61 //------------------------------------------------------------------------------
62