[fp8] fix cb scheduler step tests

[yannicks1]

this change does always compile for the next power of 2 for the batch size.

[github-actions]

👋 Hi! Thank you for contributing to vLLM support on Spyre.
Just a reminder: Make sure that your code passes all the linting checks, otherwise your PR won't be able to be merged. To do so, first install the linting requirements, then run format.sh and commit the changes. This can be done with uv directly:

uv sync --frozen --group lint --active --inexact

Or this can be done with pip:

uv pip compile --group lint > requirements-lint.txt
pip install -r requirements-lint.txt
bash format.sh

Now you are good to go 🚀

[prashantgupta24]

this will leave the test name as-is, which might be confusing (since the test will still say BS 2)

[yannicks1]

bot:test

[maxdebayser]

I'm fine with this as long as the tests pass. Just for the record, can you add an explanation either here in the PR or in the code as to why this is required?

[joerunde]

Just for the record, can you add an explanation either here in the PR or in the code as to why this is required?

+1, can we extend the 🌶️🌶️🌶️ comment right above these changes?

[wallashss]

Another +1 for the documentation of the solution and question.

If we have this constraint of batch size, should we update platform as well somehow to give feedback to the user?

[yannicks1]

just added some more comments

[yannicks1]

bot:test
TORCH_SENDNN_CACHE_ENABLE=0

[wallashss]

bot:test