mirror of
https://github.com/zama-ai/tfhe-rs.git
synced 2026-01-06 21:34:05 -05:00
35 lines
1.1 KiB
Python
35 lines
1.1 KiB
Python
#!/usr/bin/python3
|
|
|
|
from pathlib import Path
|
|
|
|
from utils import get_repo_root, format_version_major_minor, get_tfhe_version
|
|
|
|
|
|
def main():
|
|
repo_root = get_repo_root()
|
|
tfhe_current_version = format_version_major_minor(get_tfhe_version())
|
|
shortint_param_file = repo_root / "tfhe/src/shortint/parameters/mod.rs"
|
|
|
|
with open(shortint_param_file, "r", encoding="utf-8") as f:
|
|
lines = f.readlines()
|
|
for line in lines:
|
|
line = line.strip()
|
|
if "as current_params;" in line:
|
|
print("Content:", line)
|
|
expected = f"use {tfhe_current_version} as current_params;"
|
|
print("Expected use:", expected)
|
|
if not line.endswith(expected):
|
|
print(
|
|
"Use clause for current parameters does not match current TFHE-rs version"
|
|
)
|
|
exit(1)
|
|
else:
|
|
print("OK")
|
|
return
|
|
print("Did not find import line.")
|
|
exit(1)
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main()
|