ceed - set JiT source dirs on fallback ceed

CEED · Oct 18, 2024 · 8c5f31d · 8c5f31d
1 parent b5030a2
commit 8c5f31d
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/interface/ceed.c b/interface/ceed.c
@@ -658,6 +658,15 @@ int CeedGetOperatorFallbackCeed(Ceed ceed, Ceed *fallback_ceed) {
     fallback_ceed->op_fallback_parent = ceed;
     fallback_ceed->Error              = ceed->Error;
     ceed->op_fallback_ceed            = fallback_ceed;
+  {
+    const char **jit_source_dirs;
+    CeedInt      num_jit_source_dirs = 0;
+
+    CeedCall(CeedGetJitSourceRoots(ceed, &num_jit_source_dirs, &jit_source_dirs));
+    for (CeedInt i = 0; i < num_jit_source_dirs; i++) {
+       CeedCall(CeedAddJitSourceRoot(fallback_ceed, jit_source_dirs[i]);));
+    }
+    CeedCall(CeedRestoreJitSourceRoots(ceed, &jit_source_dirs));
   }
   *fallback_ceed = ceed->op_fallback_ceed;
   return CEED_ERROR_SUCCESS;