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. 389c6efa4Sjeremylt // 43d8e8822SJeremy L Thompson // SPDX-License-Identifier: BSD-2-Clause 589c6efa4Sjeremylt // 63d8e8822SJeremy L Thompson // This file is part of CEED: http://github.com/ceed 789c6efa4Sjeremylt 849aac155SJeremy L Thompson #include <ceed.h> 9ec3da8bcSJed Brown #include <ceed/backend.h> 103d576824SJeremy L Thompson #include <stdbool.h> 1189c6efa4Sjeremylt #include <string.h> 122b730f8bSJeremy L Thompson 1389c6efa4Sjeremylt #include "ceed-opt.h" 1489c6efa4Sjeremylt 15f10650afSjeremylt //------------------------------------------------------------------------------ 16f10650afSjeremylt // Backend Destroy 17f10650afSjeremylt //------------------------------------------------------------------------------ 1889c6efa4Sjeremylt static int CeedDestroy_Opt(Ceed ceed) { 1989c6efa4Sjeremylt Ceed_Opt *data; 20ad70ee2cSJeremy L Thompson 212b730f8bSJeremy L Thompson CeedCallBackend(CeedGetData(ceed, &data)); 222b730f8bSJeremy L Thompson CeedCallBackend(CeedFree(&data)); 23e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 2489c6efa4Sjeremylt } 2589c6efa4Sjeremylt 26f10650afSjeremylt //------------------------------------------------------------------------------ 27f10650afSjeremylt // Backend Init 28f10650afSjeremylt //------------------------------------------------------------------------------ 2989c6efa4Sjeremylt static int CeedInit_Opt_Blocked(const char *resource, Ceed ceed) { 30ad70ee2cSJeremy L Thompson Ceed ceed_ref; 31ad70ee2cSJeremy L Thompson Ceed_Opt *data; 32ad70ee2cSJeremy L Thompson 336574a04fSJeremy L Thompson CeedCheck(!strcmp(resource, "/cpu/self") || !strcmp(resource, "/cpu/self/opt") || !strcmp(resource, "/cpu/self/opt/blocked"), ceed, 346574a04fSJeremy L Thompson CEED_ERROR_BACKEND, "Opt backend cannot use resource: %s", resource); 352b730f8bSJeremy L Thompson CeedCallBackend(CeedSetDeterministic(ceed, true)); 3689c6efa4Sjeremylt 37ea61e9acSJeremy L Thompson // Create reference Ceed that implementation will be dispatched through unless overridden 382b730f8bSJeremy L Thompson CeedCallBackend(CeedInit("/cpu/self/ref/serial", &ceed_ref)); 392b730f8bSJeremy L Thompson CeedCallBackend(CeedSetDelegate(ceed, ceed_ref)); 4089c6efa4Sjeremylt 412b730f8bSJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "Destroy", CeedDestroy_Opt)); 423bf1f308SJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "TensorContractCreate", CeedTensorContractCreate_Opt)); 432b730f8bSJeremy L Thompson CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "OperatorCreate", CeedOperatorCreate_Opt)); 4489c6efa4Sjeremylt 4589c6efa4Sjeremylt // Set block size 462b730f8bSJeremy L Thompson CeedCallBackend(CeedCalloc(1, &data)); 47ad70ee2cSJeremy L Thompson data->block_size = 8; 482b730f8bSJeremy L Thompson CeedCallBackend(CeedSetData(ceed, data)); 49e15f9bd0SJeremy L Thompson return CEED_ERROR_SUCCESS; 5089c6efa4Sjeremylt } 5189c6efa4Sjeremylt 52f10650afSjeremylt //------------------------------------------------------------------------------ 53f10650afSjeremylt // Backend Register 54f10650afSjeremylt //------------------------------------------------------------------------------ 552b730f8bSJeremy L Thompson CEED_INTERN int CeedRegister_Opt_Blocked(void) { return CeedRegister("/cpu/self/opt/blocked", CeedInit_Opt_Blocked, 40); } 562a86cc9dSSebastian Grimberg 57f10650afSjeremylt //------------------------------------------------------------------------------ 58