PetscScalarfy and typdef signed char manually.

Author: sv2518

pyop2/compilation.py

@@ -216,7 +216,7 @@ def workaround_cflags(self):
             if version.StrictVersion("7.3") <= ver <= version.StrictVersion("7.5"):
                 # GCC bug https://gcc.gnu.org/bugzilla/show_bug.cgi?id=90055
                 # See also https://github.com/firedrakeproject/firedrake/issues/1442
-                return  # enable vectorization for paper
+                return []  # enable vectorization for paper
                 # Bug also on skylake with the vectoriser in this
                 # combination (disappears without
                 # -fno-tree-loop-vectorize!)

pyop2/sequential.py

@@ -61,6 +61,7 @@
 from pyop2.utils import cached_property, get_petsc_dir
 from pyop2.configuration import configuration
 from pyop2.codegen.rep2loopy import _PreambleGen
+from pyop2.datatypes import ScalarType
 
 
 def vectorise(wrapper, iname, batch_size):
@@ -75,8 +76,8 @@ def vectorise(wrapper, iname, batch_size):
     # create constant zero vectors
     wrapper = wrapper.copy(target=loopy.CVecTarget())
     kernel = wrapper.root_kernel
-    zeros = loopy.TemporaryVariable("_zeros", shape=loopy.auto, dtype=numpy.float64, read_only=True,
-                                    initializer=numpy.array(0.0, dtype=numpy.float64),
+    zeros = loopy.TemporaryVariable("_zeros", shape=loopy.auto, dtype=ScalarType, read_only=True,
+                                    initializer=numpy.array(0.0, dtype=ScalarType),
                                     address_space=loopy.AddressSpace.GLOBAL, zero_size=batch_size)
     tmps = kernel.temporary_variables.copy()
     tmps["_zeros"] = zeros
@@ -103,9 +104,12 @@ def vectorise(wrapper, iname, batch_size):
     wrapper = wrapper.with_root_kernel(kernel)
 
     # vector data type
-    vec_types = [("double", 8), ("int", 4)]  # scalar type, bytes
-    preamble = ["typedef {0} {0}{1} __attribute__ ((vector_size ({2})));".format(t, batch_size, batch_size * b) for t, b in vec_types]
-    preamble = "\n" + "\n".join(preamble)
+    dw_typedef = ["typedef signed char SC;"]
+    preamble_dw = "\n" + "\n".join(dw_typedef)
+    vec_types = [("double", 8), ("int", 4), ("SC", 1)]  # scalar type, bytes
+    vec_typedef = ["typedef {0} {0}{1} __attribute__ ((vector_size ({2})));".format(t, batch_size, batch_size * b) for t, b in vec_types]
+    preamble_vec = "\n" + "\n".join(vec_typedef)
+    preamble = preamble_dw + preamble_vec
 
     wrapper = loopy.register_preamble_generators(wrapper, [_PreambleGen(preamble, idx="01")])
     return wrapper