mirror of
https://github.com/facebookresearch/pytorch3d.git
synced 2025-08-01 03:12:49 +08:00
1f92c4e9d2
Reviewed By: zsol Differential Revision: D53258682 fbshipit-source-id: 3f006b5f31a2b1ffdc6323d3a3b08ac46c3162ce
43 lines
1.3 KiB
Python
Executable File
43 lines
1.3 KiB
Python
Executable File
#!/usr/bin/env python3
|
|
# Copyright (c) Meta Platforms, Inc. and affiliates.
|
|
# All rights reserved.
|
|
#
|
|
# This source code is licensed under the BSD-style license found in the
|
|
# LICENSE file in the root directory of this source tree.
|
|
|
|
import glob
|
|
import os
|
|
import subprocess
|
|
import sys
|
|
from os.path import dirname, isfile, join
|
|
|
|
|
|
def main() -> None:
|
|
# pyre-ignore[16]
|
|
if len(sys.argv) > 1:
|
|
# Parse from flags.
|
|
# pyre-ignore[16]
|
|
file_names = [
|
|
join(dirname(__file__), n) for n in sys.argv if n.startswith("bm_")
|
|
]
|
|
else:
|
|
# Get all the benchmark files (starting with "bm_").
|
|
bm_files = glob.glob(join(dirname(__file__), "bm_*.py"))
|
|
file_names = sorted(
|
|
f for f in bm_files if isfile(f) and not f.endswith("bm_main.py")
|
|
)
|
|
|
|
# Forward all important path information to the subprocesses through the
|
|
# environment.
|
|
os.environ["PATH"] = sys.path[0] + ":" + os.environ.get("PATH", "")
|
|
os.environ["LD_LIBRARY_PATH"] = (
|
|
sys.path[0] + ":" + os.environ.get("LD_LIBRARY_PATH", "")
|
|
)
|
|
os.environ["PYTHONPATH"] = ":".join(sys.path)
|
|
for file_name in file_names:
|
|
subprocess.check_call([sys.executable, file_name])
|
|
|
|
|
|
if __name__ == "__main__":
|
|
main() # pragma: no cover
|