Skip to content

Commit bf3a3ee

Browse files
Pass options to cupy
1 parent f60c509 commit bf3a3ee

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

xobjects/context_cupy.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -419,6 +419,7 @@ def build_kernels(
419419
extra_cdef=None,
420420
extra_classes=(),
421421
extra_headers=(),
422+
extra_compile_args=(),
422423
compile=True, # noqa
423424
) -> Dict[Tuple[str, tuple], "KernelCupy"]:
424425
if not compile:
@@ -454,7 +455,7 @@ def build_kernels(
454455
with open(save_source_as, "w") as fid:
455456
fid.write(specialized_source)
456457

457-
module = cupy.RawModule(code=specialized_source)
458+
module = cupy.RawModule(code=specialized_source, options=extra_compile_args)
458459

459460
out_kernels = {}
460461
for pyname, kernel in kernel_descriptions.items():

0 commit comments

Comments
 (0)