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