From 4117feeea005202d20024f25d7e1abcda2f4e87c Mon Sep 17 00:00:00 2001 From: SolitaryThinker Date: Sun, 15 Feb 2026 01:22:26 +0000 Subject: [PATCH] fix --- .github/workflows/fastvideo-kernel-publish.yml | 11 +++++++---- fastvideo/tests/ssim/test_causal_similarity.py | 4 ++++ fastvideo/tests/ssim/test_inference_similarity.py | 4 ++++ fastvideo/tests/ssim/test_lingbot_similarity.py | 2 ++ fastvideo/tests/ssim/test_longcat_similarity.py | 2 ++ fastvideo/tests/ssim/test_matrixgame_similarity.py | 4 ++++ fastvideo/tests/ssim/test_sd35_similarity.py | 2 ++ 7 files changed, 25 insertions(+), 4 deletions(-) diff --git a/.github/workflows/fastvideo-kernel-publish.yml b/.github/workflows/fastvideo-kernel-publish.yml index 06dc0c444..f8d17516c 100644 --- a/.github/workflows/fastvideo-kernel-publish.yml +++ b/.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 diff --git a/fastvideo/tests/ssim/test_causal_similarity.py b/fastvideo/tests/ssim/test_causal_similarity.py index 423514a63..42156a96f 100644 --- a/fastvideo/tests/ssim/test_causal_similarity.py +++ b/fastvideo/tests/ssim/test_causal_similarity.py @@ -19,6 +19,10 @@ if "A40" in device_name: 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}") diff --git a/fastvideo/tests/ssim/test_inference_similarity.py b/fastvideo/tests/ssim/test_inference_similarity.py index 2c70e8838..90a69a88e 100644 --- a/fastvideo/tests/ssim/test_inference_similarity.py +++ b/fastvideo/tests/ssim/test_inference_similarity.py @@ -41,6 +41,10 @@ if "A40" in device_name: 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}") diff --git a/fastvideo/tests/ssim/test_lingbot_similarity.py b/fastvideo/tests/ssim/test_lingbot_similarity.py index 101ed3d95..b90f87781 100644 --- a/fastvideo/tests/ssim/test_lingbot_similarity.py +++ b/fastvideo/tests/ssim/test_lingbot_similarity.py @@ -47,6 +47,8 @@ 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 = None logger.warning("Unsupported device for ssim tests: %s", device_name) diff --git a/fastvideo/tests/ssim/test_longcat_similarity.py b/fastvideo/tests/ssim/test_longcat_similarity.py index f609925e0..adcf5cde0 100644 --- a/fastvideo/tests/ssim/test_longcat_similarity.py +++ b/fastvideo/tests/ssim/test_longcat_similarity.py @@ -35,6 +35,8 @@ 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: logger.warning(f"Unsupported device for ssim tests: {device_name}") diff --git a/fastvideo/tests/ssim/test_matrixgame_similarity.py b/fastvideo/tests/ssim/test_matrixgame_similarity.py index c720a922b..45904b86d 100644 --- a/fastvideo/tests/ssim/test_matrixgame_similarity.py +++ b/fastvideo/tests/ssim/test_matrixgame_similarity.py @@ -22,6 +22,10 @@ if "A40" in device_name: 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}") diff --git a/fastvideo/tests/ssim/test_sd35_similarity.py b/fastvideo/tests/ssim/test_sd35_similarity.py index 8198dade4..11dcf5c2c 100644 --- a/fastvideo/tests/ssim/test_sd35_similarity.py +++ b/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