tests : fix comp_padding CUDA graph reservation (#31344)

Assisted-by: Codex
This commit is contained in:
Yonghye Kwon authored and GitHub committed 2026-10-29 07:04:28 +02:00
1 parent 7ac5727d60
commit 6d5ac3c8d5
1 file changed
+2 -1
+2 -1
View File
@@ -782,7 +782,8 @@ struct case_comp_padding : ctx_test {
ctx_opts opts;
opts.n_ctx = 256*n_seq;
opts.n_batch = 64;
opts.n_ubatch = 64;
// Reserve the DSV4.1 small-batch attention graph when n_seq == 1.
opts.n_ubatch = 32;
opts.n_seq_max = n_seq;
opts.kv_unified = n_seq == 1;