pytorch

3fa42a1b - Add a few more test cases for partitioner