@@ -41,6 +41,15 @@ def benchmarks(self) -> list[Benchmark]:
41
41
if options .sycl is None :
42
42
return []
43
43
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
+
44
53
return [
45
54
Hashtable (self ),
46
55
Bitcracker (self ),
@@ -66,6 +75,8 @@ def download_deps(self):
66
75
return
67
76
68
77
def extra_cmake_args (self ) -> list [str ]:
78
+ if options .ur_adapter == "cuda" :
79
+ return [f"-DUSE_NVIDIA_BACKEND=YES" , f"-DUSE_SM=80" ]
69
80
return []
70
81
71
82
def ld_libraries (self ) -> list [str ]:
@@ -358,6 +369,12 @@ def download_deps(self):
358
369
359
370
def extra_cmake_args (self ):
360
371
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
+ ]
361
378
return [
362
379
f"-DCMAKE_CXX_FLAGS=-O3 -fsycl -ffast-math -I{ oneapi .dnn_include ()} -I{ oneapi .mkl_include ()} -L{ oneapi .dnn_lib ()} -L{ oneapi .mkl_lib ()} "
363
380
]
@@ -415,6 +432,12 @@ def download_deps(self):
415
432
416
433
def extra_cmake_args (self ):
417
434
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
+ ]
418
441
return [
419
442
f"-DCMAKE_CXX_FLAGS=-O3 -fsycl -ffast-math -I{ oneapi .dnn_include ()} -I{ oneapi .mkl_include ()} -L{ oneapi .dnn_lib ()} -L{ oneapi .mkl_lib ()} "
420
443
]
@@ -452,6 +475,12 @@ def ld_libraries(self):
452
475
453
476
def extra_cmake_args (self ):
454
477
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
+ ]
455
484
return [
456
485
f"-DCMAKE_CXX_FLAGS=-O3 -fsycl -ffast-math -I{ oneapi .dnn_include ()} -I{ oneapi .mkl_include ()} -L{ oneapi .dnn_lib ()} -L{ oneapi .mkl_lib ()} "
457
486
]
0 commit comments