[pruner] add cuda tests for pruner (#61993)
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/61993
Repeating `test_pruner` unit tests for Linear and Conv2d models with device = 'cuda' to confirm pruner will work on GPU
- set device to cuda
- move model to device
- assert that module.weight.device is cuda
ghstack-source-id: 134554382
Test Plan:
`buck test mode/dev-nosan //caffe2/test:ao -- TestBasePruner`
https://pxl.cl/1Md9c
Reviewed By: jerryzh168
Differential Revision: D29829293
fbshipit-source-id: 1f7250e45695d0ad634d0bb7582a34fd1324e765
