1 // Copyright (c) 2017-2022, Lawrence Livermore National Security, LLC and other CEED contributors. 2 // All Rights Reserved. See the top-level LICENSE and NOTICE files for details. 3 // 4 // SPDX-License-Identifier: BSD-2-Clause 5 // 6 // This file is part of CEED: http://github.com/ceed 7 8 #include <ceed/ceed.h> 9 #include <ceed/backend.h> 10 #include <stdbool.h> 11 #include <string.h> 12 #include "ceed-xsmm.h" 13 14 //------------------------------------------------------------------------------ 15 // Backend Init 16 //------------------------------------------------------------------------------ 17 static int CeedInit_Xsmm_Serial(const char *resource, Ceed ceed) { 18 int ierr; 19 if (strcmp(resource, "/cpu/self") 20 && strcmp(resource, "/cpu/self/xsmm/serial")) 21 // LCOV_EXCL_START 22 return CeedError(ceed, CEED_ERROR_BACKEND, 23 "serial libXSMM backend cannot use resource: %s", 24 resource); 25 // LCOV_EXCL_STOP 26 ierr = CeedSetDeterministic(ceed, true); CeedChkBackend(ierr); 27 28 // Create reference CEED that implementation will be dispatched 29 // through unless overridden 30 Ceed ceed_ref; 31 CeedInit("/cpu/self/opt/serial", &ceed_ref); 32 ierr = CeedSetDelegate(ceed, ceed_ref); CeedChkBackend(ierr); 33 34 if (CEED_SCALAR_TYPE == CEED_SCALAR_FP64) { 35 ierr = CeedSetBackendFunction(ceed, "Ceed", ceed, "TensorContractCreate", 36 CeedTensorContractCreate_f64_Xsmm); 37 CeedChkBackend(ierr); 38 } else { 39 ierr = CeedSetBackendFunction(ceed, "Ceed", ceed, "TensorContractCreate", 40 CeedTensorContractCreate_f32_Xsmm); 41 CeedChkBackend(ierr); 42 } 43 44 return CEED_ERROR_SUCCESS; 45 } 46 47 //------------------------------------------------------------------------------ 48 // Backend Register 49 //------------------------------------------------------------------------------ 50 CEED_INTERN int CeedRegister_Xsmm_Serial(void) { 51 return CeedRegister("/cpu/self/xsmm/serial", CeedInit_Xsmm_Serial, 25); 52 } 53 //------------------------------------------------------------------------------ 54