yujuanqin's picture
support test_models on Intel
b295d06
raw
history blame contribute delete
411 Bytes
import os
import sys
from pathlib import Path
BINARY_DIR = Path(r"D:\yujuan\yoyo-translator-win\resources\bin")
def load_s2ts_lib():
sys.path.append(f"{BINARY_DIR}")
dll_path = os.environ.get("PATH")
new_dll_path = dll_path + f";{BINARY_DIR}"
os.environ["PATH"] = new_dll_path
os.environ["ONEDNN_MAX_CPU_ISA"] = "AVX2_VNNI"
os.environ["KMP_DUPLICATE_LIB_OK"] = "TRUE"
load_s2ts_lib()