Add HeaderDescriptorSpec entries for cute, cutlass

Author: rwgk

cuda_pathfinder/cuda/pathfinder/_headers/header_descriptor_catalog.py

@@ -145,6 +145,14 @@ class HeaderDescriptorSpec:
         conda_targets_layout=False,
         use_ctk_root_canary=False,
     ),
+    HeaderDescriptorSpec(
+        name="cute",
+        packaged_with="other",
+        header_basename="cute/tensor.hpp",
+        site_packages_dirs=("cutlass_library/source/include",),
+        conda_targets_layout=False,
+        use_ctk_root_canary=False,
+    ),
     HeaderDescriptorSpec(
         name="cutensor",
         packaged_with="other",
@@ -153,6 +161,14 @@ class HeaderDescriptorSpec:
         conda_targets_layout=False,
         use_ctk_root_canary=False,
     ),
+    HeaderDescriptorSpec(
+        name="cutlass",
+        packaged_with="other",
+        header_basename="cutlass/cutlass.h",
+        site_packages_dirs=("cutlass_library/source/include",),
+        conda_targets_layout=False,
+        use_ctk_root_canary=False,
+    ),
     HeaderDescriptorSpec(
         name="mathdx",
         packaged_with="other",

cuda_pathfinder/pyproject.toml

@@ -43,6 +43,7 @@ cu13 = [
     "nvidia-nvshmem-cu13; sys_platform != 'win32'",
 ]
 host = [
+    "nvidia-cutlass",
     "nvpl-fft; platform_system == 'Linux' and platform_machine == 'aarch64'",
 ]
 

cuda_pathfinder/tests/test_find_nvidia_headers.py

@@ -41,7 +41,9 @@
 
 NON_CTK_IMPORTLIB_METADATA_DISTRIBUTIONS_NAMES = {
     "cusparseLt": r"^nvidia-cusparselt-.*$",
+    "cute": r"^nvidia-cutlass$",
     "cutensor": r"^cutensor-.*$",
+    "cutlass": r"^nvidia-cutlass$",
     "mathdx": r"^nvidia-libmathdx-.*$",
     "nvshmem": r"^nvidia-nvshmem-.*$",
 }