Print out information about TVM and the current Python environment
()
| 35 | |
| 36 | |
| 37 | def describe(): |
| 38 | """ |
| 39 | Print out information about TVM and the current Python environment |
| 40 | """ |
| 41 | info = list((k, v) for k, v in libinfo().items()) |
| 42 | info = dict(sorted(info, key=lambda x: x[0])) |
| 43 | print("Python Environment") |
| 44 | sys_version = sys.version.replace("\n", " ") |
| 45 | uname = platform.uname() |
| 46 | uname = f"{uname.system} {uname.release} {uname.version} {uname.machine}" |
| 47 | lines = [ |
| 48 | f"TVM version = {tvm.__version__}", |
| 49 | f"Python version = {sys_version} ({sys.maxsize.bit_length() + 1} bit)", |
| 50 | f"uname = {uname}", |
| 51 | ] |
| 52 | print(textwrap.indent("\n".join(lines), prefix=" ")) |
| 53 | print("CMake Options:") |
| 54 | print(textwrap.indent(json.dumps(info, indent=2), prefix=" ")) |