Skip to content

Commit 64f9d46

Browse files
H-HuangH-Huang
committed
Update on "[13/N] Update gather with CPU/CUDA implementations"
Differential Revision: [D40181612](https://our.internmc.facebook.com/intern/diff/D40181612) [ghstack-poisoned]
2 parents 3f83c9f + 9a4c6bf commit 64f9d46

File tree

609 files changed

+53596
-36166
lines changed

Some content is hidden

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

609 files changed

+53596
-36166
lines changed

‎.circleci/scripts/setup_ci_environment.sh‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,7 +32,7 @@ if ! command -v aws >/dev/null; then
3232
fi
3333

3434
if [ -n "${USE_CUDA_DOCKER_RUNTIME:-}" ]; then
35-
DRIVER_FN="NVIDIA-Linux-x86_64-515.57.run"
35+
DRIVER_FN="NVIDIA-Linux-x86_64-515.76.run"
3636
wget "https://s3.amazonaws.com/ossci-linux/nvidia_driver/$DRIVER_FN"
3737
sudo /bin/bash "$DRIVER_FN" -s --no-drm || (sudo cat /var/log/nvidia-installer.log && false)
3838
nvidia-smi

‎.github/actions/filter-test-configs/action.yml‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -52,7 +52,8 @@ runs:
5252
.github/scripts/filter_test_configs.py \
5353
--test-matrix "${{ inputs.test-matrix }}" \
5454
--pr-number "${{ github.event.pull_request.number }}" \
55-
--tag "${{ steps.parse-ref.outputs.tag }}"
55+
--tag "${{ steps.parse-ref.outputs.tag }}" \
56+
--event-name "${{ github.event_name }}"
5657
5758
- name: Print the filtered test matrix
5859
shell: bash

‎.github/actions/upload-test-artifacts/action.yml‎

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -139,3 +139,4 @@ runs:
139139
retention-days: 14
140140
if-no-files-found: ignore
141141
path: usage_log.txt
142+
continue-on-error: true

‎.github/auto_request_review.yml‎

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@ reviewers:
77
- wconstab
88
- anjali411
99
- albanD
10-
- Krovatkin
1110
- miladm
1211
- bdhirsh
1312

‎.github/ci_commit_pins/triton.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
5ca1ed01016530056c4507661c24d6c21efc983d
1+
0d7e7532279e45672555e344646f5c19c3972331

‎.github/ci_commit_pins/vision.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
cba1c011a87dd14af10f97bcb113fa09a8e2b396
1+
e64784cdea465d833d9d0f66dc73d7abe217933d

‎.github/ci_commit_pins/xla.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
1e9b8bdc75114ac6c16305c970be37a1cd2fdb1c
1+
5ff192eac058d54cc5f22c8ddf9cf6acfa51325d

‎.github/labeler.yml‎

Lines changed: 34 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,3 +15,37 @@
1515
- torch/_subclasses/fake_tensor.py
1616
- torch/_subclasses/fake_utils.py
1717
- torch/_subclasses/meta_utils.py
18+
19+
"module: cpu":
20+
- aten/src/ATen/cpu/**
21+
- aten/src/ATen/native/cpu/**
22+
- aten/src/ATen/native/quantized/cpu/**
23+
- aten/src/ATen/native/Convolution*.cpp
24+
- aten/src/ATen/native/mkldnn/**
25+
- torch/cpu/**
26+
- torch/utils/mkldnn.py
27+
- test/test_mkldnn.py
28+
29+
"module: mkldnn":
30+
- third_party/ideep
31+
- caffe2/ideep/**
32+
- caffe2/python/ideep/**
33+
- cmake/Modules/FindMKLDNN.cmake
34+
- third_party/mkl-dnn.BUILD
35+
- torch/csrc/jit/codegen/onednn/**
36+
- test/test_jit_llga_fuser.py
37+
38+
"module: amp (automated mixed precision)":
39+
- torch/amp/**
40+
- aten/src/ATen/autocast_mode.*
41+
- torch/csrc/jit/passes/autocast.cpp
42+
- test/test_autocast.py
43+
44+
"NNC":
45+
- torch/csrc/jit/tensorexpr/**
46+
47+
"oncall: quantization":
48+
- torch/ao/quantization/**
49+
- torch/quantization/**
50+
- aten/src/ATen/quantized/**
51+
- aten/src/ATen/native/quantized/cpu/**

‎.github/merge_rules.yaml‎

Lines changed: 26 additions & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -35,6 +35,7 @@
3535
- csarofeen
3636
- ngimel
3737
- jjsjann123
38+
- kevinstephano
3839
- ptrblck
3940
mandatory_checks_name:
4041
- EasyCLA
@@ -240,9 +241,12 @@
240241
- third_party/ideep
241242
- caffe2/ideep/**
242243
- caffe2/python/ideep/**
244+
- cmake/Modules/FindMKLDNN.cmake
245+
- third_party/mkl-dnn.BUILD
243246
approved_by:
244247
- XiaobingSuper
245-
- yanbing-j
248+
- jgong5
249+
- mingfeima
246250
mandatory_checks_name:
247251
- EasyCLA
248252
- Lint
@@ -255,6 +259,7 @@
255259
approved_by:
256260
- sanchitintel
257261
- chunyuan-w
262+
- jgong5
258263
mandatory_checks_name:
259264
- EasyCLA
260265
- Lint
@@ -267,9 +272,11 @@
267272
- aten/src/ATen/native/quantized/cpu/**
268273
- aten/src/ATen/native/Convolution*.cpp
269274
- aten/src/ATen/native/mkldnn/**
275+
- test/test_mkldnn.py
270276
approved_by:
271277
- mingfeima
272278
- XiaobingSuper
279+
- jgong5
273280
mandatory_checks_name:
274281
- EasyCLA
275282
- Lint
@@ -282,7 +289,7 @@
282289
- test/test_mkldnn.py
283290
approved_by:
284291
- leslie-fang-intel
285-
- CaoE
292+
- jgong5
286293
mandatory_checks_name:
287294
- EasyCLA
288295
- Lint
@@ -296,7 +303,18 @@
296303
- test/test_autocast.py
297304
approved_by:
298305
- leslie-fang-intel
299-
- CaoE
306+
- jgong5
307+
mandatory_checks_name:
308+
- EasyCLA
309+
- Lint
310+
- pull
311+
312+
- name: NNC
313+
patterns:
314+
- torch/csrc/jit/tensorexpr/**
315+
approved_by:
316+
- EikanWang
317+
- jgong5
300318
mandatory_checks_name:
301319
- EasyCLA
302320
- Lint
@@ -307,11 +325,12 @@
307325
- torch/csrc/lazy/**
308326
- test/cpp/lazy/**
309327
- test/lazy/**
310-
- codegen/api/lazy.py
311-
- codegen/dest/lazy_ir.py
312-
- codegen/dest/lazy_ts_lowering.py
313-
- codegen/gen_lazy_tensor.py
328+
- torchgen/api/lazy.py
329+
- torchgen/dest/lazy_ir.py
330+
- torchgen/dest/lazy_ts_lowering.py
331+
- torchgen/gen_lazy_tensor.py
314332
- aten/src/ATen/native/ts_native_functions.yaml
333+
- .github/ci_commit_pins/xla.txt
315334
approved_by:
316335
- alanwaketan
317336
- JackCaoG

‎.github/requirements/README.md‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,3 +17,6 @@ The list of support files are as follows:
1717
test jobs to setup the conda environment
1818
* conda-env-macOS-X64. This is use by MacOS (x86-64) build and test
1919
jobs to setup the conda environment
20+
* Pip:
21+
* pip-requirements-macOS.txt. This is used by MacOS build and test jobs to
22+
setup the pip environment

0 commit comments

Comments
 (0)