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 f60c509 commit bf3a3eeCopy full SHA for bf3a3ee
xobjects/context_cupy.py
@@ -419,6 +419,7 @@ def build_kernels(
419
extra_cdef=None,
420
extra_classes=(),
421
extra_headers=(),
422
+ extra_compile_args=(),
423
compile=True, # noqa
424
) -> Dict[Tuple[str, tuple], "KernelCupy"]:
425
if not compile:
@@ -454,7 +455,7 @@ def build_kernels(
454
455
with open(save_source_as, "w") as fid:
456
fid.write(specialized_source)
457
- module = cupy.RawModule(code=specialized_source)
458
+ module = cupy.RawModule(code=specialized_source, options=extra_compile_args)
459
460
out_kernels = {}
461
for pyname, kernel in kernel_descriptions.items():
0 commit comments