check WCR_CUDA against WCR (microsoft#190)

Co-authored-by: hualxie <hualxie@microsoft.com>

Author: xieofxie

.aitk/scripts/requirements_check.py

@@ -4,10 +4,19 @@
 def get_lines_from_file(file_path: Path):
     with open(file_path, "r", encoding="utf-8") as f:
         lines = f.readlines()
-    return [file_path, set(line.strip() for line in lines)]
+    return (file_path, set(line.strip() for line in lines))
 
 
-def req_is_subset(req1, req2):
+def req_is_subset(req1: tuple[Path, set[str]], req2: tuple[Path, set[str]], additional: set[str] = set()):
+    if not additional.issubset(req1[1]):
+        print(f"Additional is not a subset of '{req1[0].name}'")
+        missing = additional - req1[1]
+        print(f"Missing lines in '{req1[0].name}':")
+        for req in missing:
+            print(f"  {req}")
+        raise SystemExit(1)
+    req1 = (req1[0], req1[1] - additional)
+
     if req1[1].issubset(req2[1]):
         print(f"Requirement '{req1[0].name}' is subset of '{req2[0].name}'")
     else:
@@ -26,6 +35,18 @@ def requirements_check():
     QNN_LLM_lines = get_lines_from_file(requirements_folder / "requirements-QNN_LLM.txt")
     req_is_subset(WCR_INIT_lines, WCR_lines)
     req_is_subset(QNN_LLM_lines, WCR_lines)
+    WCR_CUDA_lines = get_lines_from_file(requirements_folder / "requirements-WCR_CUDA.txt")
+    req_is_subset(
+        WCR_CUDA_lines,
+        WCR_lines,
+        additional={
+            "--extra-index-url https://download.pytorch.org/whl/cu128",
+            "# torch==2.7.0+cu128",
+            "torch==2.7.0+cu128",
+            "# torchvision==0.22.0+cu128",
+            "torchvision==0.22.0+cu128",
+        },
+    )
 
 
 if __name__ == "__main__":