diff --git a/projects/rocprofiler-compute/src/common.py b/projects/rocprofiler-compute/src/common.py index 5aeeced221..2109ede788 100644 --- a/projects/rocprofiler-compute/src/common.py +++ b/projects/rocprofiler-compute/src/common.py @@ -30,10 +30,7 @@ OMNIPERF_HOME = Path(__file__).resolve().parent # OMNIPERF INFO PROG = "omniperf" SOC_LIST = ["mi50", "mi100", "mi200"] -DISTRO_MAP = { - "platform:el8": "rhel8", - "15.3": "sle15sp3", -} +DISTRO_MAP = {"platform:el8": "rhel8", "15.3": "sle15sp3", "20.04": "ubuntu20_04"} version = os.path.join(OMNIPERF_HOME.parent, "VERSION") try: with open(version, "r") as file: diff --git a/projects/rocprofiler-compute/src/omniperf b/projects/rocprofiler-compute/src/omniperf index 47e3601c46..f4b231380d 100755 --- a/projects/rocprofiler-compute/src/omniperf +++ b/projects/rocprofiler-compute/src/omniperf @@ -203,6 +203,7 @@ def detect_roofline(): rocm_ver = mspec.rocmversion[:1] os_release = path("/etc/os-release").read_text() + ubuntu_distro = specs.search(r'VERSION_ID="(.*?)"', os_release) rhel_distro = specs.search(r'PLATFORM_ID="(.*?)"', os_release) sles_distro = specs.search(r'VERSION_ID="(.*?)"', os_release) @@ -221,6 +222,9 @@ def detect_roofline(): elif sles_distro == "15.3": # Must be a valid SLES machine distro = sles_distro + elif ubuntu_distro == "20.04": + # Must be a valid Ubuntu machine + distro = ubuntu_distro else: print("ROOFLINE ERROR: Cannot find a valid binary for your operating system") sys.exit(1)