|
4 | 4 |
|
5 | 5 | import binascii |
6 | 6 | import re |
7 | | -import textwrap |
8 | 7 | from contextlib import contextmanager |
9 | 8 |
|
10 | 9 | import pytest |
11 | 10 | from cuda.bindings import nvvm |
12 | 11 |
|
13 | | -MINIMAL_NVVMIR_FIXTURE_PARAMS = ["txt", "bitcode_static"] |
14 | | -try: |
15 | | - import llvmlite.binding as llvmlite_binding # Optional test dependency. |
16 | | -except ImportError: |
17 | | - llvmlite_binding = None |
18 | | -else: |
19 | | - MINIMAL_NVVMIR_FIXTURE_PARAMS.append("bitcode_dynamic") |
20 | | - |
21 | | -MINIMAL_NVVMIR_TXT = b"""\ |
| 12 | +MINIMAL_NVVMIR_TXT_TEMPLATE = b"""\ |
22 | 13 | target datalayout = "e-p:64:64:64-i1:8:8-i8:8:8-i16:16:16-i32:32:32-i64:64:64-i128:128:128-f32:32:32-f64:64:64-v16:16:16-v32:32:32-v64:64:64-v128:128:128-n16:32:64" |
23 | 14 |
|
24 | 15 | target triple = "nvptx64-nvidia-cuda" |
|
130 | 121 | "6e673e0000000000", |
131 | 122 | } |
132 | 123 |
|
133 | | -MINIMAL_NVVMIR_CACHE = {} |
134 | | - |
135 | 124 |
|
136 | | -@pytest.fixture(params=MINIMAL_NVVMIR_FIXTURE_PARAMS) |
| 125 | +@pytest.fixture(params=("txt", "bitcode_static")) |
137 | 126 | def minimal_nvvmir(request): |
138 | | - for pass_counter in range(2): |
139 | | - nvvmir = MINIMAL_NVVMIR_CACHE.get(request.param, -1) |
140 | | - if nvvmir != -1: |
141 | | - if nvvmir is None: |
142 | | - pytest.skip(f"UNAVAILABLE: {request.param}") |
143 | | - return nvvmir |
144 | | - if pass_counter: |
145 | | - raise AssertionError("This code path is meant to be unreachable.") |
146 | | - # Build cache entries, then try again (above). |
147 | | - major, minor, debug_major, debug_minor = nvvm.ir_version() |
148 | | - txt = MINIMAL_NVVMIR_TXT % (major, debug_major) |
149 | | - if llvmlite_binding is None: |
150 | | - bitcode_dynamic = None |
151 | | - else: |
152 | | - bitcode_dynamic = llvmlite_binding.parse_assembly(txt.decode()).as_bitcode() |
153 | | - bitcode_static = MINIMAL_NVVMIR_BITCODE_STATIC.get((major, debug_major)) |
154 | | - if bitcode_static is not None: |
155 | | - bitcode_static = binascii.unhexlify(bitcode_static) |
156 | | - MINIMAL_NVVMIR_CACHE["txt"] = txt |
157 | | - MINIMAL_NVVMIR_CACHE["bitcode_dynamic"] = bitcode_dynamic |
158 | | - MINIMAL_NVVMIR_CACHE["bitcode_static"] = bitcode_static |
159 | | - if bitcode_static is None: |
160 | | - if bitcode_dynamic is None: |
161 | | - raise RuntimeError("Please `pip install llvmlite` to generate `bitcode_static` (see PR #443)") |
162 | | - bitcode_hex = binascii.hexlify(bitcode_dynamic).decode("ascii") |
163 | | - print("\n\nMINIMAL_NVVMIR_BITCODE_STATIC = { # PLEASE ADD TO test_nvvm.py") |
164 | | - print(f" ({major}, {debug_major}): # (major, debug_major)") |
165 | | - lines = textwrap.wrap(bitcode_hex, width=80) |
166 | | - for line in lines[:-1]: |
167 | | - print(f' "{line}"') |
168 | | - print(f' "{lines[-1]}",') |
169 | | - print("}\n", flush=True) |
| 127 | + major, minor, debug_major, debug_minor = nvvm.ir_version() |
| 128 | + |
| 129 | + if request.param == "txt": |
| 130 | + return MINIMAL_NVVMIR_TXT_TEMPLATE % (major, debug_major) |
| 131 | + |
| 132 | + bitcode_static_binascii = MINIMAL_NVVMIR_BITCODE_STATIC.get((major, debug_major)) |
| 133 | + if bitcode_static_binascii: |
| 134 | + return binascii.unhexlify(bitcode_static_binascii) |
| 135 | + raise RuntimeError( |
| 136 | + "Static bitcode for NVVM IR version " |
| 137 | + f"{major}.{debug_major} is not available in this test.\n" |
| 138 | + "Maintainers: Please run the helper script to generate it and add the " |
| 139 | + "output to the MINIMAL_NVVMIR_BITCODE_STATIC dict:\n" |
| 140 | + " ../../toolshed/build_static_bitcode_input.py" |
| 141 | + ) |
170 | 142 |
|
171 | 143 |
|
172 | 144 | @pytest.fixture(params=[nvvm.compile_program, nvvm.verify_program]) |
|
0 commit comments