tree 1dbc1cf4e127e91e56d46c3ed581cd980ae9b0eb
parent dea61e7e6cad2022f7a1030a2b4bc084ce9afbc7
author xcnick <xcnick0412@gmail.com> 1642756344 -0800
committer PyTorch MergeBot <pytorchmergebot@users.noreply.github.com> 1642756582 +0000

Fix hsplit vsplit dsplit crash when section is 0 (#69342)

Summary:
Fixes https://github.com/pytorch/pytorch/issues/69270 #68970

Dim modulo by zero when split_size is 0.

Pull Request resolved: https://github.com/pytorch/pytorch/pull/69342

Reviewed By: ngimel

Differential Revision: D33692683

Pulled By: mruberry

fbshipit-source-id: aab82e5617c23c265b7dd3a8bac2dd245aaef5b4
(cherry picked from commit bcbf3b4c4d587394bc7586f32310f637fc9e3de7)