1*5aed82e4SJeremy L Thompson // Copyright (c) 2017-2024, 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 1300fb7a04SSebastian Grimberg #include "ceed-magma-common.h" 14cb23e90cSJeremy L Thompson 1500fb7a04SSebastian Grimberg //------------------------------------------------------------------------------ 1600fb7a04SSebastian Grimberg // Backend Init 1700fb7a04SSebastian Grimberg //------------------------------------------------------------------------------ 186574a04fSJeremy L Thompson static int CeedInit_Magma_Det(const char *resource, Ceed ceed) { 1938293ee6SJeremy L Thompson Ceed ceed_ref; 2038293ee6SJeremy L Thompson Ceed_Magma *data; 216dbfb411Snbeams const int nrc = 18; // number of characters in resource 2238293ee6SJeremy L Thompson 236574a04fSJeremy L Thompson CeedCheck(!strncmp(resource, "/gpu/cuda/magma/det", nrc) || !strncmp(resource, "/gpu/hip/magma/det", nrc), ceed, CEED_ERROR_BACKEND, 246574a04fSJeremy L Thompson "Magma backend cannot use resource: %s", resource); 252b730f8bSJeremy L Thompson CeedCallBackend(CeedSetDeterministic(ceed, true)); 26cb23e90cSJeremy L Thompson 2758549094SSebastian Grimberg CeedCallBackend(CeedCalloc(1, &data)); 282b730f8bSJeremy L Thompson CeedCallBackend(CeedSetData(ceed, data)); 2900fb7a04SSebastian Grimberg CeedCallBackend(CeedInit_Magma_common(ceed, resource)); 306dbfb411Snbeams 3158549094SSebastian Grimberg // Create reference Ceed that implementation will be dispatched through 32e5f091ebSnbeams #ifdef CEED_MAGMA_USE_HIP 3358549094SSebastian Grimberg CeedCallBackend(CeedInit("/gpu/hip/magma", &ceed_ref)); 34adb2481bSnbeams #else 3558549094SSebastian Grimberg CeedCallBackend(CeedInit("/gpu/cuda/magma", &ceed_ref)); 36adb2481bSnbeams #endif 3758549094SSebastian Grimberg CeedCallBackend(CeedSetDelegate(ceed, ceed_ref)); 38cb23e90cSJeremy L Thompson 3900fb7a04SSebastian Grimberg CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "Destroy", CeedDestroy_Magma)); 40e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 41cb23e90cSJeremy L Thompson } 42cb23e90cSJeremy L Thompson 4300fb7a04SSebastian Grimberg //------------------------------------------------------------------------------ 4400fb7a04SSebastian Grimberg // Backend Register 4500fb7a04SSebastian Grimberg //------------------------------------------------------------------------------ 461d013790SJed Brown CEED_INTERN int CeedRegister_Magma_Det(void) { 47e5f091ebSnbeams #ifdef CEED_MAGMA_USE_HIP 481d013790SJed Brown return CeedRegister("/gpu/hip/magma/det", CeedInit_Magma_Det, 125); 49adb2481bSnbeams #else 501d013790SJed Brown return CeedRegister("/gpu/cuda/magma/det", CeedInit_Magma_Det, 125); 51adb2481bSnbeams #endif 52cb23e90cSJeremy L Thompson } 5300fb7a04SSebastian Grimberg 5400fb7a04SSebastian Grimberg //------------------------------------------------------------------------------ 55