mirror of
https://github.com/amd/blis.git
synced 2026-05-12 10:05:38 +00:00
7e50ba669b
Some text files were missing a newline at the end of the file.
One has been added.
Also correct file format of windows/tests/inputs.yaml, which
was missed in commit 0f0277e104
AMD-Internal: [CPUPL-2870]
Change-Id: Icb83a4a27033dc0ff325cb84a1cf399e953ec549
74 lines
2.5 KiB
Python
74 lines
2.5 KiB
Python
"""Copyright (C) 2020, Advanced Micro Devices, Inc. All Rights Reserved"""
|
|
|
|
import subprocess
|
|
import sys
|
|
|
|
def config_check():
|
|
# Execute wmic shell command with sub-process
|
|
result = subprocess.run(['wmic', 'cpu', 'get', 'caption'], stdout=subprocess.PIPE, text=True).stdout
|
|
|
|
# Replace the newline character with empty char
|
|
result=result.replace('\n', '')
|
|
|
|
# parse the string into list of string
|
|
parse_string=result.split(" ")
|
|
|
|
# Strip the empty strings from list
|
|
parse_string=[list for list in parse_string if list.strip()]
|
|
|
|
vendor=parse_string[1]
|
|
family=hex(int(parse_string[3]))
|
|
model=hex(int(parse_string[5]))
|
|
stepping=hex(int(parse_string[7]))
|
|
|
|
# AMD family numbers
|
|
# Zen/ Zen+/Zen2 family number
|
|
zen_family="0x17"
|
|
# Bulldozer / Piledriver / Steamroller / Excavator family number
|
|
amd_family="0x15"
|
|
|
|
# AMD CPUID model numbers
|
|
zen_model=["0x30", "0xff"]
|
|
zen2_model=["0x00", "0xff"]
|
|
excavator_model=["0x60","0x7f"]
|
|
steamroller_model=["0x30", "0x3f"]
|
|
piledriver_model=["0x02", "0x10", "0x1f"]
|
|
bulldozer_model=["0x00", "0x01"]
|
|
|
|
# Check the CPU configuration Intel64/AMD64
|
|
if vendor.count("Intel64"):
|
|
return
|
|
elif vendor.count("AMD64"):
|
|
# Check the AMD family name
|
|
if family == zen_family:
|
|
if (zen_model[0] <= model and model <= zen_model[1]) :
|
|
family="zen2"
|
|
elif (zen2_model[0] <= model and model <= zen2_model[1]) :
|
|
family="zen"
|
|
else:
|
|
print("Unknown model number")
|
|
elif family == amd_family:
|
|
# check for specific models of excavator family
|
|
if (excavator_model[0] <= model and model <= excavator_model[1]) :
|
|
family="excavator"
|
|
# check for specific models of steamroller family
|
|
elif (steamroller_model[0] <= model and model <= steamroller_model[1]) :
|
|
family="steamroller"
|
|
# check for specific models of piledriver family
|
|
elif (model == piledriver_model[0] or (piledriver_model[1] <= model and model <= piledriver_model[2])) :
|
|
family="piledriver"
|
|
# check for specific models of bulldozer family
|
|
elif (model == bulldozer_model[0] or model == bulldozer_model[1]) :
|
|
family="bulldozer"
|
|
else:
|
|
print("Unknown model number")
|
|
else:
|
|
print("Unknown family")
|
|
else:
|
|
print("UNKNOWN CPU")
|
|
return family
|
|
|
|
# Function call for config family names
|
|
FAMILY=config_check()
|
|
print(FAMILY)
|