[Metax][CI] update ci test files (#6340)

scripts/run_ci_metax.sh

@@ -95,14 +95,20 @@ METAX_CI_CASELIST=(
     "tests/operators/test_token_penalty.py"
     "tests/operators/test_stop_generation_multi_ends.py"
     "tests/operators/test_get_padding_offset.py"
-    "tests/operators/test_speculate_get_padding_offset.py"
     "tests/operators/test_rebuild_padding.py"
     "tests/operators/test_share_external_data.py"
     "tests/operators/test_rejection_top_p_sampling.py"
     "tests/layers/test_min_sampling.py"
 )
 for path in "${METAX_CI_CASELIST[@]}"; do
-    FILTERED_PATHS+=("$run_path/$path")
+    local TEST_FILE_PATH=$run_path/$path
+
+    if [ -e "$TEST_FILE_PATH" ]; then
+        FILTERED_PATHS+=("$TEST_FILE_PATH")
+    else
+        echo "Test file: [ $path ] does not exist, skip it."
+    fi
+
     # if ! is_excluded "$path"; then
     #     FILTERED_PATHS+=("$run_path/$path")
     # fi

tests/metax_ci/test_ernie_21b.py

@@ -10,6 +10,7 @@ os.environ["FLAGS_weight_only_linear_arch"] = "80"
 os.environ["FD_METAX_KVCACHE_MEM"] = "8"
 os.environ["ENABLE_V1_KVCACHE_SCHEDULER"] = "1"
 os.environ["FD_ENC_DEC_BLOCK_NUM"] = "2"
+os.environ["FD_SAMPLING_CLASS"] = "rejection"
 
 
 MODEL_PATH = "/data/models/PaddlePaddle/ERNIE-4.5-21B-A3B-Thinking"

tests/metax_ci/test_ernie_28b_vl.py

@@ -16,6 +16,7 @@ os.environ["FLAGS_weight_only_linear_arch"] = "80"
 os.environ["FD_METAX_KVCACHE_MEM"] = "8"
 os.environ["ENABLE_V1_KVCACHE_SCHEDULER"] = "1"
 os.environ["FD_ENC_DEC_BLOCK_NUM"] = "2"
+os.environ["FD_SAMPLING_CLASS"] = "rejection"
 
 
 MATERIAL_PATH = "/data/material"