17d8d0e25Snbeams // Copyright (c) 2017-2018, Lawrence Livermore National Security, LLC. 27d8d0e25Snbeams // Produced at the Lawrence Livermore National Laboratory. LLNL-CODE-734707. 37d8d0e25Snbeams // All Rights reserved. See files LICENSE and NOTICE for details. 47d8d0e25Snbeams // 57d8d0e25Snbeams // This file is part of CEED, a collection of benchmarks, miniapps, software 67d8d0e25Snbeams // libraries and APIs for efficient high-order finite element and spectral 77d8d0e25Snbeams // element discretizations for exascale applications. For more information and 87d8d0e25Snbeams // source code availability see http://github.com/ceed. 97d8d0e25Snbeams // 107d8d0e25Snbeams // The CEED research is supported by the Exascale Computing Project 17-SC-20-SC, 117d8d0e25Snbeams // a collaborative effort of two U.S. Department of Energy organizations (Office 127d8d0e25Snbeams // of Science and the National Nuclear Security Administration) responsible for 137d8d0e25Snbeams // the planning and preparation of a capable exascale ecosystem, including 147d8d0e25Snbeams // software, applications, hardware, advanced system engineering and early 157d8d0e25Snbeams // testbed platforms, in support of the nation's exascale computing imperative. 167d8d0e25Snbeams 17ec3da8bcSJed Brown #include <ceed/ceed.h> 18ec3da8bcSJed Brown #include <ceed/backend.h> 193d576824SJeremy L Thompson #include <hip/hip_runtime.h> 207d8d0e25Snbeams #include <stdio.h> 213d576824SJeremy L Thompson #include <string.h> 227d8d0e25Snbeams #include "ceed-hip-gen.h" 233d576824SJeremy L Thompson #include "../hip/ceed-hip.h" 247d8d0e25Snbeams 257d8d0e25Snbeams //------------------------------------------------------------------------------ 267d8d0e25Snbeams // Apply QFunction 277d8d0e25Snbeams //------------------------------------------------------------------------------ 287d8d0e25Snbeams static int CeedQFunctionApply_Hip_gen(CeedQFunction qf, CeedInt Q, 297d8d0e25Snbeams CeedVector *U, CeedVector *V) { 307d8d0e25Snbeams int ierr; 317d8d0e25Snbeams Ceed ceed; 32e15f9bd0SJeremy L Thompson ierr = CeedQFunctionGetCeed(qf, &ceed); CeedChkBackend(ierr); 33e15f9bd0SJeremy L Thompson return CeedError(ceed, CEED_ERROR_BACKEND, 34e15f9bd0SJeremy L Thompson "Backend does not implement QFunctionApply"); 357d8d0e25Snbeams } 367d8d0e25Snbeams 377d8d0e25Snbeams //------------------------------------------------------------------------------ 387d8d0e25Snbeams // Destroy QFunction 397d8d0e25Snbeams //------------------------------------------------------------------------------ 407d8d0e25Snbeams static int CeedQFunctionDestroy_Hip_gen(CeedQFunction qf) { 417d8d0e25Snbeams int ierr; 427d8d0e25Snbeams CeedQFunction_Hip_gen *data; 43e15f9bd0SJeremy L Thompson ierr = CeedQFunctionGetData(qf, &data); CeedChkBackend(ierr); 447d8d0e25Snbeams Ceed ceed; 45e15f9bd0SJeremy L Thompson ierr = CeedQFunctionGetCeed(qf, &ceed); CeedChkBackend(ierr); 467d8d0e25Snbeams ierr = hipFree(data->d_c); CeedChk_Hip(ceed, ierr); 47e15f9bd0SJeremy L Thompson ierr = CeedFree(&data->qFunctionSource); CeedChkBackend(ierr); 48e15f9bd0SJeremy L Thompson ierr = CeedFree(&data); CeedChkBackend(ierr); 49e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 507d8d0e25Snbeams } 517d8d0e25Snbeams 527d8d0e25Snbeams //------------------------------------------------------------------------------ 537d8d0e25Snbeams // Create QFunction 547d8d0e25Snbeams //------------------------------------------------------------------------------ 557d8d0e25Snbeams int CeedQFunctionCreate_Hip_gen(CeedQFunction qf) { 567d8d0e25Snbeams int ierr; 577d8d0e25Snbeams Ceed ceed; 587d8d0e25Snbeams CeedQFunctionGetCeed(qf, &ceed); 597d8d0e25Snbeams CeedQFunction_Hip_gen *data; 60e15f9bd0SJeremy L Thompson ierr = CeedCalloc(1, &data); CeedChkBackend(ierr); 61e15f9bd0SJeremy L Thompson ierr = CeedQFunctionSetData(qf, data); CeedChkBackend(ierr); 627d8d0e25Snbeams 63*3d3250a0SJeremy L Thompson // Read QFunction source 6443e1b16fSJeremy L Thompson ierr = CeedQFunctionGetKernelName(qf, &data->qFunctionName); 6543e1b16fSJeremy L Thompson CeedChkBackend(ierr); 66*3d3250a0SJeremy L Thompson ierr = CeedQFunctionLoadSourceToBuffer(qf, &data->qFunctionSource); 67*3d3250a0SJeremy L Thompson CeedChkBackend(ierr); 68*3d3250a0SJeremy L Thompson if (!data->qFunctionSource) 69*3d3250a0SJeremy L Thompson // LCOV_EXCL_START 70*3d3250a0SJeremy L Thompson return CeedError(ceed, CEED_ERROR_UNSUPPORTED, 71*3d3250a0SJeremy L Thompson "/gpu/hip/gen backend requires QFunction source code file"); 72*3d3250a0SJeremy L Thompson // LCOV_EXCL_STOP 737d8d0e25Snbeams 747d8d0e25Snbeams ierr = CeedSetBackendFunction(ceed, "QFunction", qf, "Apply", 75e15f9bd0SJeremy L Thompson CeedQFunctionApply_Hip_gen); CeedChkBackend(ierr); 767d8d0e25Snbeams ierr = CeedSetBackendFunction(ceed, "QFunction", qf, "Destroy", 77e15f9bd0SJeremy L Thompson CeedQFunctionDestroy_Hip_gen); CeedChkBackend(ierr); 78e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 797d8d0e25Snbeams } 807d8d0e25Snbeams //------------------------------------------------------------------------------ 81