@@ -41,6 +41,15 @@ def benchmarks(self) -> list[Benchmark]:
4141 if options .sycl is None :
4242 return []
4343
44+ if options .ur_adapter == "cuda" :
45+ return [
46+ Hashtable (self ),
47+ Bitcracker (self ),
48+ CudaSift (self ),
49+ QuickSilver (self ),
50+ SobelFilter (self ),
51+ ]
52+
4453 return [
4554 Hashtable (self ),
4655 Bitcracker (self ),
@@ -66,6 +75,8 @@ def download_deps(self):
6675 return
6776
6877 def extra_cmake_args (self ) -> list [str ]:
78+ if options .ur_adapter == "cuda" :
79+ return [f"-DUSE_NVIDIA_BACKEND=YES" , f"-DUSE_SM=80" ]
6980 return []
7081
7182 def ld_libraries (self ) -> list [str ]:
@@ -358,6 +369,12 @@ def download_deps(self):
358369
359370 def extra_cmake_args (self ):
360371 oneapi = get_oneapi ()
372+ if options .ur_adapter == "cuda" :
373+ return [
374+ f"-DUSE_NVIDIA_BACKEND=YES" ,
375+ f"-DUSE_SM=80" ,
376+ f"-DCMAKE_CXX_FLAGS=-O3 -fsycl -ffast-math -I{ oneapi .dnn_include ()} -I{ oneapi .mkl_include ()} -L{ oneapi .dnn_lib ()} -L{ oneapi .mkl_lib ()} " ,
377+ ]
361378 return [
362379 f"-DCMAKE_CXX_FLAGS=-O3 -fsycl -ffast-math -I{ oneapi .dnn_include ()} -I{ oneapi .mkl_include ()} -L{ oneapi .dnn_lib ()} -L{ oneapi .mkl_lib ()} "
363380 ]
@@ -415,6 +432,12 @@ def download_deps(self):
415432
416433 def extra_cmake_args (self ):
417434 oneapi = get_oneapi ()
435+ if options .ur_adapter == "cuda" :
436+ return [
437+ f"-DUSE_NVIDIA_BACKEND=YES" ,
438+ f"-DUSE_SM=80" ,
439+ f"-DCMAKE_CXX_FLAGS=-O3 -fsycl -ffast-math -I{ oneapi .dnn_include ()} -I{ oneapi .mkl_include ()} -L{ oneapi .dnn_lib ()} -L{ oneapi .mkl_lib ()} " ,
440+ ]
418441 return [
419442 f"-DCMAKE_CXX_FLAGS=-O3 -fsycl -ffast-math -I{ oneapi .dnn_include ()} -I{ oneapi .mkl_include ()} -L{ oneapi .dnn_lib ()} -L{ oneapi .mkl_lib ()} "
420443 ]
@@ -452,6 +475,12 @@ def ld_libraries(self):
452475
453476 def extra_cmake_args (self ):
454477 oneapi = get_oneapi ()
478+ if options .ur_adapter == "cuda" :
479+ return [
480+ f"-DUSE_NVIDIA_BACKEND=YES" ,
481+ f"-DUSE_SM=80" ,
482+ f"-DCMAKE_CXX_FLAGS=-O3 -fsycl -ffast-math -I{ oneapi .dnn_include ()} -I{ oneapi .mkl_include ()} -L{ oneapi .dnn_lib ()} -L{ oneapi .mkl_lib ()} " ,
483+ ]
455484 return [
456485 f"-DCMAKE_CXX_FLAGS=-O3 -fsycl -ffast-math -I{ oneapi .dnn_include ()} -I{ oneapi .mkl_include ()} -L{ oneapi .dnn_lib ()} -L{ oneapi .mkl_lib ()} "
457486 ]
0 commit comments