Skip to content

Commit ab6b724

Browse files
Xia-WeiwenXia-Weiwen
committed
Update on "[Quant] Add fused LinearTanh module for onednn backend"
 **Summary** This PR adds fused `QLinearTanh` module for onednn backend, which will be used for int8 inference with onednn backend. Cannot call this module with other quantization backends otherwise an error is thrown. **Test plan** python test_quantization.py TestStaticQuantizedModule cc jgong5 mingfeima XiaobingSuper sanchitintel ashokei jingxu10 [ghstack-poisoned]
2 parents 1aa42eb + 631a7e5 commit ab6b724

File tree

202 files changed

+5170
-2803
lines changed

Some content is hidden

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

202 files changed

+5170
-2803
lines changed

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -21,6 +21,9 @@ outputs:
2121
is-test-matrix-empty:
2222
description: True if the filtered test configs matrix is empty. False otherwise.
2323
value: ${{ steps.filter.outputs.is-test-matrix-empty }}
24+
keep-going:
25+
description: True if keep-going label was on PR.
26+
value: ${{ steps.filter.outputs.keep-going }}
2427

2528
runs:
2629
using: composite

‎.github/ci_commit_pins/xla.txt‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
06cc6d4f596e764502e3ce5ab998c8986fd52e5b
1+
d43ed3c4d24ea17df0a57fed7f338a9c7b7d1ef0

‎.github/scripts/filter_test_configs.py‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -164,6 +164,7 @@ def main() -> None:
164164
# workflow dispatcher
165165
tag_regex = re.compile(r"^ciflow/\w+/(?P<pr_number>\d+)$")
166166

167+
labels = set()
167168
if pr_number:
168169
# If a PR number is set, query all the labels from that PR
169170
labels = get_labels(int(pr_number))
@@ -202,6 +203,8 @@ def main() -> None:
202203
# quickly check it without the need to parse the JSON string
203204
set_output("is-test-matrix-empty", filtered_test_matrix_len == 0)
204205

206+
set_output("keep-going", "keep-going" in labels)
207+
205208

206209
if __name__ == "__main__":
207210
main()

‎.github/templates/common.yml.j2‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -109,7 +109,7 @@ concurrency:
109109

110110
{%- macro checkout(submodules="recursive", deep_clone=True, directory="", repository="pytorch/pytorch", branch="", checkout_pr_head=True) -%}
111111
- name: Checkout !{{ 'PyTorch' if repository == "pytorch/pytorch" else repository }}
112-
uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9
112+
uses: malfet/checkout@silent-checkout
113113
with:
114114
{%- if branch %}
115115
ref: !{{ branch }}
@@ -127,6 +127,7 @@ concurrency:
127127
{%- if directory %}
128128
path: !{{ directory }}
129129
{%- endif %}
130+
quiet-checkout: true
130131
- name: Clean !{{ 'PyTorch' if repository == "pytorch/pytorch" else repository }} checkout
131132
run: |
132133
# Remove any artifacts from the previous checkouts

‎.github/workflows/_binary-build-linux.yml‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -143,24 +143,26 @@ jobs:
143143
mkdir "${GITHUB_WORKSPACE}"
144144
145145
- name: Checkout PyTorch to pytorch dir
146-
uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9
146+
uses: malfet/checkout@silent-checkout
147147
with:
148148
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
149149
submodules: recursive
150150
path: pytorch
151+
quiet-checkout: true
151152
- name: Clean PyTorch checkout
152153
run: |
153154
# Remove any artifacts from the previous checkouts
154155
git clean -fxd
155156
working-directory: pytorch
156157

157158
- name: Checkout pytorch/builder to builder dir
158-
uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9
159+
uses: malfet/checkout@silent-checkout
159160
with:
160161
ref: main
161162
submodules: recursive
162163
repository: pytorch/builder
163164
path: builder
165+
quiet-checkout: true
164166
- name: Clean pytorch/builder checkout
165167
run: |
166168
# Remove any artifacts from the previous checkouts

‎.github/workflows/_binary-test-linux.yml‎

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ on:
6565
description: Github Token
6666

6767
jobs:
68-
build:
68+
test:
6969
runs-on: ${{ inputs.runs_on }}
7070
timeout-minutes: 240
7171
env:
@@ -140,7 +140,7 @@ jobs:
140140
mkdir "${GITHUB_WORKSPACE}"
141141
142142
- name: Checkout PyTorch to pytorch dir
143-
uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9
143+
uses: malfet/checkout@silent-checkout
144144
with:
145145
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}
146146
submodules: recursive
@@ -152,7 +152,7 @@ jobs:
152152
working-directory: pytorch
153153

154154
- name: Checkout pytorch/builder to builder dir
155-
uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9
155+
uses: malfet/checkout@silent-checkout
156156
with:
157157
ref: main
158158
submodules: recursive

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

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -45,6 +45,7 @@ jobs:
4545
outputs:
4646
test-matrix: ${{ steps.filter.outputs.test-matrix }}
4747
is-test-matrix-empty: ${{ steps.filter.outputs.is-test-matrix-empty }}
48+
keep-going: ${{ steps.filter.outputs.keep-going }}
4849
steps:
4950
- name: Checkout PyTorch
5051
uses: pytorch/pytorch/.github/actions/checkout-pytorch@master
@@ -124,6 +125,7 @@ jobs:
124125
SHARD_NUMBER: ${{ matrix.shard }}
125126
NUM_TEST_SHARDS: ${{ matrix.num_shards }}
126127
PR_BODY: ${{ github.event.pull_request.body }}
128+
CONTINUE_THROUGH_ERROR: ${{ needs.filter.outputs.keep-going }}
127129
SCCACHE_BUCKET: ossci-compiler-cache-circleci-v2
128130
SCCACHE_S3_KEY_PREFIX: ${{ github.workflow }}
129131
SHM_SIZE: ${{ contains(inputs.build-environment, 'cuda') && '2g' || '1g' }}
@@ -176,6 +178,7 @@ jobs:
176178
-e NUM_TEST_SHARDS \
177179
-e PR_BODY \
178180
-e COMMIT_MESSAGES \
181+
-e CONTINUE_THROUGH_ERROR \
179182
-e PYTORCH_RETRY_TEST_CASES \
180183
-e PYTORCH_OVERRIDE_FLAKY_SIGNAL \
181184
-e PR_LABELS \

‎.github/workflows/create_release.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,7 @@ jobs:
1818
name: Create Release
1919
runs-on: ubuntu-latest
2020
steps:
21-
- uses: zhouzhuojie/checkout@05b13c9a0d21f08f6d5e64a1d5042246d13619d9
21+
- uses: malfet/checkout@silent-checkout
2222
with:
2323
submodules: 'recursive'
2424
ref: ${{ github.event_name == 'pull_request' && github.event.pull_request.head.sha || github.sha }}

‎.github/workflows/docker-release.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -75,7 +75,7 @@ jobs:
7575
- name: Setup job specific variables
7676
run: |
7777
set -eou pipefail
78-
# To get QEMU binaries in our PATh
78+
# To get QEMU binaries in our PATH
7979
echo "${RUNNER_TEMP}/bin" >> "${GITHUB_PATH}"
8080
# Generate PyTorch version to use
8181
echo "PYTORCH_VERSION=$(python3 .github/scripts/generate_pytorch_version.py)" >> "${GITHUB_ENV}"

‎.github/workflows/generated-linux-binary-libtorch-cxx11-abi-nightly.yml‎

Lines changed: 16 additions & 8 deletions
Some generated files are not rendered by default. Learn more about customizing how changed files appear on GitHub.

0 commit comments

Comments
 (0)