File tree 2 files changed +10
-3
lines changed
2 files changed +10
-3
lines changed Original file line number Diff line number Diff line change @@ -34,6 +34,8 @@ class E2EExpr(BooleanExpression):
34
34
"hip_dev_kit" ,
35
35
"zstd" ,
36
36
"vulkan" ,
37
+ "hip_options" ,
38
+ "cuda_options" ,
37
39
"true" ,
38
40
"false" ,
39
41
}
Original file line number Diff line number Diff line change @@ -434,7 +434,7 @@ def open_check_file(file_name):
434
434
)
435
435
)
436
436
437
- cuda_options = cuda_options = (
437
+ cuda_options = (
438
438
(" -L" + config .cuda_libs_dir if config .cuda_libs_dir else "" )
439
439
+ " -lcuda "
440
440
+ " -I"
@@ -447,6 +447,10 @@ def open_check_file(file_name):
447
447
+ " /I"
448
448
+ config .cuda_include
449
449
)
450
+ if platform .system () == "Windows" :
451
+ cuda_options += (
452
+ " --cuda-path=" + os .path .dirname (os .path .dirname (config .cuda_libs_dir )) + f'"'
453
+ )
450
454
451
455
config .substitutions .append (("%cuda_options" , cuda_options ))
452
456
@@ -483,7 +487,7 @@ def open_check_file(file_name):
483
487
)
484
488
)
485
489
486
- hip_options = hip_options = (
490
+ hip_options = (
487
491
(" -L" + config .hip_libs_dir if config .hip_libs_dir else "" )
488
492
+ " -lamdhip64 "
489
493
+ " -I"
@@ -500,7 +504,8 @@ def open_check_file(file_name):
500
504
+ " /I"
501
505
+ config .hip_include
502
506
)
503
-
507
+ if platform .system () == "Windows" :
508
+ hip_options += " --rocm-path=" + os .path .dirname (config .hip_libs_dir ) + f'"'
504
509
with test_env ():
505
510
sp = subprocess .getstatusoutput (
506
511
config .dpcpp_compiler + " -fsycl " + check_hip_file + hip_options
You can’t perform that action at this time.
0 commit comments