mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-06-12 01:08:27 +00:00
717f2efef7
[CK] add composable kernel support on gfx1250 (#6978) ## Motivation Add composable kernel support on gfx1250. ## Technical Details <!-- Explain the changes along with any relevant GitHub links. --> ## Test Plan <!-- Explain any relevant testing done to verify this PR. --> ## Test Result <!-- Briefly summarize test outcomes. --> ## Submission Checklist - [ ] Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests. --------- Co-authored-by: Qun Lin <qlin@amd.com> Co-authored-by: jialuo12_amdeng <jia.luo@amd.com> Co-authored-by: Andriy Roshchenko <andriy.roshchenko@amd.com> Co-authored-by: hsivasun_amdeng <haresh.sivasuntharampillai@amd.com>
80 lines
2.9 KiB
Python
80 lines
2.9 KiB
Python
import re
|
|
import sys
|
|
|
|
|
|
def extract_and_calculate_max(file_path):
|
|
"""
|
|
Extract NumVgprs and ScratchSize values from a .s file, and calculate their maximum values
|
|
Parameters:
|
|
file_path (str): Path to the .s file
|
|
Returns:
|
|
tuple: (max_vgprs, max_scratch, vgpr_count, scratch_count)
|
|
Returns (None, None, 0, 0) if no values are found or error occurs
|
|
"""
|
|
# Initialize lists to store values
|
|
num_vgprs_list = []
|
|
scratch_size_list = []
|
|
|
|
# Define regular expressions for matching
|
|
# Pattern explanation: Match lines starting with optional whitespace, followed by keyword, colon, and numbers
|
|
vgpr_pattern = re.compile(r"\s*NumVgprs:\s*(\d+)")
|
|
scratch_pattern = re.compile(r"\s*ScratchSize:\s*(\d+)")
|
|
|
|
try:
|
|
# Open and read the file with encoding handling to avoid errors
|
|
with open(file_path, "r", encoding="utf-8", errors="ignore") as f:
|
|
for line in f:
|
|
# Match NumVgprs
|
|
vgpr_match = vgpr_pattern.search(line)
|
|
if vgpr_match:
|
|
num = int(vgpr_match.group(1))
|
|
num_vgprs_list.append(num)
|
|
|
|
# Match ScratchSize
|
|
scratch_match = scratch_pattern.search(line)
|
|
if scratch_match:
|
|
num = int(scratch_match.group(1))
|
|
scratch_size_list.append(num)
|
|
|
|
# Calculate maximum values
|
|
max_vgprs = max(num_vgprs_list) if num_vgprs_list else None
|
|
max_scratch = max(scratch_size_list) if scratch_size_list else None
|
|
return max_vgprs, max_scratch, len(num_vgprs_list), len(scratch_size_list)
|
|
|
|
except Exception as e:
|
|
print(f"Error processing file {file_path}: {str(e)}", file=sys.stderr)
|
|
return None, None, 0, 0
|
|
|
|
|
|
if __name__ == "__main__":
|
|
# Check command line arguments
|
|
if len(sys.argv) < 3:
|
|
print("Usage: python parse_vgpr_scratch.py <file_pattern>")
|
|
print("Example: python parse_vgpr_scratch.py *.s")
|
|
sys.exit(1)
|
|
|
|
# Get file pattern from command line argument
|
|
# Windows
|
|
# file_pattern = sys.argv[1]
|
|
# glob.glob(file_pattern)
|
|
# Get all files matching the pattern in current directory
|
|
# Linux
|
|
file_list = sys.argv[2:]
|
|
|
|
if not file_list:
|
|
print("No files found in the list", file=sys.stderr)
|
|
sys.exit(1)
|
|
|
|
# Process each file and output results in single line
|
|
for file_path in file_list:
|
|
max_vgprs, max_scratch, vgpr_count, scratch_count = extract_and_calculate_max(
|
|
file_path
|
|
)
|
|
# Format output values (display 'N/A' if no value found)
|
|
vgpr_output = max_vgprs if max_vgprs is not None else "N/A"
|
|
scratch_output = max_scratch if max_scratch is not None else "N/A"
|
|
# Single line output with file name
|
|
print(
|
|
f"{file_path}: NumVgprs_Max={vgpr_output}, ScratchSize_Max={scratch_output}"
|
|
)
|