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. 37d8d0e25Snbeams // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 57d8d0e25Snbeams // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 77d8d0e25Snbeams 849aac155SJeremy L Thompson #include <ceed.h> 9ec3da8bcSJed Brown #include <ceed/backend.h> 103d576824SJeremy L Thompson #include <hip/hip_runtime.h> 112b730f8bSJeremy L Thompson 1249aac155SJeremy L Thompson #include "../hip/ceed-hip-common.h" 137d8d0e25Snbeams #include "ceed-hip-gen.h" 147d8d0e25Snbeams 157d8d0e25Snbeams //------------------------------------------------------------------------------ 167d8d0e25Snbeams // Apply QFunction 177d8d0e25Snbeams //------------------------------------------------------------------------------ 182b730f8bSJeremy L Thompson static int CeedQFunctionApply_Hip_gen(CeedQFunction qf, CeedInt Q, CeedVector *U, CeedVector *V) { 197d8d0e25Snbeams Ceed ceed; 20b7453713SJeremy L Thompson 212b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionGetCeed(qf, &ceed)); 222b730f8bSJeremy L Thompson return CeedError(ceed, CEED_ERROR_BACKEND, "Backend does not implement QFunctionApply"); 237d8d0e25Snbeams } 247d8d0e25Snbeams 257d8d0e25Snbeams //------------------------------------------------------------------------------ 267d8d0e25Snbeams // Destroy QFunction 277d8d0e25Snbeams //------------------------------------------------------------------------------ 287d8d0e25Snbeams static int CeedQFunctionDestroy_Hip_gen(CeedQFunction qf) { 297d8d0e25Snbeams Ceed ceed; 30b7453713SJeremy L Thompson CeedQFunction_Hip_gen *data; 31b7453713SJeremy L Thompson 32b7453713SJeremy L Thompson CeedCallBackend(CeedQFunctionGetData(qf, &data)); 332b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionGetCeed(qf, &ceed)); 342b730f8bSJeremy L Thompson CeedCallHip(ceed, hipFree(data->d_c)); 35*09095acaSJeremy L Thompson CeedCallBackend(CeedFree(&data->qfunction_source)); 362b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&data)); 37e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 387d8d0e25Snbeams } 397d8d0e25Snbeams 407d8d0e25Snbeams //------------------------------------------------------------------------------ 417d8d0e25Snbeams // Create QFunction 427d8d0e25Snbeams //------------------------------------------------------------------------------ 437d8d0e25Snbeams int CeedQFunctionCreate_Hip_gen(CeedQFunction qf) { 447d8d0e25Snbeams Ceed ceed; 457d8d0e25Snbeams CeedQFunction_Hip_gen *data; 46b7453713SJeremy L Thompson 47b7453713SJeremy L Thompson CeedQFunctionGetCeed(qf, &ceed); 482b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(1, &data)); 492b730f8bSJeremy L Thompson CeedCallBackend(CeedQFunctionSetData(qf, data)); 507d8d0e25Snbeams 513d3250a0SJeremy L Thompson // Read QFunction source 52*09095acaSJeremy L Thompson CeedCallBackend(CeedQFunctionGetKernelName(qf, &data->qfunction_name)); 5323d4529eSJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_SUCCESS, "----- Loading QFunction User Source -----\n"); 54*09095acaSJeremy L Thompson CeedCallBackend(CeedQFunctionLoadSourceToBuffer(qf, &data->qfunction_source)); 5523d4529eSJeremy L Thompson CeedDebug256(ceed, CEED_DEBUG_COLOR_SUCCESS, "----- Loading QFunction User Source Complete! -----\n"); 56*09095acaSJeremy L Thompson CeedCheck(data->qfunction_source, ceed, CEED_ERROR_UNSUPPORTED, "/gpu/hip/gen backend requires QFunction source code file"); 577d8d0e25Snbeams 582b730f8bSJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "QFunction", qf, "Apply", CeedQFunctionApply_Hip_gen)); 592b730f8bSJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "QFunction", qf, "Destroy", CeedQFunctionDestroy_Hip_gen)); 60e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 617d8d0e25Snbeams } 622a86cc9dSSebastian Grimberg 637d8d0e25Snbeams //------------------------------------------------------------------------------ 64