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-avx.h" 13 14 //------------------------------------------------------------------------------ 15 // Backend Init 16 //------------------------------------------------------------------------------ 17 static int CeedInit_Avx(const char *resource, Ceed ceed) { 18 int ierr; 19 if (strcmp(resource, "/cpu/self") 20 && strcmp(resource, "/cpu/self/avx/serial")) 21 // LCOV_EXCL_START 22 return CeedError(ceed, CEED_ERROR_BACKEND, 23 "AVX backend cannot use resource: %s", resource); 24 // LCOV_EXCL_STOP 25 ierr = CeedSetDeterministic(ceed, true); CeedChkBackend(ierr); 26 27 // Create reference CEED that implementation will be dispatched 28 // through unless overridden 29 Ceed ceed_ref; 30 CeedInit("/cpu/self/opt/serial", &ceed_ref); 31 ierr = CeedSetDelegate(ceed, ceed_ref); CeedChkBackend(ierr); 32 33 if (CEED_SCALAR_TYPE == CEED_SCALAR_FP64) { 34 ierr = CeedSetBackendFunction(ceed, "Ceed", ceed, "TensorContractCreate", 35 CeedTensorContractCreate_f64_Avx); 36 CeedChkBackend(ierr); 37 } else { 38 ierr = CeedSetBackendFunction(ceed, "Ceed", ceed, "TensorContractCreate", 39 CeedTensorContractCreate_f32_Avx); 40 CeedChkBackend(ierr); 41 } 42 43 return CEED_ERROR_SUCCESS; 44 } 45 46 //------------------------------------------------------------------------------ 47 // Backend Register 48 //------------------------------------------------------------------------------ 49 CEED_INTERN int CeedRegister_Avx_Serial(void) { 50 return CeedRegister("/cpu/self/avx/serial", CeedInit_Avx, 35); 51 } 52 //------------------------------------------------------------------------------ 53