Merge dsv4-moe-routing-primitives into deepseek_v4_compressed_attention

Author: parambole

.github/workflows/build_and_test_maxtext.yml

@@ -199,7 +199,7 @@ jobs:
       base_image: maxtext-unit-test-tpu:py312
       cloud_runner: linux-x86-ct6e-180-4tpu
       pytest_marker: 'not cpu_only and not gpu_only and not integration_test and not post_training'
-      pytest_addopts: '--ignore=tests/post_training'
+      pytest_addopts: '--ignore=tests/post_training --ignore=tests/unit/deepseek_v4_vs_reference_test.py'
       xla_python_client_mem_fraction: 0.75
       tf_force_gpu_allow_growth: false
       container_resource_option: "--privileged"

.github/workflows/run_tests_coordinator.yml

@@ -133,13 +133,13 @@ jobs:
 
       pytest_extra_args: >-
         ${{ fromJSON('{
-            "tpu-unit": "--ignore=tests/post_training",
+            "tpu-unit": "--ignore=tests/post_training --ignore=tests/unit/deepseek_v4_vs_reference_test.py",
             "tpu-integration": "--ignore=tests/post_training",
             "tpu-post-training-unit": "",
             "tpu-post-training-integration": "",
-            "gpu-unit": "--ignore=tests/post_training",
+            "gpu-unit": "--ignore=tests/post_training --ignore=tests/unit/deepseek_v4_vs_reference_test.py",
             "gpu-integration": "--ignore=tests/post_training",
-            "cpu-unit": "--ignore=tests/post_training",
+            "cpu-unit": "--ignore=tests/post_training --ignore=tests/unit/deepseek_v4_vs_reference_test.py",
             "cpu-post-training-unit": ""
           }')[inputs.flavor] }}
         ${{ inputs.additional_pytest_args }}