Skip to content

Commit 1d3b822

Browse files
Add compile flags to Pyopencl context
1 parent 2ff2cd4 commit 1d3b822

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

xobjects/context_cupy.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -416,10 +416,10 @@ def build_kernels(
416416
specialize=True,
417417
apply_to_source=(),
418418
save_source_as=None,
419+
extra_compile_args=(),
419420
extra_cdef=None,
420421
extra_classes=(),
421422
extra_headers=(),
422-
extra_compile_args=(),
423423
compile=True, # noqa
424424
) -> Dict[Tuple[str, tuple], "KernelCupy"]:
425425
if not compile:

xobjects/context_pyopencl.py

Lines changed: 7 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -180,6 +180,7 @@ def build_kernels(
180180
specialize=True,
181181
apply_to_source=(),
182182
save_source_as=None,
183+
extra_compile_args=(),
183184
extra_cdef=None,
184185
extra_classes=(),
185186
extra_headers=(),
@@ -218,8 +219,13 @@ def build_kernels(
218219
with open(save_source_as, "w") as fid:
219220
fid.write(specialized_source)
220221

222+
extra_compile_args = (
223+
*extra_compile_args,
224+
"-cl-std=CL2.0",
225+
"-DXO_CONTEXT_CL",
226+
)
221227
prg = cl.Program(self.context, specialized_source).build(
222-
options="-cl-std=CL2.0",
228+
options=extra_compile_args,
223229
)
224230

225231
out_kernels = {}

0 commit comments

Comments
 (0)