[pt2] Skip if curr_size is None (#101170)
Summary: There is a chance curr_size is None (got an error None cannot set item), so skip if it's None
Test Plan: unit test in D44736488
Differential Revision: D45767829
Pull Request resolved: https://github.com/pytorch/pytorch/pull/101170
Approved by: https://github.com/ezyang