Skip to content

Commit f417ff9

Browse files
Fix typo in context_cupy
1 parent a76e63b commit f417ff9

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

xobjects/context_cupy.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -455,10 +455,9 @@ def build_kernels(
455455
with open(save_source_as, "w") as fid:
456456
fid.write(specialized_source)
457457

458-
extra_compile_args = (*extra_compile_args, "-DXO_CONTEXT_CUDA")
459458
extra_include_paths = self.get_installed_c_source_paths()
460459
include_flags = [f"-I{path}" for path in extra_include_paths]
461-
xtr_compile_args.extend(include_flags)
460+
extra_compile_args = (*extra_compile_args, *include_flags, "-DXO_CONTEXT_CUDA")
462461

463462
module = cupy.RawModule(
464463
code=specialized_source, options=extra_compile_args

0 commit comments

Comments
 (0)