[TensorExpr] Fix lowering for aten::div.

[ZolotukhinM]

Stack from ghstack:

• [TensorExpr] Add a envvar to disable LLVM backend and use IR Eval instead. #48355 [TensorExpr] Add a envvar to disable LLVM backend and use IR Eval instead.
• [TensorExpr] IREval: fix div for Half dtype. #48354 [TensorExpr] IREval: fix div for Half dtype.
• [TensorExpr] Fix lowering for aten::div. #48329 [TensorExpr] Fix lowering for aten::div.
• [TensorExpr] Fix aten::atan2 lowering and disable aten::pow lowering on CPU. #48326 [TensorExpr] Fix aten::atan2 lowering and disable aten::pow lowering on CPU.

Differential Revision: D25130750

[eellison]

LGTM, one small update to make

Edit: nvm, looks good, was looking at old code in updatte suggestion

[dr-ci]

💊 CI failures summary and remediations

As of commit 63c23be (more details on the Dr. CI page):

---

• 2/2 failures possibly* introduced in this PR
  • 1/2 non-CircleCI failure(s)

---

🕵️ 1 new failure recognized by patterns

The following CI failures do not appear to be due to upstream breakages:

pytorch_linux_xenial_py3_6_gcc5_4_test (1/1)

Step: "Run tests" (full log | diagnosis details | 🔁 rerun)

Nov 21 07:23:37 [E request_callback_no_python.cpp:636] Received error while processing request type 258: RuntimeError: Can not pickle torch.futures.Future

Nov 21 07:23:37 At: 
Nov 21 07:23:37   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(98): serialize 
Nov 21 07:23:37   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(150): serialize 
Nov 21 07:23:37  
Nov 21 07:23:37 [E request_callback_no_python.cpp:636] Received error while processing request type 258: RuntimeError: Can not pickle torch.futures.Future 
Nov 21 07:23:37  
Nov 21 07:23:37 At: 
Nov 21 07:23:37   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(98): serialize 
Nov 21 07:23:37   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(150): serialize 
Nov 21 07:23:37  
Nov 21 07:23:37 [E request_callback_no_python.cpp:636] Received error while processing request type 258: RuntimeError: Can not pickle torch.futures.Future 
Nov 21 07:23:37  
Nov 21 07:23:37 At: 
Nov 21 07:23:37   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(98): serialize 
Nov 21 07:23:37   /opt/conda/lib/python3.6/site-packages/torch/distributed/rpc/internal.py(150): serialize 
Nov 21 07:23:37  
Nov 21 07:23:37 ok (1.638s) 
Nov 21 07:23:39   test_return_future_remote (__main__.ProcessGroupRpcTestWithSpawn) ... RPC was initialized with the PROCESS_GROUP backend which is deprecated and slated to be removed and superseded by the TENSORPIPE backend. It is recommended to migrate to the TENSORPIPE backend. 
Nov 21 07:23:39 RPC was initialized with the PROCESS_GROUP backend which is deprecated and slated to be removed and superseded by the TENSORPIPE backend. It is recommended to migrate to the TENSORPIPE backend. 
Nov 21 07:23:39 RPC was initialized with the PROCESS_GROUP backend which is deprecated and slated to be removed and superseded by the TENSORPIPE backend. It is recommended to migrate to the TENSORPIPE backend. 
Nov 21 07:23:39 RPC was initialized with the PROCESS_GROUP backend which is deprecated and slated to be removed and superseded by the TENSORPIPE backend. It is recommended to migrate to the TENSORPIPE backend. 

---

ci.pytorch.org: 1 failed

• Failed: pr/pytorch-linux-bionic-rocm3.9-py3.6

---

This comment was automatically generated by Dr. CI (expand for details).

Follow this link to opt-out of these comments for your Pull Requests.

Please report bugs/suggestions on the GitHub issue tracker or post in the (internal) Dr. CI Users group.

See how this bot performed.

This comment has been revised 20 times.

[facebook-github-bot]

@ZolotukhinM merged this pull request in b967119.