Function main

python/tvm/exec/query_rpc_tracker.py:26–44

Main function

()

def main():
    """Main function"""
    parser = argparse.ArgumentParser()
    parser.add_argument("--host", type=str, default="", help="the hostname of the tracker")
    parser.add_argument("--port", type=int, default=None, help="The port of the RPC")
    args = parser.parse_args()
    logging.basicConfig(level=logging.INFO)

    # default to local host or environment variable
    if not args.host:
        args.host = os.environ.get("TVM_TRACKER_HOST", "127.0.0.1")

    if not args.port:
        args.port = int(os.environ.get("TVM_TRACKER_PORT", "9190"))

    conn = rpc.connect_tracker(args.host, args.port)
    # pylint: disable=superfluous-parens
    print(f"Tracker address {args.host}:{args.port}\n")
    print(f"{conn.text_summary()}")


if __name__ == "__main__":