[dtensor] add a few dunder methods to pointwise ops (pytorch#130754)

fixes pytorch#130671

Pull Request resolved: pytorch#130754
Approved by: https://github.com/Skylion007, https://github.com/awgu, https://github.com/msaroufim
ghstack dependencies: pytorch#130753

Author: wanchaol

torch/distributed/_tensor/ops/pointwise_ops.py

@@ -58,6 +58,14 @@
 
 pointwise_ops = [
     # please keep the entries below alphabetically sorted
+    aten.__ilshift__.Scalar,
+    aten.__ilshift__.Tensor,
+    aten.__irshift__.Scalar,
+    aten.__irshift__.Tensor,
+    aten.__lshift__.Scalar,
+    aten.__lshift__.Tensor,
+    aten.__rshift__.Scalar,
+    aten.__rshift__.Tensor,
     aten._conj.default,
     aten.abs.default,
     aten.abs.out,