omni

ecmwf-ifs · Jan 17, 2024 · e3234d8 · e3234d8
1 parent ad91a28
commit e3234d8
Showing 1 changed file with 10 additions and 4 deletions.
diff --git a/transformations/tests/test_raw_stack_allocator.py b/transformations/tests/test_raw_stack_allocator.py
@@ -436,11 +436,17 @@ def test_raw_stack_allocator_temporaries(frontend, block_dim, horizontal, direct
         pragmas = FindNodes(Pragma).visit(driver.body)
 
         if directive == 'openacc':
-            assert pragmas[0].content.lower() == 'data create(z_jprb_stack, '\
-                                                 'z_selected_real_kind_13_300_stack, ll_stack)'
+            if frontend == OMNI:
+                assert pragmas[0].content.lower() == 'data create(z_selected_real_kind_13_300_stack, ll_stack)'
+            else:
+                assert pragmas[0].content.lower() == 'data create(z_jprb_stack, '\
+                                                     'z_selected_real_kind_13_300_stack, ll_stack)'
 
         if directive == 'openmp':
-            assert pragmas[0].content.lower() == 'target allocate(z_jprb_stack, '\
-                                                 'z_selected_real_kind_13_300_stack, ll_stack)'
+            if frontend == OMNI:
+                assert pragmas[0].content.lower() == 'target allocate(z_selected_real_kind_13_300_stack, ll_stack)'
+            else:
+                assert pragmas[0].content.lower() == 'target allocate(z_jprb_stack, '\
+                                                     'z_selected_real_kind_13_300_stack, ll_stack)'
 
     rmtree(basedir)