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 13*00fb7a04SSebastian Grimberg #include "ceed-magma-common.h" 14cb23e90cSJeremy L Thompson 15*00fb7a04SSebastian Grimberg //------------------------------------------------------------------------------ 16*00fb7a04SSebastian Grimberg // Backend Init 17*00fb7a04SSebastian Grimberg //------------------------------------------------------------------------------ 186574a04fSJeremy L Thompson static int CeedInit_Magma_Det(const char *resource, Ceed ceed) { 196dbfb411Snbeams const int nrc = 18; // number of characters in resource 206574a04fSJeremy L Thompson CeedCheck(!strncmp(resource, "/gpu/cuda/magma/det", nrc) || !strncmp(resource, "/gpu/hip/magma/det", nrc), ceed, CEED_ERROR_BACKEND, 216574a04fSJeremy L Thompson "Magma backend cannot use resource: %s", resource); 222b730f8bSJeremy L Thompson CeedCallBackend(CeedSetDeterministic(ceed, true)); 23cb23e90cSJeremy L Thompson 246dbfb411Snbeams Ceed_Magma *data; 2558549094SSebastian Grimberg CeedCallBackend(CeedCalloc(1, &data)); 262b730f8bSJeremy L Thompson CeedCallBackend(CeedSetData(ceed, data)); 27*00fb7a04SSebastian Grimberg CeedCallBackend(CeedInit_Magma_common(ceed, resource)); 286dbfb411Snbeams 2958549094SSebastian Grimberg // Create reference Ceed that implementation will be dispatched through 3058549094SSebastian Grimberg Ceed ceed_ref; 31e5f091ebSnbeams #ifdef CEED_MAGMA_USE_HIP 3258549094SSebastian Grimberg CeedCallBackend(CeedInit("/gpu/hip/magma", &ceed_ref)); 33adb2481bSnbeams #else 3458549094SSebastian Grimberg CeedCallBackend(CeedInit("/gpu/cuda/magma", &ceed_ref)); 35adb2481bSnbeams #endif 3658549094SSebastian Grimberg CeedCallBackend(CeedSetDelegate(ceed, ceed_ref)); 37cb23e90cSJeremy L Thompson 38*00fb7a04SSebastian Grimberg CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "Destroy", CeedDestroy_Magma)); 39*00fb7a04SSebastian Grimberg 40e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 41cb23e90cSJeremy L Thompson } 42cb23e90cSJeremy L Thompson 43*00fb7a04SSebastian Grimberg //------------------------------------------------------------------------------ 44*00fb7a04SSebastian Grimberg // Backend Register 45*00fb7a04SSebastian 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 } 53*00fb7a04SSebastian Grimberg 54*00fb7a04SSebastian Grimberg //------------------------------------------------------------------------------ 55