limit max_split_per_batch to 16

[valarLip]

Motivation

Technical Details

Test Plan

Test Result

Submission Checklist

• Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

[Copilot]

Pull request overview

This PR makes two main changes: (1) replaces the "topk": -1 parameter with "max_split_per_batch": 16 in the MLA attention metadata builder to properly limit batch splitting, and (2) applies code formatting improvements to the linear operations module.

• Corrects the split_params configuration in aiter_mla.py by replacing an incorrect parameter name
• Reformats function signatures and reorganizes imports in linear.py for better code readability

Reviewed changes

Copilot reviewed 2 out of 2 changed files in this pull request and generated 1 comment.

File	Description
atom/model_ops/attentions/aiter_mla.py	Replaces "topk": -1 with "max_split_per_batch": 16 in the split_params dictionary passed to get_mla_metadata_v1
atom/model_ops/linear.py	Code formatting improvements including import reorganization, multi-line function signatures, and whitespace cleanup; introduces a duplicate import issue

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.