forked from mooncake-track/Mooncake
29 lines
713 B
Python
29 lines
713 B
Python
#!/usr/bin/env python3
|
|
"""
|
|
Minimal CLI module for transfer_engine_bench.
|
|
"""
|
|
|
|
import os
|
|
import sys
|
|
import subprocess
|
|
|
|
|
|
def main():
|
|
"""
|
|
Main entry point for the transfer_engine_bench command.
|
|
Simply runs the transfer_engine_bench binary with all arguments passed through.
|
|
"""
|
|
# Get the path to the transfer_engine_bench binary
|
|
package_dir = os.path.dirname(os.path.abspath(__file__))
|
|
bin_path = os.path.join(package_dir, "transfer_engine_bench")
|
|
|
|
# Make sure the binary is executable
|
|
os.chmod(bin_path, 0o755)
|
|
|
|
# Run the binary with all arguments passed through
|
|
return subprocess.call([bin_path] + sys.argv[1:])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
sys.exit(main())
|