Sparse: Remove dispatch in parallel region

[peterbell10]

Stack from ghstack:

• CTCLoss: Remove dispatching in parallel region #60599 CTCLoss: Remove dispatching in parallel region
• Sparse: Remove dispatch in parallel region #60598 Sparse: Remove dispatch in parallel region
• Embedding: Remove dispatch in parallel region #60597 Embedding: Remove dispatch in parallel region
• BatchNorm: Remove dispatch in parallel region #60596 BatchNorm: Remove dispatch in parallel region

Ref #56794

Differential Revision: D29446192

[facebook-github-bot]

💊 CI failures summary and remediations

As of commit db4df9b (more details on the Dr. CI page and at hud.pytorch.org/pr/60598):

---

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

---

🕵️ 3 new failures recognized by patterns

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

Lint / mypy (1/3)

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

2021-06-24T14:08:14.6553763Z torch/utils/benchm...ction "median" in typed context [no-untyped-call]

2021-06-24T14:08:01.8687967Z env:
2021-06-24T14:08:01.8688540Z   pythonLocation: /opt/hostedtoolcache/Python/3.8.10/x64
2021-06-24T14:08:01.8689264Z   LD_LIBRARY_PATH: /opt/hostedtoolcache/Python/3.8.10/x64/lib
2021-06-24T14:08:01.8689815Z ##[endgroup]
2021-06-24T14:08:01.8772442Z + for CONFIG in mypy*.ini
2021-06-24T14:08:01.8774723Z + mypy --config=mypy-strict.ini
2021-06-24T14:08:12.8711630Z torch/utils/benchmark/utils/common.py:166:34: error: Call to untyped function "median" in typed context  [no-untyped-call]
2021-06-24T14:08:12.8713276Z torch/utils/benchmark/utils/common.py:168:31: error: Call to untyped function "percentile" in typed context  [no-untyped-call]
2021-06-24T14:08:12.8715119Z torch/utils/benchmark/utils/common.py:169:31: error: Call to untyped function "percentile" in typed context  [no-untyped-call]
2021-06-24T14:08:12.8716631Z torch/utils/benchmark/utils/common.py:281:18: error: Call to untyped function "round" in typed context  [no-untyped-call]
2021-06-24T14:08:14.6553763Z torch/utils/benchmark/utils/timer.py:304:24: error: Call to untyped function "median" in typed context  [no-untyped-call]
2021-06-24T14:08:18.3688540Z Found 5 errors in 2 files (checked 141 source files)
2021-06-24T14:08:19.1892227Z ##[error]Process completed with exit code 1.
2021-06-24T14:08:19.1978211Z Post job cleanup.
2021-06-24T14:08:19.2878417Z [command]/usr/bin/git version
2021-06-24T14:08:19.2931517Z git version 2.32.0
2021-06-24T14:08:19.9309343Z [command]/usr/bin/git config --local --name-only --get-regexp core\.sshCommand
2021-06-24T14:08:19.9311194Z [command]/usr/bin/git submodule foreach --recursive git config --local --name-only --get-regexp 'core\.sshCommand' && git config --local --unset-all 'core.sshCommand' || :
2021-06-24T14:08:19.9312810Z [command]/usr/bin/git config --local --name-only --get-regexp http\.https\:\/\/github\.com\/\.extraheader
2021-06-24T14:08:19.9313800Z http.https://github.com/.extraheader
2021-06-24T14:08:19.9315252Z [command]/usr/bin/git config --local --unset-all http.https://github.com/.extraheader

pytorch_xla_linux_bionic_py3_6_clang9_build (2/3)

Step: "(Optional) Merge target branch" (full log | diagnosis details | 🔁 rerun)

Automatic merge failed; fix conflicts and then commit the result.

CONFLICT (add/add): Merge conflict in .github/templates/linux_ci_workflow.yml.j2
Auto-merging .github/templates/linux_ci_workflow.yml.j2
CONFLICT (add/add): Merge conflict in .github/scripts/lint_native_functions.py
Auto-merging .github/scripts/lint_native_functions.py
CONFLICT (add/add): Merge conflict in .github/scripts/generate_ci_workflows.py
Auto-merging .github/scripts/generate_ci_workflows.py
CONFLICT (add/add): Merge conflict in .circleci/verbatim-sources/job-specs/pytorch-job-specs.yml
Auto-merging .circleci/verbatim-sources/job-specs/pytorch-job-specs.yml
CONFLICT (add/add): Merge conflict in .circleci/config.yml
Auto-merging .circleci/config.yml
Automatic merge failed; fix conflicts and then commit the result.


Exited with code exit status 1

pytorch_linux_xenial_py3_6_gcc5_4_build (3/3)

Step: "(Optional) Merge target branch" (full log | diagnosis details | 🔁 rerun)

Automatic merge failed; fix conflicts and then commit the result.

CONFLICT (add/add): Merge conflict in .github/templates/linux_ci_workflow.yml.j2
Auto-merging .github/templates/linux_ci_workflow.yml.j2
CONFLICT (add/add): Merge conflict in .github/scripts/lint_native_functions.py
Auto-merging .github/scripts/lint_native_functions.py
CONFLICT (add/add): Merge conflict in .github/scripts/generate_ci_workflows.py
Auto-merging .github/scripts/generate_ci_workflows.py
CONFLICT (add/add): Merge conflict in .circleci/verbatim-sources/job-specs/pytorch-job-specs.yml
Auto-merging .circleci/verbatim-sources/job-specs/pytorch-job-specs.yml
CONFLICT (add/add): Merge conflict in .circleci/config.yml
Auto-merging .circleci/config.yml
Automatic merge failed; fix conflicts and then commit the result.


Exited with code exit status 1

---

ci.pytorch.org: 1 failed

• Failed: pr/pytorch-linux-bionic-rocm4.2-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 to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

[ngimel]

@ngimel has imported this pull request. If you are a Facebook employee, you can view this diff on Phabricator.

[facebook-github-bot]

@ngimel merged this pull request in 54673fc.