xref: /libCEED/backends/hip-shared/ceed-hip-shared.c (revision 3d8e882215d238700cdceb37404f76ca7fa24eaa)
1*3d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors.
2*3d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details.
37d8d0e25Snbeams //
4*3d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause
57d8d0e25Snbeams //
6*3d8e8822SJeremy L Thompson // This file is part of CEED:  http://github.com/ceed
77d8d0e25Snbeams 
8ec3da8bcSJed Brown #include <ceed/ceed.h>
9ec3da8bcSJed Brown #include <ceed/backend.h>
103d576824SJeremy L Thompson #include <stdbool.h>
117d8d0e25Snbeams #include <string.h>
127d8d0e25Snbeams #include "ceed-hip-shared.h"
137d8d0e25Snbeams 
147d8d0e25Snbeams //------------------------------------------------------------------------------
157d8d0e25Snbeams // Backend init
167d8d0e25Snbeams //------------------------------------------------------------------------------
177d8d0e25Snbeams static int CeedInit_Hip_shared(const char *resource, Ceed ceed) {
187d8d0e25Snbeams   int ierr;
19f87d896cSJeremy L Thompson 
20f87d896cSJeremy L Thompson   if (strcmp(resource, "/gpu/hip/shared"))
217d8d0e25Snbeams     // LCOV_EXCL_START
22e15f9bd0SJeremy L Thompson     return CeedError(ceed, CEED_ERROR_BACKEND,
23e15f9bd0SJeremy L Thompson                      "Hip backend cannot use resource: %s", resource);
247d8d0e25Snbeams   // LCOV_EXCL_STOP
25e15f9bd0SJeremy L Thompson   ierr = CeedSetDeterministic(ceed, true); CeedChkBackend(ierr);
267d8d0e25Snbeams 
276dbfb411Snbeams   Ceed_Hip *data;
28e15f9bd0SJeremy L Thompson   ierr = CeedCalloc(1, &data); CeedChkBackend(ierr);
29e15f9bd0SJeremy L Thompson   ierr = CeedSetData(ceed, data); CeedChkBackend(ierr);
30f87d896cSJeremy L Thompson   ierr = CeedHipInit(ceed, resource); CeedChkBackend(ierr);
317d8d0e25Snbeams 
32437930d1SJeremy L Thompson   Ceed ceed_ref;
33437930d1SJeremy L Thompson   CeedInit("/gpu/hip/ref", &ceed_ref);
34437930d1SJeremy L Thompson   ierr = CeedSetDelegate(ceed, ceed_ref); CeedChkBackend(ierr);
356dbfb411Snbeams 
367d8d0e25Snbeams   ierr = CeedSetBackendFunction(ceed, "Ceed", ceed, "BasisCreateTensorH1",
377d8d0e25Snbeams                                 CeedBasisCreateTensorH1_Hip_shared);
38e15f9bd0SJeremy L Thompson   CeedChkBackend(ierr);
397d8d0e25Snbeams   ierr = CeedSetBackendFunction(ceed, "Ceed", ceed, "Destroy",
40e15f9bd0SJeremy L Thompson                                 CeedDestroy_Hip); CeedChkBackend(ierr);
41e15f9bd0SJeremy L Thompson   CeedChkBackend(ierr);
42e15f9bd0SJeremy L Thompson   return CEED_ERROR_SUCCESS;
437d8d0e25Snbeams }
447d8d0e25Snbeams 
457d8d0e25Snbeams //------------------------------------------------------------------------------
467d8d0e25Snbeams // Register backend
477d8d0e25Snbeams //------------------------------------------------------------------------------
481d013790SJed Brown CEED_INTERN int CeedRegister_Hip_Shared(void) {
491d013790SJed Brown   return CeedRegister("/gpu/hip/shared", CeedInit_Hip_shared, 25);
507d8d0e25Snbeams }
517d8d0e25Snbeams //------------------------------------------------------------------------------
52