[PyTorch][JIT] use a better hash table in alias analysis

[swolchok]

Stack from ghstack:

• -> [PyTorch][JIT] use a better hash table in alias analysis #69854

Differential Revision: D33039733

[pytorch-probot]

CI Flow Status

⚛️ CI Flow

Ruleset - Version: v1
Ruleset - File: https://github.com/pytorch/pytorch/blob/570a64c2d5b4eda8fe1b0ffee5905075f9311ea9/.github/generated-ciflow-ruleset.json
PR ciflow labels: ciflow/default

Workflows	Labels (bold enabled)	Status
Triggered Workflows
linux-bionic-py3.7-clang9	ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/noarch, ciflow/trunk	✅ triggered
linux-docs	ciflow/all, ciflow/cpu, ciflow/default, ciflow/docs, ciflow/linux, ciflow/trunk	✅ triggered
linux-vulkan-bionic-py3.7-clang9	ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk, ciflow/vulkan	✅ triggered
linux-xenial-cuda11.3-py3.7-gcc7	ciflow/all, ciflow/cuda, ciflow/default, ciflow/linux, ciflow/trunk	✅ triggered
linux-xenial-cuda11.3-py3.7-gcc7-bazel-test	ciflow/all, ciflow/bazel, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk	✅ triggered
linux-xenial-py3-clang5-mobile-build	ciflow/all, ciflow/default, ciflow/linux, ciflow/mobile, ciflow/trunk	✅ triggered
linux-xenial-py3-clang5-mobile-custom-build-static	ciflow/all, ciflow/default, ciflow/linux, ciflow/mobile, ciflow/trunk	✅ triggered
linux-xenial-py3.7-clang7-asan	ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/sanitizers, ciflow/trunk	✅ triggered
linux-xenial-py3.7-clang7-onnx	ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/onnx, ciflow/trunk	✅ triggered
linux-xenial-py3.7-gcc5.4	ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk	✅ triggered
linux-xenial-py3.7-gcc7	ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk	✅ triggered
linux-xenial-py3.7-gcc7-no-ops	ciflow/all, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk	✅ triggered
pytorch-linux-xenial-py3-clang5-android-ndk-r19c-gradle-custom-build-single	ciflow/all, ciflow/android, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk	✅ triggered
pytorch-linux-xenial-py3-clang5-android-ndk-r19c-gradle-custom-build-single-full-jit	ciflow/all, ciflow/android, ciflow/cpu, ciflow/default, ciflow/linux, ciflow/trunk	✅ triggered
win-vs2019-cpu-py3	ciflow/all, ciflow/cpu, ciflow/default, ciflow/trunk, ciflow/win	✅ triggered
win-vs2019-cuda11.3-py3	ciflow/all, ciflow/cuda, ciflow/default, ciflow/trunk, ciflow/win	✅ triggered
Skipped Workflows
caffe2-linux-xenial-py3.7-gcc5.4	ciflow/all, ciflow/cpu, ciflow/linux, ciflow/trunk	🚫 skipped
docker-builds	ciflow/all, ciflow/trunk	🚫 skipped
ios-12-5-1-arm64	ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk	🚫 skipped
ios-12-5-1-arm64-coreml	ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk	🚫 skipped
ios-12-5-1-arm64-custom-ops	ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk	🚫 skipped
ios-12-5-1-arm64-full-jit	ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk	🚫 skipped
ios-12-5-1-arm64-metal	ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk	🚫 skipped
ios-12-5-1-x86-64	ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk	🚫 skipped
ios-12-5-1-x86-64-coreml	ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk	🚫 skipped
ios-12-5-1-x86-64-full-jit	ciflow/all, ciflow/ios, ciflow/macos, ciflow/trunk	🚫 skipped
libtorch-linux-xenial-cuda10.2-py3.7-gcc7	ciflow/all, ciflow/cuda, ciflow/libtorch, ciflow/linux, ciflow/trunk	🚫 skipped
libtorch-linux-xenial-cuda11.3-py3.7-gcc7	ciflow/all, ciflow/cuda, ciflow/libtorch, ciflow/linux, ciflow/trunk	🚫 skipped
linux-binary-conda	ciflow/binaries, ciflow/binaries/conda	🚫 skipped
linux-binary-libtorch-cxx11-abi	ciflow/binaries, ciflow/binaries/libtorch	🚫 skipped
linux-binary-libtorch-pre-cxx11	ciflow/binaries, ciflow/binaries/libtorch	🚫 skipped
linux-binary-manywheel	ciflow/binaries, ciflow/binaries/wheel	🚫 skipped
linux-bionic-cuda10.2-py3.9-gcc7	ciflow/all, ciflow/cuda, ciflow/linux, ciflow/slow, ciflow/trunk	🚫 skipped
linux-bionic-py3.6-clang9	ciflow/xla	🚫 skipped
linux-docs-push	ciflow/all, ciflow/cpu, ciflow/linux, ciflow/scheduled	🚫 skipped
linux-xenial-cuda11.3-py3.7-gcc7-no-ops	ciflow/all, ciflow/cuda, ciflow/linux, ciflow/trunk	🚫 skipped
macos-10-15-py3-arm64	ciflow/all, ciflow/macos, ciflow/trunk	🚫 skipped
macos-10-15-py3-lite-interpreter-x86-64	ciflow/all, ciflow/macos, ciflow/trunk	🚫 skipped
macos-11-py3-x86-64	ciflow/all, ciflow/macos, ciflow/trunk	🚫 skipped
parallelnative-linux-xenial-py3.7-gcc5.4	ciflow/all, ciflow/cpu, ciflow/linux, ciflow/trunk	🚫 skipped
periodic-libtorch-linux-bionic-cuda11.5-py3.7-gcc7	ciflow/all, ciflow/cuda, ciflow/libtorch, ciflow/linux, ciflow/scheduled	🚫 skipped
periodic-libtorch-linux-xenial-cuda11.1-py3.7-gcc7	ciflow/all, ciflow/cuda, ciflow/libtorch, ciflow/linux, ciflow/scheduled	🚫 skipped
periodic-linux-bionic-cuda11.5-py3.7-gcc7	ciflow/all, ciflow/cuda, ciflow/linux, ciflow/scheduled	🚫 skipped
periodic-linux-xenial-cuda10.2-py3-gcc7-slow-gradcheck	ciflow/all, ciflow/cuda, ciflow/linux, ciflow/scheduled, ciflow/slow, ciflow/slow-gradcheck	🚫 skipped
periodic-linux-xenial-cuda11.1-py3.7-gcc7-debug	ciflow/all, ciflow/cuda, ciflow/linux, ciflow/scheduled	🚫 skipped
periodic-win-vs2019-cuda11.1-py3	ciflow/all, ciflow/cuda, ciflow/scheduled, ciflow/win	🚫 skipped
periodic-win-vs2019-cuda11.5-py3	ciflow/all, ciflow/cuda, ciflow/scheduled, ciflow/win	🚫 skipped
pytorch-linux-xenial-py3-clang5-android-ndk-r19c-build	ciflow/all, ciflow/android, ciflow/cpu, ciflow/linux, ciflow/trunk	🚫 skipped

You can add a comment to the PR and tag @pytorchbot with the following commands:

# ciflow rerun, "ciflow/default" will always be added automatically
@pytorchbot ciflow rerun

# ciflow rerun with additional labels "-l <ciflow/label_name>", which is equivalent to adding these labels manually and trigger the rerun
@pytorchbot ciflow rerun -l ciflow/scheduled -l ciflow/slow

For more information, please take a look at the CI Flow Wiki.

[facebook-github-bot]

🔗 Helpful links

• 🧪  See artifacts and rendered test results at hud.pytorch.org/pr/69854
• 📄  Preview docs built from this PR
• 📄  Preview C++ docs built from this PR
• ↩️  [fb-only] Re-run with SSH instructions
• 🔧  Opt-in to CIFlow to control what jobs run on your PRs

💊 CI failures summary and remediations

As of commit 013a31b (more details on the Dr. CI page):

---

• 1/1 failures introduced in this PR

---

🕵️ 1 new failure recognized by patterns

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

Lint / quick-checks (1/1)

Step: "Ensure no trailing spaces" (full log | diagnosis details | 🔁 rerun)

2022-02-03T18:05:39.7336626Z python: can't open..._launches.py': [Errno 2] No such file or directory

2022-02-03T18:05:39.6993089Z ##[group]Run set -eux
2022-02-03T18:05:39.6993362Z �[36;1mset -eux�[0m
2022-02-03T18:05:39.6993772Z �[36;1mpython torch/testing/_check_kernel_launches.py |& tee "${GITHUB_WORKSPACE}"/cuda_kernel_launch_checks.txt�[0m
2022-02-03T18:05:39.7041567Z shell: /bin/bash -e {0}
2022-02-03T18:05:39.7041822Z env:
2022-02-03T18:05:39.7042270Z   pythonLocation: /opt/hostedtoolcache/Python/3.10.2/x64
2022-02-03T18:05:39.7042657Z   LD_LIBRARY_PATH: /opt/hostedtoolcache/Python/3.10.2/x64/lib
2022-02-03T18:05:39.7042976Z ##[endgroup]
2022-02-03T18:05:39.7122538Z + python torch/testing/_check_kernel_launches.py
2022-02-03T18:05:39.7123849Z + tee /home/runner/work/pytorch/pytorch/cuda_kernel_launch_checks.txt
2022-02-03T18:05:39.7336626Z python: can't open file '/home/runner/work/pytorch/pytorch/torch/testing/_check_kernel_launches.py': [Errno 2] No such file or directory
2022-02-03T18:05:39.7395661Z ##[group]Run (! git --no-pager grep -I -no $'#include <cub/' --  ./aten  ':(exclude)aten/src/ATen/cuda/cub*.cuh' || (echo "The above files have direct cub include; please include ATen/cuda/cub.cuh instead and wrap your cub calls in at::native namespace if necessary"; false))
2022-02-03T18:05:39.7396675Z �[36;1m(! git --no-pager grep -I -no $'#include <cub/' --  ./aten  ':(exclude)aten/src/ATen/cuda/cub*.cuh' || (echo "The above files have direct cub include; please include ATen/cuda/cub.cuh instead and wrap your cub calls in at::native namespace if necessary"; false))�[0m
2022-02-03T18:05:39.7465089Z shell: /bin/bash -e {0}
2022-02-03T18:05:39.7465353Z env:
2022-02-03T18:05:39.7465666Z   pythonLocation: /opt/hostedtoolcache/Python/3.10.2/x64
2022-02-03T18:05:39.7466071Z   LD_LIBRARY_PATH: /opt/hostedtoolcache/Python/3.10.2/x64/lib
2022-02-03T18:05:39.7466383Z ##[endgroup]
2022-02-03T18:05:39.7845066Z ##[group]Run (! git --no-pager grep -I -no $'cudaStreamSynchronize' --  ./aten ./c10 ':(exclude)aten/src/ATen/test' ':(exclude)c10/cuda/CUDAFunctions.h' || (echo "The above files call raw cuda APIs directly; please use at::cuda wrappers instead"; false))
2022-02-03T18:05:39.7846051Z �[36;1m(! git --no-pager grep -I -no $'cudaStreamSynchronize' --  ./aten ./c10 ':(exclude)aten/src/ATen/test' ':(exclude)c10/cuda/CUDAFunctions.h' || (echo "The above files call raw cuda APIs directly; please use at::cuda wrappers instead"; false))�[0m
2022-02-03T18:05:39.7895730Z shell: /bin/bash -e {0}

---

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

Please report bugs/suggestions to the (internal) Dr. CI Users group.

Click here to manually regenerate this comment.

[github-actions]

Hey swolchok.
You've committed this PR, but it does not have both a 'release notes: ...' and 'topics: ...' label. Please add one of each to the PR. The 'release notes: ...' label should represent the part of PyTorch that this PR changes (fx, autograd, distributed, etc) and the 'topics: ...' label should represent the kind of PR it is (not user facing, new feature, bug fix, perf improvement, etc). The list of valid labels can be found here for the 'release notes: ...' and here for the 'topics: ...'.
For changes that are 'topic: not user facing' there is no need for a release notes label.