mirror of
https://github.com/NVIDIA/nvbench.git
synced 2026-03-14 20:27:24 +00:00
* Add cuda architectures to build wheel for * Package scripts in wheel * Separate cuda major version extraction to fix architecutre selection logic * Add back statement printing cuda version * [pre-commit.ci] auto code formatting --------- Co-authored-by: pre-commit-ci[bot] <66853113+pre-commit-ci[bot]@users.noreply.github.com>
33 lines
1.0 KiB
Python
33 lines
1.0 KiB
Python
file_version = (1, 0, 0)
|
|
|
|
file_version_string = "{}.{}.{}".format(
|
|
file_version[0], file_version[1], file_version[2]
|
|
)
|
|
|
|
|
|
def check_file_version(filename, root_node):
|
|
try:
|
|
version_node = root_node["meta"]["version"]["json"]
|
|
except KeyError:
|
|
print("WARNING:")
|
|
print(" {} is written in an older, unversioned format. ".format(filename))
|
|
print(" It may not read correctly.")
|
|
print(" Reader expects JSON file version {}.".format(file_version_string))
|
|
return
|
|
|
|
# TODO We could do something fancy here using semantic versioning, but
|
|
# for now just warn on mismatch.
|
|
if version_node["string"] != file_version_string:
|
|
print("WARNING:")
|
|
print(
|
|
" {} was written using a different NVBench JSON file version.".format(
|
|
filename
|
|
)
|
|
)
|
|
print(" It may not read correctly.")
|
|
print(
|
|
" (file version: {} reader version: {})".format(
|
|
version_node["string"], file_version_string
|
|
)
|
|
)
|