cuda-python/toolshed/build_static_bitcode_input.py at main · hap2003/cuda-python

Repository navigation

#!/usr/bin/env python3

# SPDX-License-Identifier: LicenseRef-NVIDIA-SOFTWARE-LICENSE

"""

Helper to produce static bitcode input for test_nvvm.py.

Usage:

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 os

import sys

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."

main()