2025-12-05 nightly release (0a2cebd)

Author: pytorchbot

.github/workflows/unittest_ci.yml

@@ -36,7 +36,7 @@ jobs:
     strategy:
       fail-fast: false
       matrix:
-        cuda-tag: ["cu126", "cu128"]
+        cuda-tag: ["cu126", "cu128", "cu129"]
         os:
           - linux.g5.12xlarge.nvidia.gpu
         python:
@@ -55,18 +55,20 @@ jobs:
             cuda-tag: "cu126"
           - is_pr: true
             cuda-tag: "cu128"
+          - is_pr: true
+            cuda-tag: "cu129"
             python:
               version: "3.9"
           - is_pr: true
-            cuda-tag: "cu128"
+            cuda-tag: "cu129"
             python:
               version: "3.10"
           - is_pr: true
-            cuda-tag: "cu128"
+            cuda-tag: "cu129"
             python:
               version: "3.11"
           - is_pr: true
-            cuda-tag: "cu128"
+            cuda-tag: "cu129"
             python:
               version: "3.12"
     uses: pytorch/test-infra/.github/workflows/linux_job_v2.yml@main

torchrec/distributed/test_utils/test_model_parallel_base.py

@@ -253,7 +253,7 @@ def test_sharding_ebc_input_validation_enabled(self, mock_jk: Mock) -> None:
 
         with self.assertRaisesRegex(ValueError, "keys must be unique"):
             model(kjt)
-        mock_jk.assert_called_once_with("pytorch/torchrec:enable_kjt_validation")
+        mock_jk.assert_any_call("pytorch/torchrec:enable_kjt_validation")
 
     @patch("torch._utils_internal.justknobs_check")
     def test_sharding_ebc_validate_input_only_once(self, mock_jk: Mock) -> None:
@@ -271,7 +271,7 @@ def test_sharding_ebc_validate_input_only_once(self, mock_jk: Mock) -> None:
             model(kjt)
             model(kjt)
 
-        mock_jk.assert_called_once_with("pytorch/torchrec:enable_kjt_validation")
+        mock_jk.assert_any_call("pytorch/torchrec:enable_kjt_validation")
         matched_logs = list(
             filter(lambda s: "Validating input features..." in s, logs.output)
         )
@@ -294,7 +294,7 @@ def test_sharding_ebc_input_validation_disabled(self, mock_jk: Mock) -> None:
         except ValueError:
             self.fail("Input validation should not be enabled.")
 
-        mock_jk.assert_called_once_with("pytorch/torchrec:enable_kjt_validation")
+        mock_jk.assert_any_call("pytorch/torchrec:enable_kjt_validation")
 
     def _create_sharded_model(
         self, embedding_dim: int = 128, num_embeddings: int = 256