13d8e8822SJeremy L Thompson // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors. 23d8e8822SJeremy L Thompson // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 3cb23e90cSJeremy L Thompson // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 5cb23e90cSJeremy L Thompson // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 7cb23e90cSJeremy L Thompson 849aac155SJeremy L Thompson #include <ceed.h> 9ec3da8bcSJed Brown #include <ceed/backend.h> 106dbfb411Snbeams #include <stdlib.h> 112b730f8bSJeremy L Thompson #include <string.h> 122b730f8bSJeremy L Thompson 13cb23e90cSJeremy L Thompson #include "ceed-magma.h" 14cb23e90cSJeremy L Thompson 15*6574a04fSJeremy L Thompson static int CeedInit_Magma_Det(const char *resource, Ceed ceed) { 166dbfb411Snbeams const int nrc = 18; // number of characters in resource 17*6574a04fSJeremy L Thompson CeedCheck(!strncmp(resource, "/gpu/cuda/magma/det", nrc) || !strncmp(resource, "/gpu/hip/magma/det", nrc), ceed, CEED_ERROR_BACKEND, 18*6574a04fSJeremy L Thompson "Magma backend cannot use resource: %s", resource); 192b730f8bSJeremy L Thompson CeedCallBackend(CeedSetDeterministic(ceed, true)); 20cb23e90cSJeremy L Thompson 216dbfb411Snbeams Ceed_Magma *data; 222b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(sizeof(Ceed_Magma), &data)); 232b730f8bSJeremy L Thompson CeedCallBackend(CeedSetData(ceed, data)); 246dbfb411Snbeams 256dbfb411Snbeams // get/set device ID 266dbfb411Snbeams const char *device_spec = strstr(resource, ":device_id="); 276dbfb411Snbeams const int deviceID = (device_spec) ? atoi(device_spec + 11) : -1; 286dbfb411Snbeams 296dbfb411Snbeams int currentDeviceID; 306dbfb411Snbeams magma_getdevice(¤tDeviceID); 316dbfb411Snbeams if (deviceID >= 0 && currentDeviceID != deviceID) { 326dbfb411Snbeams magma_setdevice(deviceID); 336dbfb411Snbeams currentDeviceID = deviceID; 346dbfb411Snbeams } 356dbfb411Snbeams // create a queue that uses the null stream 366dbfb411Snbeams data->device = currentDeviceID; 376dbfb411Snbeams 38cb23e90cSJeremy L Thompson // Create reference CEED that implementation will be dispatched 39cb23e90cSJeremy L Thompson // through unless overridden 40cb23e90cSJeremy L Thompson Ceed ceedref; 41e5f091ebSnbeams #ifdef CEED_MAGMA_USE_HIP 422b730f8bSJeremy L Thompson CeedCallBackend(CeedInit("/gpu/hip/magma", &ceedref)); 43adb2481bSnbeams #else 442b730f8bSJeremy L Thompson CeedCallBackend(CeedInit("/gpu/cuda/magma", &ceedref)); 45adb2481bSnbeams #endif 462b730f8bSJeremy L Thompson CeedCallBackend(CeedSetDelegate(ceed, ceedref)); 47cb23e90cSJeremy L Thompson 48cb23e90cSJeremy L Thompson // Create reference CEED for restriction 49cb23e90cSJeremy L Thompson Ceed restrictionceedref; 50e5f091ebSnbeams #ifdef CEED_MAGMA_USE_HIP 51a31f51a5Snbeams CeedInit("/gpu/hip/ref", &restrictionceedref); 52a31f51a5Snbeams #else 53461525f5SNatalie Beams CeedInit("/gpu/cuda/ref", &restrictionceedref); 54a31f51a5Snbeams #endif 552b730f8bSJeremy L Thompson CeedCallBackend(CeedSetObjectDelegate(ceed, restrictionceedref, "ElemRestriction")); 56cb23e90cSJeremy L Thompson 57e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 58cb23e90cSJeremy L Thompson } 59cb23e90cSJeremy L Thompson 601d013790SJed Brown CEED_INTERN int CeedRegister_Magma_Det(void) { 61e5f091ebSnbeams #ifdef CEED_MAGMA_USE_HIP 621d013790SJed Brown return CeedRegister("/gpu/hip/magma/det", CeedInit_Magma_Det, 125); 63adb2481bSnbeams #else 641d013790SJed Brown return CeedRegister("/gpu/cuda/magma/det", CeedInit_Magma_Det, 125); 65adb2481bSnbeams #endif 66cb23e90cSJeremy L Thompson } 67