Files
OpenLane/docker/current_platform.py
Mohamed Gaber ab6b79bce7 Don't require click for Docker platform extraction (#1205)
~ `docker/utils.py current-docker-platform` isolated into standalone `docker/current_platform.py` script that does not require any non-default Python libraries
2022-07-18 12:03:28 +02:00

24 lines
536 B
Python

import sys
import platform
def current_docker_platform() -> str:
arch = platform.machine()
if arch in ["x86_64", "amd64"]:
return "amd64"
elif arch in ["aarch64", "arm64"]:
return "arm64v8"
elif arch in ["ppc64le"]:
return "ppc64le"
else:
print(
f"Unsupported architecture '{platform.machine()}' Falling back to x86-64 for Docker.",
file=sys.stderr,
)
return "amd64"
if __name__ == "__main__":
print(current_docker_platform(), end="")