[threaded pg] adapt test_pointwise_ops.py (#90713)
Differential Revision: [D42153660](https://our.internmc.facebook.com/intern/diff/D42153660)
Pull Request resolved: https://github.com/pytorch/pytorch/pull/90713
Approved by: https://github.com/wanchaol
