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" 237d8d0e25Snbeams 247d8d0e25Snbeams //------------------------------------------------------------------------------ 257d8d0e25Snbeams // Apply QFunction 267d8d0e25Snbeams //------------------------------------------------------------------------------ 277d8d0e25Snbeams static int CeedQFunctionApply_Hip_gen(CeedQFunction qf, CeedInt Q, 287d8d0e25Snbeams CeedVector *U, CeedVector *V) { 297d8d0e25Snbeams int ierr; 307d8d0e25Snbeams Ceed ceed; 31e15f9bd0SJeremy L Thompson ierr = CeedQFunctionGetCeed(qf, &ceed); CeedChkBackend(ierr); 32e15f9bd0SJeremy L Thompson return CeedError(ceed, CEED_ERROR_BACKEND, 33e15f9bd0SJeremy L Thompson "Backend does not implement QFunctionApply"); 347d8d0e25Snbeams } 357d8d0e25Snbeams 367d8d0e25Snbeams //------------------------------------------------------------------------------ 377d8d0e25Snbeams // Destroy QFunction 387d8d0e25Snbeams //------------------------------------------------------------------------------ 397d8d0e25Snbeams static int CeedQFunctionDestroy_Hip_gen(CeedQFunction qf) { 407d8d0e25Snbeams int ierr; 417d8d0e25Snbeams CeedQFunction_Hip_gen *data; 42e15f9bd0SJeremy L Thompson ierr = CeedQFunctionGetData(qf, &data); CeedChkBackend(ierr); 437d8d0e25Snbeams Ceed ceed; 44e15f9bd0SJeremy L Thompson ierr = CeedQFunctionGetCeed(qf, &ceed); CeedChkBackend(ierr); 457d8d0e25Snbeams ierr = hipFree(data->d_c); CeedChk_Hip(ceed, ierr); 46e15f9bd0SJeremy L Thompson ierr = CeedFree(&data->qFunctionSource); CeedChkBackend(ierr); 47e15f9bd0SJeremy L Thompson ierr = CeedFree(&data); CeedChkBackend(ierr); 48e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 497d8d0e25Snbeams } 507d8d0e25Snbeams 517d8d0e25Snbeams //------------------------------------------------------------------------------ 527d8d0e25Snbeams // Create QFunction 537d8d0e25Snbeams //------------------------------------------------------------------------------ 547d8d0e25Snbeams int CeedQFunctionCreate_Hip_gen(CeedQFunction qf) { 557d8d0e25Snbeams int ierr; 567d8d0e25Snbeams Ceed ceed; 577d8d0e25Snbeams CeedQFunctionGetCeed(qf, &ceed); 587d8d0e25Snbeams CeedQFunction_Hip_gen *data; 59e15f9bd0SJeremy L Thompson ierr = CeedCalloc(1, &data); CeedChkBackend(ierr); 60e15f9bd0SJeremy L Thompson ierr = CeedQFunctionSetData(qf, data); CeedChkBackend(ierr); 617d8d0e25Snbeams 623d3250a0SJeremy L Thompson // Read QFunction source 6343e1b16fSJeremy L Thompson ierr = CeedQFunctionGetKernelName(qf, &data->qFunctionName); 6443e1b16fSJeremy L Thompson CeedChkBackend(ierr); 65*46dc0734SJeremy L Thompson CeedDebug256(ceed, 2, "----- Loading QFunction User Source -----\n"); 663d3250a0SJeremy L Thompson ierr = CeedQFunctionLoadSourceToBuffer(qf, &data->qFunctionSource); 673d3250a0SJeremy L Thompson CeedChkBackend(ierr); 68*46dc0734SJeremy L Thompson CeedDebug256(ceed, 2, "----- Loading QFunction User Source Complete! -----\n"); 693d3250a0SJeremy L Thompson if (!data->qFunctionSource) 703d3250a0SJeremy L Thompson // LCOV_EXCL_START 713d3250a0SJeremy L Thompson return CeedError(ceed, CEED_ERROR_UNSUPPORTED, 723d3250a0SJeremy L Thompson "/gpu/hip/gen backend requires QFunction source code file"); 733d3250a0SJeremy L Thompson // LCOV_EXCL_STOP 747d8d0e25Snbeams 757d8d0e25Snbeams ierr = CeedSetBackendFunction(ceed, "QFunction", qf, "Apply", 76e15f9bd0SJeremy L Thompson CeedQFunctionApply_Hip_gen); CeedChkBackend(ierr); 777d8d0e25Snbeams ierr = CeedSetBackendFunction(ceed, "QFunction", qf, "Destroy", 78e15f9bd0SJeremy L Thompson CeedQFunctionDestroy_Hip_gen); CeedChkBackend(ierr); 79e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 807d8d0e25Snbeams } 817d8d0e25Snbeams //------------------------------------------------------------------------------ 82