lee/BitNet
1
0
Fork 0
mirror of https://github.com/microsoft/BitNet.git synced 2026-05-03 11:20:36 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix llama-bench path error on Windows

Browse Source
This commit is contained in:
potassiummmm
2024-10-31 16:50:47 +08:00
parent 9b29748910
commit 80b94aecb2
1 changed files with 8 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
utils/e2e_benchmark.py
+8 -1
View File
@@ -2,6 +2,7 @@ import os
import sys
import logging
import argparse
import platform
import subprocess
def run_command(command, shell=False, log_step=None):
@@ -22,7 +23,13 @@ def run_command(command, shell=False, log_step=None):
sys.exit(1)
def run_benchmark():
bench_path = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "build/bin/llama-bench")
build_dir = os.path.join(os.path.dirname(os.path.dirname(os.path.abspath(__file__))), "build")
if platform.system() == "Windows":
bench_path = os.path.join(build_dir, "bin", "Release", "llama-bench.exe")
if not os.path.exists(bench_path):
bench_path = os.path.join(build_dir, "bin", "llama-bench")
else:
bench_path = os.path.join(build_dir, "bin", "llama-bench")
if not os.path.exists(bench_path):
logging.error(f"Benchmark binary not found, please build first.")
sys.exit(1)