1 // Copyright (c) 2017-2026, 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.h>
9 #include <ceed/backend.h>
10 #include <stdbool.h>
11 #include <string.h>
12
13 #include "ceed-avx.h"
14
15 //------------------------------------------------------------------------------
16 // Backend Init
17 //------------------------------------------------------------------------------
CeedInit_Avx(const char * resource,Ceed ceed)18 static int CeedInit_Avx(const char *resource, Ceed ceed) {
19 Ceed ceed_ref;
20
21 CeedCheck(!strcmp(resource, "/cpu/self") || !strcmp(resource, "/cpu/self/avx") || !strcmp(resource, "/cpu/self/avx/blocked"), ceed,
22 CEED_ERROR_BACKEND, "AVX backend cannot use resource: %s", resource);
23 CeedCallBackend(CeedSetDeterministic(ceed, true));
24
25 // Create reference Ceed that implementation will be dispatched through unless overridden
26 CeedCallBackend(CeedInit("/cpu/self/opt/blocked", &ceed_ref));
27 CeedCallBackend(CeedSetDelegate(ceed, ceed_ref));
28 CeedCallBackend(CeedDestroy(&ceed_ref));
29
30 CeedCallBackend(CeedSetBackendFunction(ceed, "Ceed", ceed, "TensorContractCreate", CeedTensorContractCreate_Avx));
31 return CEED_ERROR_SUCCESS;
32 }
33
34 //------------------------------------------------------------------------------
35 // Backend Register
36 //------------------------------------------------------------------------------
CeedRegister_Avx_Blocked(void)37 CEED_INTERN int CeedRegister_Avx_Blocked(void) { return CeedRegister("/cpu/self/avx/blocked", CeedInit_Avx, 30); }
38
39 //------------------------------------------------------------------------------
40