[FixBug] compute early stopping with real batch size (#3418)

* [FixBug] compute early stopping with real batch size

* update

* fix test_sampler
This commit is contained in:
Zero Rains
2025-08-19 13:09:21 +08:00
committed by GitHub
parent 3a7a20d191
commit 8b12c80f90
4 changed files with 69 additions and 6 deletions

View File

@@ -26,7 +26,6 @@ done
failed_tests_file="failed_tests.log"
> "$failed_tests_file"
disabled_tests=(
layers/test_sampler.py
layers/test_append_attention.py
layers/test_attention.py
operators/test_rejection_top_p_sampling.py
@@ -36,7 +35,6 @@ disabled_tests=(
operators/test_stop_generation.py
operators/test_air_topp_sampling.py
operators/test_fused_moe.py
layers/test_repetition_early_stopper.py
operators/test_stop_generation_multi_ends.py
graph_optimization/test_cuda_graph.py
)