Skip to content

Commit bc3ffca

Browse files
tugsbayasgalantugsbayasgalan
committed
Update on "[BE] Make maybe_aliasing_or_mutating proper tag"
 For better tracking, we need to make maybe aliasing/mutating ops with proper tag. We need to special case native_batch_norm because it is not a CIA but has a wrong schema. I guess native_batch_norm will be removed at some point, so until then we just keep it around. Differential Revision: [D60347117](https://our.internmc.facebook.com/intern/diff/D60347117/) [ghstack-poisoned]
2 parents 5d86bdf + 394eae6 commit bc3ffca

File tree

325 files changed

+5937
-2250
lines changed

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

325 files changed

+5937
-2250
lines changed

‎.ci/docker/common/install_cuda.sh‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@
33
set -ex
44

55
NCCL_VERSION=v2.21.5-1
6-
CUDNN_VERSION=9.1.0.70
6+
CUDNN_VERSION=9.5.1.17
77

88
function install_cusparselt_040 {
99
# cuSparseLt license: https://docs.nvidia.com/cuda/cusparselt/license.html
@@ -39,6 +39,7 @@ function install_cusparselt_062 {
3939
}
4040

4141
function install_118 {
42+
CUDNN_VERSION=9.1.0.70
4243
echo "Installing CUDA 11.8 and cuDNN ${CUDNN_VERSION} and NCCL ${NCCL_VERSION} and cuSparseLt-0.4.0"
4344
rm -rf /usr/local/cuda-11.8 /usr/local/cuda
4445
# install CUDA 11.8.0 in the same container
@@ -105,6 +106,7 @@ function install_121 {
105106
}
106107

107108
function install_124 {
109+
CUDNN_VERSION=9.1.0.70
108110
echo "Installing CUDA 12.4.1 and cuDNN ${CUDNN_VERSION} and NCCL ${NCCL_VERSION} and cuSparseLt-0.6.2"
109111
rm -rf /usr/local/cuda-12.4 /usr/local/cuda
110112
# install CUDA 12.4.1 in the same container

‎.ci/docker/common/install_cuda_aarch64.sh‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,7 @@
44
set -ex
55

66
NCCL_VERSION=v2.21.5-1
7-
CUDNN_VERSION=9.1.0.70
7+
CUDNN_VERSION=9.5.1.17
88

99
function install_cusparselt_062 {
1010
# cuSparseLt license: https://docs.nvidia.com/cuda/cusparselt/license.html
@@ -18,6 +18,7 @@ function install_cusparselt_062 {
1818
}
1919

2020
function install_124 {
21+
CUDNN_VERSION=9.1.0.70
2122
echo "Installing CUDA 12.4.1 and cuDNN ${CUDNN_VERSION} and NCCL ${NCCL_VERSION} and cuSparseLt-0.6.2"
2223
rm -rf /usr/local/cuda-12.4 /usr/local/cuda
2324
# install CUDA 12.4.1 in the same container

‎.ci/docker/common/install_cudnn.sh‎

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -4,7 +4,9 @@ if [[ -n "${CUDNN_VERSION}" ]]; then
44
# cuDNN license: https://developer.nvidia.com/cudnn/license_agreement
55
mkdir tmp_cudnn
66
pushd tmp_cudnn
7-
if [[ ${CUDA_VERSION:0:2} == "12" ]]; then
7+
if [[ ${CUDA_VERSION:0:4} == "12.6" ]]; then
8+
CUDNN_NAME="cudnn-linux-x86_64-9.5.1.17_cuda12-archive"
9+
elif [[ ${CUDA_VERSION:0:2} == "12" ]]; then
810
CUDNN_NAME="cudnn-linux-x86_64-9.1.0.70_cuda12-archive"
911
elif [[ ${CUDA_VERSION:0:2} == "11" ]]; then
1012
CUDNN_NAME="cudnn-linux-x86_64-9.1.0.70_cuda11-archive"

‎.ci/pytorch/test.sh‎

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1410,7 +1410,11 @@ if ! [[ "${BUILD_ENVIRONMENT}" == *libtorch* || "${BUILD_ENVIRONMENT}" == *-baze
14101410
(cd test && python -c "import torch; print(torch.__config__.show())")
14111411
(cd test && python -c "import torch; print(torch.__config__.parallel_info())")
14121412
fi
1413-
if [[ "${BUILD_ENVIRONMENT}" == *aarch64* && "${TEST_CONFIG}" != *perf_cpu_aarch64* ]]; then
1413+
if [[ "${TEST_CONFIG}" == *numpy_2* ]]; then
1414+
# Install numpy-2.0.2 and test inductor tracing
1415+
python -mpip install --pre numpy==2.0.2
1416+
python test/run_test.py --include dynamo/test_unspec.py
1417+
elif [[ "${BUILD_ENVIRONMENT}" == *aarch64* && "${TEST_CONFIG}" != *perf_cpu_aarch64* ]]; then
14141418
test_linux_aarch64
14151419
elif [[ "${TEST_CONFIG}" == *backward* ]]; then
14161420
test_forward_backward_compatibility

‎.github/ci_commit_pins/xla.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
f71c02d1f457d58371e013632efb016c01bd1866
1+
2ec22641e390cda25ec7c61fcbce07507727d584

‎.github/scripts/generate_binary_build_matrix.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -73,7 +73,7 @@
7373
"nvidia-cuda-nvrtc-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7474
"nvidia-cuda-runtime-cu12==12.6.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7575
"nvidia-cuda-cupti-cu12==12.6.80; platform_system == 'Linux' and platform_machine == 'x86_64' | "
76-
"nvidia-cudnn-cu12==9.1.0.70; platform_system == 'Linux' and platform_machine == 'x86_64' | "
76+
"nvidia-cudnn-cu12==9.5.1.17; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7777
"nvidia-cublas-cu12==12.6.3.3; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7878
"nvidia-cufft-cu12==11.3.0.4; platform_system == 'Linux' and platform_machine == 'x86_64' | "
7979
"nvidia-curand-cu12==10.3.7.77; platform_system == 'Linux' and platform_machine == 'x86_64' | "

‎.github/workflows/_linux-test.yml‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -330,6 +330,14 @@ jobs:
330330
test_config: ${{ matrix.config }}
331331
job_identifier: ${{ github.workflow }}_${{ inputs.build-environment }}
332332

333+
- name: Upload the benchmark results
334+
uses: pytorch/test-infra/.github/actions/upload-benchmark-results@main
335+
with:
336+
benchmark-results-dir: test/test-reports
337+
dry-run: false
338+
schema-version: v3
339+
github-token: ${{ secrets.GITHUB_TOKEN }}
340+
333341
- name: Print remaining test logs
334342
shell: bash
335343
if: always() && steps.test.conclusion

‎.github/workflows/_mac-test.yml‎

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -223,6 +223,14 @@ jobs:
223223
use-gha: true
224224
file-suffix: ${{ github.job }}-${{ matrix.config }}-${{ matrix.shard }}-${{ matrix.num_shards }}-${{ matrix.runner }}_${{ steps.get-job-id.outputs.job-id }}
225225

226+
- name: Upload the benchmark results
227+
uses: pytorch/test-infra/.github/actions/upload-benchmark-results@main
228+
with:
229+
benchmark-results-dir: test/test-reports
230+
dry-run: false
231+
schema-version: v3
232+
github-token: ${{ secrets.GITHUB_TOKEN }}
233+
226234
- name: Clean up disk space
227235
if: always()
228236
continue-on-error: true

‎.github/workflows/generated-linux-binary-manywheel-main.yml‎

Lines changed: 1 addition & 1 deletion
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)