Branch Elimination + Construction Utils

dace/codegen/codegen.py

@@ -172,23 +172,38 @@ def generate_code(sdfg: SDFG, validate=True) -> List[CodeObject]:
         import filecmp
         import shutil
         import tempfile
-        with tempfile.TemporaryDirectory() as tmp_dir:
-            sdfg.save(f'{tmp_dir}/test.sdfg', hash=False)
-            sdfg2 = SDFG.from_file(f'{tmp_dir}/test.sdfg')
-            sdfg2.save(f'{tmp_dir}/test2.sdfg', hash=False)
+        import os
+        with tempfile.NamedTemporaryFile(suffix="_.sdfg", delete=False) as tmp1, \
+            tempfile.NamedTemporaryFile(suffix="_.sdfg", delete=False) as tmp2:
+            tmp1_path = tmp1.name
+            tmp2_path = tmp2.name
+
+        try:
+            sdfg.save(tmp1_path, hash=False)
+            sdfg2 = SDFG.from_file(tmp1_path)
+            sdfg2.save(tmp2_path, hash=False)
+
             print('Testing SDFG serialization...')
-            if not filecmp.cmp(f'{tmp_dir}/test.sdfg', f'{tmp_dir}/test2.sdfg'):
-                with open(f'{tmp_dir}/test.sdfg', 'r') as f1:
-                    with open(f'{tmp_dir}/test2.sdfg', 'r') as f2:
-                        diff = difflib.unified_diff(f1.readlines(),
-                                                    f2.readlines(),
-                                                    fromfile='test.sdfg  (first save)',
-                                                    tofile='test2.sdfg (after roundtrip)')
+            if not filecmp.cmp(tmp1_path, tmp2_path):
+                with open(tmp1_path, 'r') as f1, open(tmp2_path, 'r') as f2:
+                    diff = difflib.unified_diff(f1.readlines(),
+                                                f2.readlines(),
+                                                fromfile='test.sdfg  (first save)',
+                                                tofile='test2.sdfg (after roundtrip)')
                 diff = ''.join(diff)
-                shutil.move(f'{tmp_dir}/test.sdfg', 'test.sdfg')
-                shutil.move(f'{tmp_dir}/test2.sdfg', 'test2.sdfg')
+
+                shutil.copy(tmp1_path, 'test.sdfg')
+                shutil.copy(tmp2_path, 'test2.sdfg')
                 raise RuntimeError(f'SDFG serialization failed - files do not match:\n{diff}')
 
+        finally:
+            # Clean up the temporary files
+            try:
+                os.remove(tmp1_path)
+                os.remove(tmp2_path)
+            except OSError:
+                pass
+
     if config.Config.get_bool('optimizer', 'detect_control_flow'):
         # NOTE: This should likely be done either earlier in the future, or changed entirely in modular codegen.
         # It is being done here to ensure that for now the semantics of the setting are preserved and legacy tests,

dace/config_schema.yml

@@ -261,7 +261,7 @@ required:
                         type: str
                         title: Arguments
                         description: Compiler argument flags
-                        default: '-std=c++14 -fPIC -Wall -Wextra -O3 -march=native -ffast-math -Wno-unused-parameter -Wno-unused-label'
+                        default: '-fopenmp -fPIC -Wall -Wextra -O3 -march=native -ffast-math -Wno-unused-parameter -Wno-unused-label'
                         default_Windows: '/O2 /fp:fast /arch:AVX2 /D_USRDLL /D_WINDLL /D__restrict__=__restrict'
 
                     libs:
@@ -303,7 +303,7 @@ required:
                         type: str
                         title: nvcc Arguments
                         description: Compiler argument flags for CUDA
-                        default: '-Xcompiler -march=native --use_fast_math -Xcompiler -Wno-unused-parameter'
+                        default: '--expt-relaxed-constexpr -Xcompiler -march=native --use_fast_math -Xcompiler -Wno-unused-parameter'
                         default_Windows: '-O3 --use_fast_math'
 
                     hip_args:

dace/dtypes.py

@@ -1267,6 +1267,9 @@ def isconstant(var):
 string = stringtype()
 MPI_Request = opaque('MPI_Request')
 
+FLOAT_TYPES = {float64, float32, float16}
+INT_TYPES = {int8, int16, int32, int64, uintp, uint8, uint16, uint32, uint64}
+
 
 @undefined_safe_enum
 @extensible_enum