Update new test dataset(batch size 1024, seq length 1024)

[InputNamePlz]

Ticket

Link to Github Issue

Problem description

Provide context for the problem.

What's changed

Describe the approach used to solve the problem.
Summarize the changes made and its impact.

Checklist

• All post commit CI passes
• Blackhole Post commit CI with demo tests passes (if applicable)
• Model regression CI passes (if applicable)
• Device performance regression CI passes (if applicable)
• (For models and ops writers) Single-card demo tests CI passes (if applicable) See recommended dev flow.
• Galaxy quick CI passes (if applicable)
• Galaxy demo tests, for Llama CI passes, if applicable, because of current Llama work
• (For runtime and ops writers) T3000 unit tests CI passes (if applicable, since this is run on push to main)
• (For models and ops writers) T3000 demo tests CI passes (if applicable, since this is required for release)
• New/Existing tests provide coverage for changes