File tree 1 file changed +2
-3
lines changed
1 file changed +2
-3
lines changed Original file line number Diff line number Diff line change 88
88
print ("Adding path to opencl-aot tool to PATH" )
89
89
os .environ ['PATH' ] = os .path .pathsep .join ((os .getenv ('PATH' ), config .sycl_tools_dir ))
90
90
91
- backend = lit_config .params .get ('SYCL_BE' , "PI_OPENCL" )
91
+ backend = lit_config .params .get ('SYCL_BE' , "PI_OPENCL" )
92
+ config .environment ['SYCL_BE' ] = backend
92
93
93
94
get_device_count_by_type_path = os .path .join (config .llvm_tools_dir , "get_device_count_by_type" )
94
95
@@ -159,8 +160,6 @@ def getDeviceCount(device_type):
159
160
if platform .system () == "Linux" :
160
161
gpu_run_on_linux_substitute = "env SYCL_DEVICE_TYPE=GPU "
161
162
gpu_check_on_linux_substitute = "| FileCheck %s"
162
- if cuda :
163
- gpu_run_on_linux_substitute += " SYCL_BE=PI_CUDA "
164
163
165
164
config .substitutions .append ( ('%GPU_RUN_PLACEHOLDER' , gpu_run_substitute ) )
166
165
config .substitutions .append ( ('%GPU_RUN_ON_LINUX_PLACEHOLDER' , gpu_run_on_linux_substitute ) )
You can’t perform that action at this time.
0 commit comments