fix

Author: SolitaryThinker

.github/workflows/fastvideo-kernel-publish.yml

@@ -64,12 +64,12 @@ jobs:
             # - torch-version: '2.7.1'
             #   cuda-version: '12.8.0'
             #   torch-cuda-short: 'cu128'
-            # - torch-version: '2.9.1'
-            #   cuda-version: '12.8.0'
-            #   torch-cuda-short: 'cu128'
-            - torch-version: '2.10.0'
+            - torch-version: '2.9.1'
               cuda-version: '12.8.0'
               torch-cuda-short: 'cu128'
+            # - torch-version: '2.10.0'
+            #   cuda-version: '12.8.0'
+            #   torch-cuda-short: 'cu128'
 
     steps:
       - name: Free up disk space
@@ -223,3 +223,6 @@ jobs:
         uses: pypa/gh-action-pypi-publish@release/v1
         with:
           packages-dir: fastvideo-kernel/dist/
+          # PyPI does not allow replacing an existing file with the same name.
+          # This makes re-runs idempotent by skipping files already uploaded.
+          skip-existing: true

fastvideo/tests/ssim/test_causal_similarity.py

@@ -19,6 +19,10 @@
     device_reference_folder = "A40" + device_reference_folder_suffix
 elif "L40S" in device_name:
     device_reference_folder = "L40S" + device_reference_folder_suffix
+elif "H100" in device_name:
+    device_reference_folder = "H100" + device_reference_folder_suffix
+elif "H200" in device_name:
+    device_reference_folder = "H200" + device_reference_folder_suffix
 else:
     # device_reference_folder = "L40S" + device_reference_folder_suffix
     logger.warning(f"Unsupported device for ssim tests: {device_name}")

fastvideo/tests/ssim/test_inference_similarity.py

@@ -41,6 +41,10 @@ def _shutdown_executor(generator: VideoGenerator | None) -> None:
     device_reference_folder = "A40" + device_reference_folder_suffix
 elif "L40S" in device_name:
     device_reference_folder = "L40S" + device_reference_folder_suffix
+elif "H100" in device_name:
+    device_reference_folder = "H100" + device_reference_folder_suffix
+elif "H200" in device_name:
+    device_reference_folder = "H200" + device_reference_folder_suffix
 else:
     # device_reference_folder = "L40S" + device_reference_folder_suffix
     logger.warning(f"Unsupported device for ssim tests: {device_name}")

fastvideo/tests/ssim/test_lingbot_similarity.py

@@ -47,6 +47,8 @@ def _find_lingbotworld_examples_root() -> str | None:
     device_reference_folder = "L40S" + device_reference_folder_suffix
 elif "H100" in device_name:
     device_reference_folder = "H100" + device_reference_folder_suffix
+elif "H200" in device_name:
+    device_reference_folder = "H200" + device_reference_folder_suffix
 else:
     device_reference_folder = None
     logger.warning("Unsupported device for ssim tests: %s", device_name)

fastvideo/tests/ssim/test_longcat_similarity.py

@@ -35,6 +35,8 @@
     device_reference_folder = "L40S" + device_reference_folder_suffix
 elif "H100" in device_name:
     device_reference_folder = "H100" + device_reference_folder_suffix
+elif "H200" in device_name:
+    device_reference_folder = "H200" + device_reference_folder_suffix
 else:
     logger.warning(f"Unsupported device for ssim tests: {device_name}")
 

fastvideo/tests/ssim/test_matrixgame_similarity.py

@@ -22,6 +22,10 @@
     device_reference_folder = "A40" + device_reference_folder_suffix
 elif "L40S" in device_name:
     device_reference_folder = "L40S" + device_reference_folder_suffix
+elif "H100" in device_name:
+    device_reference_folder = "H100" + device_reference_folder_suffix
+elif "H200" in device_name:
+    device_reference_folder = "H200" + device_reference_folder_suffix
 else:
     # device_reference_folder = "L40S" + device_reference_folder_suffix
     logger.warning(f"Unsupported device for ssim tests: {device_name}")

fastvideo/tests/ssim/test_sd35_similarity.py

@@ -22,6 +22,8 @@ def _device_reference_folder() -> str:
         return "L40S" + suffix
     if "H100" in device_name:
         return "H100" + suffix
+    if "H200" in device_name:
+        return "H200" + suffix
     if "RTX 4090" in device_name or "4090" in device_name:
         return "RTX4090" + suffix