mirror of
https://github.com/p-e-w/heretic.git
synced 2026-09-26 05:51:25 -07:00
experiment: try to standardize test environment
This commit is contained in:
@@ -268,6 +268,10 @@ def run():
|
|||||||
|
|
||||||
print(get_accelerator_info())
|
print(get_accelerator_info())
|
||||||
|
|
||||||
|
torch.backends.mkldnn.enabled = False # ty:ignore[invalid-assignment]
|
||||||
|
torch.set_num_threads(1)
|
||||||
|
torch.set_num_interop_threads(1)
|
||||||
|
|
||||||
if settings.print_debug_information:
|
if settings.print_debug_information:
|
||||||
print()
|
print()
|
||||||
print(torch.__config__.show().strip())
|
print(torch.__config__.show().strip())
|
||||||
|
|||||||
@@ -2,6 +2,7 @@
|
|||||||
# Copyright (C) 2025-2026 Philipp Emanuel Weidmann <pew@worldwidemann.com> + contributors
|
# Copyright (C) 2025-2026 Philipp Emanuel Weidmann <pew@worldwidemann.com> + contributors
|
||||||
|
|
||||||
import hashlib
|
import hashlib
|
||||||
|
import os
|
||||||
import subprocess
|
import subprocess
|
||||||
import sys
|
import sys
|
||||||
from pathlib import Path
|
from pathlib import Path
|
||||||
@@ -23,6 +24,12 @@ script_directory = Path(__file__).resolve().parent
|
|||||||
|
|
||||||
project_directory = script_directory.parent
|
project_directory = script_directory.parent
|
||||||
|
|
||||||
|
environment = os.environ | {
|
||||||
|
"OMP_NUM_THREADS": "1",
|
||||||
|
"MKL_NUM_THREADS": "1",
|
||||||
|
"OPENBLAS_NUM_THREADS": "1",
|
||||||
|
}
|
||||||
|
|
||||||
for test_directory in script_directory.iterdir():
|
for test_directory in script_directory.iterdir():
|
||||||
if test_directory.is_dir():
|
if test_directory.is_dir():
|
||||||
config_file = test_directory / "config.toml"
|
config_file = test_directory / "config.toml"
|
||||||
@@ -44,6 +51,7 @@ for test_directory in script_directory.iterdir():
|
|||||||
test_directory,
|
test_directory,
|
||||||
"heretic",
|
"heretic",
|
||||||
],
|
],
|
||||||
|
env=environment,
|
||||||
check=True,
|
check=True,
|
||||||
)
|
)
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user