Avoid using true division in split_dim (#83527)
This makes it more amenable to tracing with dynamic shapes,
where we don't support SymFloats yet.
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/83527
Approved by: https://github.com/ngimel
