Coverage for cuda/pathfinder/_utils/path_sort.py: 100.00%
12 statements
« prev ^ index » next coverage.py v7.16.0, created at 2026-09-03 02:41 +0000
« prev ^ index » next coverage.py v7.16.0, created at 2026-09-03 02:41 +0000
1# SPDX-FileCopyrightText: Copyright (c) 2025-2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
2# SPDX-License-Identifier: Apache-2.0
4"""Deterministic path ordering helpers."""
6from __future__ import annotations
8import os
9import re
11_DIGIT_RUN = re.compile(r"(\d+)")
14def numeric_aware_path_sort_key(path: str) -> tuple[tuple[int, int, str], ...]:
15 """Return a key that compares embedded digit runs by numeric value."""
16 key: list[tuple[int, int, str]] = [] 1abcdefgkhi
17 for part in _DIGIT_RUN.split(os.path.normcase(path)): 1abcdefgkhi
18 if part.isdigit(): 1abcdefgkhi
19 normalized = part.lstrip("0") or "0" 1abcdefghi
20 key.append((1, len(normalized), normalized)) 1abcdefghi
21 else:
22 key.append((0, 0, part)) 1abcdefgkhi
23 return tuple(key) 1abcdefgkhi