Enable PiecewiseLinearTransform test on ROCm (#25632)
Summary:
thrust segfaults should have been fixed in ROCm rocThrust
Pull Request resolved: https://github.com/pytorch/pytorch/pull/25632
Differential Revision: D17179503
Pulled By: bddppq
fbshipit-source-id: 4d3854eacb30c945119d58250bccf399ccbc6105
