Files
swift-mirror/benchmark/scripts/build_script_helper.py
2025-08-27 10:37:10 -04:00

69 lines
2.0 KiB
Python
Executable File

#!/usr/bin/env python3
import argparse
import os
import shutil
import subprocess
def perform_build(args, swiftbuild_path, config, binary_name, opt_flag):
assert config in ["debug", "release"]
assert binary_name in ["Benchmark_O", "Benchmark_Osize", "Benchmark_Onone"]
assert opt_flag in ["-O", "-Osize", "-Onone"]
inner_build_dir = os.path.join(args.build_path, binary_name)
swiftbuild_args = [
swiftbuild_path,
"--package-path",
args.package_path,
"--scratch-path",
inner_build_dir,
"--configuration",
config,
"-Xswiftc",
"-Xllvm",
"-Xswiftc",
"-align-module-to-page-size",
"-Xswiftc",
opt_flag
]
if config == "debug":
swiftbuild_args += [
"-Xswiftc",
"-DDEBUG"
]
if args.verbose:
swiftbuild_args.append("--verbose")
subprocess.call(swiftbuild_args)
# Copy the benchmark file into the final ./bin directory.
binpath = os.path.join(inner_build_dir, config, "SwiftBench")
finalpath = os.path.join(args.build_path, "bin", binary_name)
shutil.copy(binpath, finalpath)
def main():
parser = argparse.ArgumentParser()
parser.add_argument("--verbose", "-v", action="store_true")
parser.add_argument("--package-path", type=str, required=True)
parser.add_argument("--build-path", type=str, required=True)
parser.add_argument("--toolchain", type=str, required=True)
args = parser.parse_args()
# Create our bin directory so we can copy in the binaries.
bin_dir = os.path.join(args.build_path, "bin")
if not os.path.isdir(bin_dir):
os.makedirs(bin_dir)
swiftbuild_path = os.path.join(args.toolchain, "usr", "bin", "swift-build")
perform_build(args, swiftbuild_path, "debug", "Benchmark_Onone", "-Onone")
perform_build(args, swiftbuild_path, "release", "Benchmark_Osize", "-Osize")
perform_build(args, swiftbuild_path, "release", "Benchmark_O", "-O")
if __name__ == "__main__":
main()