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. 3cb23e90cSJeremy L Thompson // 4*3d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 5cb23e90cSJeremy L Thompson // 6*3d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 7cb23e90cSJeremy L Thompson 8ec3da8bcSJed Brown #include <ceed/ceed.h> 9ec3da8bcSJed Brown #include <ceed/backend.h> 106dbfb411Snbeams #include <string.h> 116dbfb411Snbeams #include <stdlib.h> 12cb23e90cSJeremy L Thompson #include "ceed-magma.h" 13cb23e90cSJeremy L Thompson 141d013790SJed Brown CEED_INTERN int CeedInit_Magma_Det(const char *resource, Ceed ceed) { 15cb23e90cSJeremy L Thompson int ierr; 166dbfb411Snbeams const int nrc = 18; // number of characters in resource 176dbfb411Snbeams if (strncmp(resource, "/gpu/cuda/magma/det", nrc) 186dbfb411Snbeams && strncmp(resource, "/gpu/hip/magma/det", nrc)) 19cb23e90cSJeremy L Thompson // LCOV_EXCL_START 20e15f9bd0SJeremy L Thompson return CeedError(ceed, CEED_ERROR_BACKEND, 21e15f9bd0SJeremy L Thompson "Magma backend cannot use resource: %s", resource); 22cb23e90cSJeremy L Thompson // LCOV_EXCL_STOP 23e15f9bd0SJeremy L Thompson ierr = CeedSetDeterministic(ceed, true); CeedChkBackend(ierr); 24cb23e90cSJeremy L Thompson 256dbfb411Snbeams Ceed_Magma *data; 266dbfb411Snbeams ierr = CeedCalloc(sizeof(Ceed_Magma), &data); CeedChkBackend(ierr); 276dbfb411Snbeams ierr = CeedSetData(ceed, data); CeedChkBackend(ierr); 286dbfb411Snbeams 296dbfb411Snbeams // get/set device ID 306dbfb411Snbeams const char *device_spec = strstr(resource, ":device_id="); 316dbfb411Snbeams const int deviceID = (device_spec) ? atoi(device_spec+11) : -1; 326dbfb411Snbeams 336dbfb411Snbeams int currentDeviceID; 346dbfb411Snbeams magma_getdevice(¤tDeviceID); 356dbfb411Snbeams if (deviceID >= 0 && currentDeviceID != deviceID) { 366dbfb411Snbeams magma_setdevice(deviceID); 376dbfb411Snbeams currentDeviceID = deviceID; 386dbfb411Snbeams } 396dbfb411Snbeams // create a queue that uses the null stream 406dbfb411Snbeams data->device = currentDeviceID; 416dbfb411Snbeams 42cb23e90cSJeremy L Thompson // Create reference CEED that implementation will be dispatched 43cb23e90cSJeremy L Thompson // through unless overridden 44cb23e90cSJeremy L Thompson Ceed ceedref; 45adb2481bSnbeams #ifdef HAVE_HIP 4640461fa4Snbeams CeedInit("/gpu/hip/magma", &ceedref); 47adb2481bSnbeams #else 4840461fa4Snbeams CeedInit("/gpu/cuda/magma", &ceedref); 49adb2481bSnbeams #endif 50e15f9bd0SJeremy L Thompson ierr = CeedSetDelegate(ceed, ceedref); CeedChkBackend(ierr); 51cb23e90cSJeremy L Thompson 52cb23e90cSJeremy L Thompson // Create reference CEED for restriction 53cb23e90cSJeremy L Thompson Ceed restrictionceedref; 54a31f51a5Snbeams #ifdef HAVE_HIP 55a31f51a5Snbeams CeedInit("/gpu/hip/ref", &restrictionceedref); 56a31f51a5Snbeams #else 57461525f5SNatalie Beams CeedInit("/gpu/cuda/ref", &restrictionceedref); 58a31f51a5Snbeams #endif 59cb23e90cSJeremy L Thompson ierr = CeedSetObjectDelegate(ceed, restrictionceedref, "ElemRestriction"); 60e15f9bd0SJeremy L Thompson CeedChkBackend(ierr); 61cb23e90cSJeremy L Thompson 62e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 63cb23e90cSJeremy L Thompson } 64cb23e90cSJeremy L Thompson 651d013790SJed Brown CEED_INTERN int CeedRegister_Magma_Det(void) { 66adb2481bSnbeams #ifdef HAVE_HIP 671d013790SJed Brown return CeedRegister("/gpu/hip/magma/det", CeedInit_Magma_Det, 125); 68adb2481bSnbeams #else 691d013790SJed Brown return CeedRegister("/gpu/cuda/magma/det", CeedInit_Magma_Det, 125); 70adb2481bSnbeams #endif 71cb23e90cSJeremy L Thompson } 72