|
6 | 6 | import re
|
7 | 7 | import subprocess
|
8 | 8 | import tempfile
|
| 9 | +import textwrap |
9 | 10 | from distutils.spawn import find_executable
|
10 | 11 |
|
11 | 12 | import lit.formats
|
|
155 | 156 | # Check for Level Zero SDK
|
156 | 157 | check_l0_file='l0_include.cpp'
|
157 | 158 | with open(check_l0_file, 'w') as fp:
|
158 |
| - fp.write('#include<level_zero/ze_api.h>\n') |
159 |
| - fp.write('int main() { uint32_t t; zeDriverGet(&t,nullptr); return t; }') |
| 159 | + print(textwrap.dedent( |
| 160 | + ''' |
| 161 | + #include <level_zero/ze_api.h> |
| 162 | + int main() { uint32_t t; zeDriverGet(&t, nullptr); return t; } |
| 163 | + ''' |
| 164 | + ), file=fp) |
160 | 165 |
|
161 | 166 | config.level_zero_libs_dir=lit_config.params.get("level_zero_libs_dir", config.level_zero_libs_dir)
|
162 | 167 | config.level_zero_include=lit_config.params.get("level_zero_include", (config.level_zero_include if config.level_zero_include else config.sycl_include))
|
|
177 | 182 | # Check for sycl-preview library
|
178 | 183 | check_preview_breaking_changes_file='preview_breaking_changes_link.cpp'
|
179 | 184 | with open(check_preview_breaking_changes_file, 'w') as fp:
|
180 |
| - fp.write('#include <sycl/sycl.hpp>\n') |
181 |
| - fp.write('namespace sycl { inline namespace _V1 { namespace detail {\n') |
182 |
| - fp.write('extern void PreviewMajorReleaseMarker();\n') |
183 |
| - fp.write('}}}\n') |
184 |
| - fp.write('int main() { sycl::detail::PreviewMajorReleaseMarker(); return 0; }\n') |
| 185 | + print(textwrap.dedent( |
| 186 | + ''' |
| 187 | + #include <sycl/sycl.hpp> |
| 188 | + namespace sycl { inline namespace _V1 { namespace detail { |
| 189 | + extern void PreviewMajorReleaseMarker(); |
| 190 | + }}} |
| 191 | + int main() { sycl::detail::PreviewMajorReleaseMarker(); return 0; } |
| 192 | + ''' |
| 193 | + ), file=fp) |
185 | 194 |
|
186 | 195 | sp = subprocess.getstatusoutput(config.dpcpp_compiler+' -fsycl -fpreview-breaking-changes ' + check_preview_breaking_changes_file)
|
187 | 196 | if sp[0] == 0:
|
|
190 | 199 | # Check for CUDA SDK
|
191 | 200 | check_cuda_file='cuda_include.cpp'
|
192 | 201 | with open(check_cuda_file, 'w') as fp:
|
193 |
| - fp.write('#include <cuda.h>\n') |
194 |
| - fp.write('int main() { CUresult r = cuInit(0); return r; }') |
| 202 | + print(textwrap.dedent( |
| 203 | + ''' |
| 204 | + #include <cuda.h> |
| 205 | + int main() { CUresult r = cuInit(0); return r; } |
| 206 | + ''' |
| 207 | + ), file=fp) |
195 | 208 |
|
196 | 209 | config.cuda_libs_dir=lit_config.params.get("cuda_libs_dir", config.cuda_libs_dir)
|
197 | 210 | config.cuda_include=lit_config.params.get("cuda_include", (config.cuda_include if config.cuda_include else config.sycl_include))
|
|
0 commit comments