Adding arg for configuring KernelAgent Test Timeout

[Jack-Khuu]

KernelAgent currently hardcodes the test time limit when evaluating a kernel candidate to 30s.

In simpler cases this is fine, but for more complex models, this starts to timeout for valid kernels or more comprehensive tests.

• For example, custom tests or tests with multiple input example (i.e GPUMode's test inputs)

This PR adds an optional arg test-timeout-s that configures this

Note: This happens more often in the kernelagent path than Fuser, since the subgraphs are typically smaller with quicker tests

---

Test

python -m Fuser.auto_agent --problem /home/jackkhuu/KernelAgent/problem.py --ka-model gpt-5 --router-model gpt-5 --extract-model gpt-5 --dispatch-model gpt-5 --compose-model gpt-5 --verify --ignore-router-config --no-router-cache --test-timeout-s 120

where problem.py is an edited version of gpumode trimul

[msaroufim]

A bit of a code smell that you have to pipe a timeout argument through so many data structures but otherwise lgtm

[Jack-Khuu]

A bit of a code smell that you have to pipe a timeout argument through so many data structures but otherwise lgtm

Yeaaaa, I had a hydra/yaml config that makes the repo much easy to play with, that I'll revive once the opt modules come in