File tree Expand file tree Collapse file tree 2 files changed +8
-2
lines changed Expand file tree Collapse file tree 2 files changed +8
-2
lines changed Original file line number Diff line number Diff 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 :
Original file line number Diff line number Diff 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 = {}
You can’t perform that action at this time.
0 commit comments