Skip to content

Commit 78bb6d6

Browse files
XiaobingSuperXiaobingSuper
committed
Update on "add mkldnn batch_norm backward"
Differential Revision: [D22440965](https://our.internmc.facebook.com/intern/diff/D22440965) [ghstack-poisoned]
2 parents 3fcf455 + 756047b commit 78bb6d6

File tree

295 files changed

+13601
-7327
lines changed

Some content is hidden

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

295 files changed

+13601
-7327
lines changed

‎.circleci/cimodel/data/pytorch_build_data.py‎

Lines changed: 0 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -3,9 +3,6 @@
33

44
CONFIG_TREE_DATA = [
55
("xenial", [
6-
(None, [
7-
X("nightly"),
8-
]),
96
("rocm", [
107
("3.3", [
118
X("3.6"),

‎.circleci/cimodel/data/simple/android_definitions.py‎

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,4 @@
1-
import cimodel.data.simple.util.branch_filters
1+
import cimodel.data.simple.util.branch_filters as branch_filters
22
from cimodel.data.simple.util.docker_constants import DOCKER_IMAGE_NDK
33

44

@@ -37,7 +37,7 @@ def gen_tree(self):
3737
}
3838

3939
if self.is_master_only:
40-
props_dict["filters"] = cimodel.data.simple.util.branch_filters.gen_filter_dict()
40+
props_dict["filters"] = branch_filters.gen_filter_dict(branch_filters.NON_PR_BRANCH_LIST)
4141

4242
return [{self.template_name: props_dict}]
4343

@@ -47,12 +47,14 @@ def __init__(self,
4747
job_name,
4848
template_name,
4949
dependencies,
50-
is_master_only=True):
50+
is_master_only=True,
51+
is_pr_only=False):
5152

5253
self.job_name = job_name
5354
self.template_name = template_name
5455
self.dependencies = dependencies
5556
self.is_master_only = is_master_only
57+
self.is_pr_only = is_pr_only
5658

5759
def gen_tree(self):
5860

@@ -62,7 +64,9 @@ def gen_tree(self):
6264
}
6365

6466
if self.is_master_only:
65-
props_dict["filters"] = cimodel.data.simple.util.branch_filters.gen_filter_dict()
67+
props_dict["filters"] = branch_filters.gen_filter_dict(branch_filters.NON_PR_BRANCH_LIST)
68+
elif self.is_pr_only:
69+
props_dict["filters"] = branch_filters.gen_filter_dict(branch_filters.PR_BRANCH_LIST)
6670

6771
return [{self.template_name: props_dict}]
6872

@@ -77,7 +81,14 @@ def gen_tree(self):
7781
"pytorch-linux-xenial-py3-clang5-android-ndk-r19c-gradle-build-x86_32",
7882
"pytorch_android_gradle_build-x86_32",
7983
["pytorch_linux_xenial_py3_clang5_android_ndk_r19c_x86_32_build"],
80-
is_master_only=False),
84+
is_master_only=False,
85+
is_pr_only=True),
86+
AndroidGradleJob(
87+
"pytorch-linux-xenial-py3-clang5-android-ndk-r19c-gradle-custom-build-single",
88+
"pytorch_android_gradle_custom_build_single",
89+
[],
90+
is_master_only=False,
91+
is_pr_only=True),
8192
AndroidGradleJob(
8293
"pytorch-linux-xenial-py3-clang5-android-ndk-r19c-gradle-build",
8394
"pytorch_android_gradle_build",

‎.circleci/cimodel/data/simple/docker_definitions.py‎

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
"pytorch-linux-bionic-py3.6-clang9",
1212
"pytorch-linux-bionic-cuda10.2-cudnn7-py3.6-clang9",
1313
"pytorch-linux-bionic-py3.8-gcc9",
14+
"pytorch-linux-bionic-rocm3.5.1-py3.6",
1415
"pytorch-linux-xenial-cuda10-cudnn7-py3-gcc7",
1516
"pytorch-linux-xenial-cuda10.1-cudnn7-py3-gcc7",
1617
"pytorch-linux-xenial-cuda10.2-cudnn7-py3-gcc7",
@@ -25,8 +26,8 @@
2526
"pytorch-linux-xenial-py3.6-gcc5.4",
2627
"pytorch-linux-xenial-py3.6-gcc7.2",
2728
"pytorch-linux-xenial-py3.6-gcc7",
28-
"pytorch-linux-xenial-pynightly",
2929
"pytorch-linux-xenial-rocm3.3-py3.6",
30+
"pytorch-linux-xenial-rocm3.5.1-py3.6",
3031
]
3132

3233

‎.circleci/cimodel/data/simple/util/branch_filters.py‎

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -4,6 +4,11 @@
44
r"/release\/.*/",
55
]
66

7+
PR_BRANCH_LIST = [
8+
r"/gh\/.*\/head/",
9+
r"/pull\/.*/",
10+
]
11+
712
RC_PATTERN = r"/v[0-9]+(\.[0-9]+)*-rc[0-9]+/"
813

914
def gen_filter_dict(

‎.circleci/cimodel/data/simple/util/docker_constants.py‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22

33
# ARE YOU EDITING THIS NUMBER? MAKE SURE YOU READ THE GUIDANCE AT THE
44
# TOP OF .circleci/config.yml
5-
DOCKER_IMAGE_TAG = "fff7795428560442086f7b2bb6004b65245dc11a"
5+
DOCKER_IMAGE_TAG = "ab1632df-fa59-40e6-8c23-98e004f61148"
66

77
DOCKER_IMAGE_TAG_ROCM = "be76e8fd-44e2-484d-b090-07e0cc3a56f0"
88

0 commit comments

Comments
 (0)