iframe-proxy | Sunbelt Computer Software

executable file

54 lines (40 loc) · 1.62 KB

#!/usr/bin/env python3
# SPDX-FileCopyrightText: Copyright (c) 2025 NVIDIA CORPORATION & AFFILIATES.
# SPDX-License-Identifier: Apache-2.0
Helper to produce static bitcode input for test_nvvm.py.
    python toolshed/build_static_bitcode_input.py
It will print a ready-to-paste MINIMAL_NVVMIR_BITCODE_STATIC entry for the
current NVVM IR version detected at runtime.
import binascii
import textwrap
import llvmlite.binding  # HINT: pip install llvmlite
from cuda.bindings import nvvm
def get_minimal_nvvmir_txt_template():
    cuda_bindings_tests_dir = os.path.normpath("cuda_bindings/tests")
    assert os.path.isdir(cuda_bindings_tests_dir), (
        "Please run this helper script from the cuda-python top-level directory."
    sys.path.insert(0, os.path.abspath(cuda_bindings_tests_dir))
    import test_nvvm
    return test_nvvm.MINIMAL_NVVMIR_TXT_TEMPLATE
def main():
    major, _minor, debug_major, _debug_minor = nvvm.ir_version()
    txt = get_minimal_nvvmir_txt_template() % (major, debug_major)
    bitcode_dynamic = llvmlite.binding.parse_assembly(txt.decode()).as_bitcode()
    bitcode_hex = binascii.hexlify(bitcode_dynamic).decode("ascii")
    print("\n\nMINIMAL_NVVMIR_BITCODE_STATIC = { # PLEASE ADD TO test_nvvm.py")
    print(f"    ({major}, {debug_major}):  # (major, debug_major)")
    lines = textwrap.wrap(bitcode_hex, width=80)
    for line in lines[:-1]:
        print(f'    "{line}"')
    print(f'    "{lines[-1]}",')
    print("}\n", flush=True)
    print()
if __name__ == "__main__":
    assert len(sys.argv) == 1, "This helper script does not take any arguments."

Sunbelt Computer Software

PL/B Language Development and Support

Provide feedback

Saved searches

Use saved searches to filter your results more quickly

Uh oh!

Sunbelt Computer Software

PL/B Language Development and Support

Uh oh!

FilesExpand file tree

build_static_bitcode_input.py

Latest commit

History

build_static_bitcode_input.py

File metadata and controls