We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 877a93a commit 034283eCopy full SHA for 034283e
pyop2/sequential.py
@@ -162,9 +162,11 @@ def code_to_compile(self):
162
iname = "n"
163
has_matrix = any(arg._is_mat for arg in self._args)
164
has_rw = any(arg.access == RW for arg in self._args)
165
+ is_cplx = any(arg.dtype.name == 'complex128' for arg in self._args)
166
if isinstance(self._kernel.code, loopy.LoopKernel) and not (has_matrix or has_rw):
167
wrapper = loopy.inline_callable_kernel(wrapper, self._kernel.name)
- wrapper = vectorise(wrapper, iname, configuration["simd_width"])
168
+ if not is_cplx:
169
+ wrapper = vectorise(wrapper, iname, configuration["simd_width"])
170
code = loopy.generate_code_v2(wrapper)
171
172
if self._kernel._cpp:
0 commit comments