pytorch

e770241b - Test added for test_prod_gpu, under review, passes with changes successfully