Skip to content

Commit 1efc4bb

Browse files
authored
Merge branch 'DeepRec-AI:main' into main
2 parents b750f29 + 3bc930a commit 1efc4bb

File tree

746 files changed

+67530
-12799
lines changed

Some content is hidden

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

746 files changed

+67530
-12799
lines changed

.bazelrc

+15-4
Original file line numberDiff line numberDiff line change
@@ -111,25 +111,29 @@ build --define framework_shared_object=true
111111
build --define open_source_build=true
112112
test --define open_source_build=true
113113

114+
# For workaround https://github.com/bazelbuild/bazel/issues/8772 with Bazel >= 0.29.1
115+
build --java_toolchain=//third_party/toolchains/java:tf_java_toolchain
116+
build --host_java_toolchain=//third_party/toolchains/java:tf_java_toolchain
117+
114118
# Please note that OneDNN on MacOS or windows is still not supported.
115119
# If you would like to use a local OneDNN instead of downloading, please set the
116120
# environment variable "TF_MKL_ROOT" every time before build.
117121
build:mkl --define=build_with_mkl=true --define=enable_mkl=true
118122
build:mkl --define=tensorflow_dnnl_contraction_kernel=0
119-
build:mkl -c opt
123+
build:mkl -c opt --copt=-O3
120124

121125
# config to build OneDNN backend with a user specified threadpool.
122126
build:mkl_threadpool --define=build_with_mkl=true --define=enable_mkl=true
123127
build:mkl_threadpool --define=tensorflow_dnnl_contraction_kernel=0
124128
build:mkl_threadpool --define=build_with_mkl_opensource=true
125129
build:mkl_threadpool --define=build_with_mkldnn_threadpool=true
126-
build:mkl_threadpool -c opt
130+
build:mkl_threadpool -c opt --copt=-O3
127131

128132
# Config setting to build with oneDNN and without the binary blob
129133
build:mkl_opensource_only --define=build_with_mkl=true --define=enable_mkl=true
130134
build:mkl_opensource_only --define=tensorflow_dnnl_contraction_kernel=0
131135
build:mkl_opensource_only --define=build_with_mkl_opensource=true
132-
build:mkl_opensource_only -c opt
136+
build:mkl_opensource_only -c opt --copt=-O3
133137

134138
# This config option is used to enable OneDNN open source library only,
135139
# without depending on OneDNN binary version.
@@ -138,6 +142,13 @@ build:mkl_open_source_only --define=build_with_mkl_dnn_v1_only=true
138142
build:mkl_open_source_only --define=build_with_mkl=true --define=enable_mkl=true
139143
build:mkl_open_source_only --define=tensorflow_dnnl_contraction_kernel=0
140144

145+
# Config setting to build oneDNN with Compute Library for the Arm Architecture (ACL).
146+
# This build is for the inference regime only.
147+
build:mkl_aarch64 --define=build_with_mkl_aarch64=true
148+
build:mkl_aarch64 --define=tensorflow_mkldnn_contraction_kernel=0
149+
build:mkl_aarch64 --define=build_with_openmp=true
150+
build:mkl_aarch64 -c opt
151+
141152
# This config refers to building with CUDA available. It does not necessarily
142153
# mean that we build CUDA op kernels.
143154
build:using_cuda --define=using_cuda=true
@@ -197,7 +208,7 @@ build --define=use_fast_cpp_protos=true
197208
build --define=allow_oversize_protos=true
198209

199210
build --spawn_strategy=standalone
200-
build -c opt
211+
build -c opt --copt=-O3
201212

202213
# Make Bazel print out all options from rc files.
203214
build --announce_rc

.bazelversion

+1-1
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
0.26.1
1+
5.3.1

.github/ISSUE_TEMPLATE/00-bug-performance-issue.md

+4-14
Original file line numberDiff line numberDiff line change
@@ -4,32 +4,22 @@ about: Use this template for reporting a bug or a performance issue.
44

55
---
66

7-
<em>Please make sure that this is a bug. As per our [GitHub Policy](https://github.com/tensorflow/tensorflow/blob/master/ISSUES.md), we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:bug_template</em>
8-
97
**System information**
10-
- Have I written custom code (as opposed to using a stock example script provided in TensorFlow):
11-
- OS Platform and Distribution (e.g., Linux Ubuntu 16.04):
12-
- Mobile device (e.g. iPhone 8, Pixel 2, Samsung Galaxy) if the issue happens on mobile device:
13-
- TensorFlow installed from (source or binary):
14-
- TensorFlow version (use command below):
8+
- OS Platform and Distribution (e.g., Linux Ubuntu 20.04):
9+
- DeepRec version or commit id:
1510
- Python version:
1611
- Bazel version (if compiling from source):
1712
- GCC/Compiler version (if compiling from source):
1813
- CUDA/cuDNN version:
19-
- GPU model and memory:
20-
21-
You can collect some of this information using our environment capture
22-
[script](https://github.com/tensorflow/tensorflow/tree/master/tools/tf_env_collect.sh)
23-
You can also obtain the TensorFlow version with: 1. TF 1.0: `python -c "import
24-
tensorflow as tf; print(tf.GIT_VERSION, tf.VERSION)"` 2. TF 2.0: `python -c
25-
"import tensorflow as tf; print(tf.version.GIT_VERSION, tf.version.VERSION)"`
2614

2715
**Describe the current behavior**
2816

2917
**Describe the expected behavior**
3018

3119
**Code to reproduce the issue**
20+
3221
Provide a reproducible test case that is the bare minimum necessary to generate the problem.
3322

3423
**Other info / logs**
24+
3525
Include any logs or source code that would be helpful to diagnose the problem. If including tracebacks, please include the full traceback. Large logs and files should be attached.

.github/ISSUE_TEMPLATE/10-build-installation-issue.md

+4-12
Original file line numberDiff line numberDiff line change
@@ -4,26 +4,18 @@ about: Use this template for build/installation issues
44

55
---
66

7-
<em>Please make sure that this is a build/installation issue. As per our [GitHub Policy](https://github.com/tensorflow/tensorflow/blob/master/ISSUES.md), we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:build_template</em>
8-
97
**System information**
10-
- OS Platform and Distribution (e.g., Linux Ubuntu 16.04):
11-
- Mobile device (e.g. iPhone 8, Pixel 2, Samsung Galaxy) if the issue happens on mobile device:
12-
- TensorFlow installed from (source or binary):
13-
- TensorFlow version:
8+
- OS Platform and Distribution (e.g., Linux Ubuntu 20.04):
9+
- DeepRec version or commit id:
1410
- Python version:
15-
- Installed using virtualenv? pip? conda?:
1611
- Bazel version (if compiling from source):
1712
- GCC/Compiler version (if compiling from source):
1813
- CUDA/cuDNN version:
19-
- GPU model and memory:
20-
21-
2214

2315
**Describe the problem**
2416

2517
**Provide the exact sequence of commands / steps that you executed before running into the problem**
2618

27-
2819
**Any other info / logs**
29-
Include any logs or source code that would be helpful to diagnose the problem. If including tracebacks, please include the full traceback. Large logs and files should be attached.
20+
21+
Include any logs or source code that would be helpful to diagnose the problem.

.github/ISSUE_TEMPLATE/20-documentation-issue.md

+3-45
Original file line numberDiff line numberDiff line change
@@ -5,51 +5,9 @@ labels: 'type:docs'
55

66
---
77

8-
Thank you for submitting a TensorFlow documentation issue. Per our GitHub
9-
policy, we only address code/doc bugs, performance issues, feature requests, and
10-
build/installation issues on GitHub.
8+
The DeepRec docs are open source! To get involved, read the user documents: https://deeprec.readthedocs.io/en/latest/index.html
119

12-
The TensorFlow docs are open source! To get involved, read the documentation
13-
contributor guide: https://www.tensorflow.org/community/contribute/docs
10+
**URL(s) with the issue**
1411

15-
## URL(s) with the issue:
12+
**Description of issue (what needs changing)**
1613

17-
Please provide a link to the documentation entry, for example:
18-
https://www.tensorflow.org/versions/r2.0/api_docs/python/tf/MyMethod
19-
20-
## Description of issue (what needs changing):
21-
22-
### Clear description
23-
24-
For example, why should someone use this method? How is it useful?
25-
26-
### Correct links
27-
28-
Is the link to the source code correct?
29-
30-
### Parameters defined
31-
32-
Are all parameters defined and formatted correctly?
33-
34-
### Returns defined
35-
36-
Are return values defined?
37-
38-
### Raises listed and defined
39-
40-
Are the errors defined? For example,
41-
https://www.tensorflow.org/versions/r2.0/api_docs/python/tf/feature_column/categorical_column_with_vocabulary_file#raises
42-
43-
### Usage example
44-
45-
Is there a usage example?
46-
47-
### Request visuals, if applicable
48-
49-
Are there currently visuals? If not, will it clarify the content?
50-
51-
### Submit a pull request?
52-
53-
Are you planning to also submit a pull request to fix the issue? See the docs
54-
contributor guide: https://www.tensorflow.org/community/contribute/docs and the
55-
docs style guide: https://www.tensorflow.org/community/contribute/docs_style

.github/ISSUE_TEMPLATE/30-feature-request.md

+1-4
Original file line numberDiff line numberDiff line change
@@ -4,15 +4,12 @@ about: Use this template for raising a feature request
44

55
---
66

7-
<em>Please make sure that this is a feature request. As per our [GitHub Policy](https://github.com/tensorflow/tensorflow/blob/master/ISSUES.md), we only address code/doc bugs, performance issues, feature requests and build/installation issues on GitHub. tag:feature_template</em>
8-
97

108
**System information**
11-
- TensorFlow version (you are using):
9+
- DeepRec version (you are using):
1210
- Are you willing to contribute it (Yes/No):
1311

1412

15-
1613
**Describe the feature and the current behavior/state.**
1714

1815
**Will this change the current api? How?**
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,8 @@
1+
---
2+
name: Other Issues
3+
about: Use this template for any other non-support related issues
4+
5+
---
6+
7+
This template is for miscellaneous issues not covered by the other issue categories.
8+

.github/ISSUE_TEMPLATE/40-tflite-op-request.md

-24
This file was deleted.

.github/ISSUE_TEMPLATE/50-other-issues.md

-13
This file was deleted.
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,55 @@
1+
name: GPU Build Serving
2+
3+
on:
4+
workflow_dispatch:
5+
push:
6+
branches:
7+
- main
8+
paths-ignore:
9+
- 'docs/**'
10+
- 'modelzoo/**'
11+
- 'sparse_operation_kit/**'
12+
- 'triton/**'
13+
- '*.md'
14+
pull_request_review:
15+
types: [submitted]
16+
branches:
17+
- main
18+
paths-ignore:
19+
- 'docs/**'
20+
- 'modelzoo/**'
21+
- 'sparse_operation_kit/**'
22+
- 'triton/**'
23+
- '*.md'
24+
25+
env:
26+
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-gpu-cibuild-py38-cu116-ubuntu20.04
27+
JOBNAME: deeprec-ci-gpu-${{ github.run_id }}
28+
29+
jobs:
30+
build:
31+
if: ${{ github.event_name == 'workflow_dispatch' || github.event_name == 'push' || (github.event_name == 'pull_request_review' && github.event.review.state == 'approved') }}
32+
runs-on: [self-hosted, gpu]
33+
steps:
34+
- name: Checkout Code
35+
uses: actions/checkout@v2
36+
37+
- name: Pull Image And Start Container
38+
if: ${{ success() }}
39+
run:
40+
cibuild/pull_image_and_start_container.sh gpu ${IMAGE} ${JOBNAME}
41+
42+
- name: Upload Source
43+
if: ${{ success() }}
44+
run:
45+
cibuild/upload.sh ${JOBNAME} ../cibuild /mnt
46+
47+
- name: Build Serving
48+
if: ${{ success() }}
49+
run:
50+
docker exec ${JOBNAME} /mnt/cibuild/serving-gpu.sh /mnt
51+
52+
- name: Remove Container
53+
if: ${{ always() }}
54+
run:
55+
cibuild/remove_container.sh ${JOBNAME}

.github/workflows/ubuntu18.04-py3.6-cibuild-build-serving.yaml

+1-1
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ on:
2323
- '*.md'
2424

2525
env:
26-
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py36-ubuntu18.04
26+
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py38-ubuntu20.04
2727
JOBNAME: deeprec-ci-cpu-${{ github.run_id }}
2828

2929
jobs:

.github/workflows/ubuntu18.04-py3.6-cibuild-build-wheel.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ on:
2323
- '*.md'
2424

2525
env:
26-
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py36-ubuntu18.04
26+
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py38-ubuntu20.04
2727
JOBNAME: deeprec-ci-cpu-${{ github.run_id }}
2828

2929
jobs:
@@ -52,4 +52,4 @@ jobs:
5252
- name: Remove Container
5353
if: ${{ always() }}
5454
run:
55-
cibuild/remove_container.sh ${JOBNAME}
55+
cibuild/remove_container.sh ${JOBNAME}

.github/workflows/ubuntu18.04-py3.6-cibuild-c-unit-test.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ on:
2323
- '*.md'
2424

2525
env:
26-
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py36-ubuntu18.04
26+
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py38-ubuntu20.04
2727
JOBNAME: deeprec-ci-cpu-${{ github.run_id }}
2828

2929
jobs:
@@ -52,4 +52,4 @@ jobs:
5252
- name: Remove Container
5353
if: ${{ always() }}
5454
run:
55-
cibuild/remove_container.sh ${JOBNAME}
55+
cibuild/remove_container.sh ${JOBNAME}

.github/workflows/ubuntu18.04-py3.6-cibuild-cc-unit-test.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ on:
2323
- '*.md'
2424

2525
env:
26-
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py36-ubuntu18.04
26+
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py38-ubuntu20.04
2727
JOBNAME: deeprec-ci-cpu-${{ github.run_id }}
2828

2929
jobs:
@@ -52,4 +52,4 @@ jobs:
5252
- name: Remove Container
5353
if: ${{ always() }}
5454
run:
55-
cibuild/remove_container.sh ${JOBNAME}
55+
cibuild/remove_container.sh ${JOBNAME}

.github/workflows/ubuntu18.04-py3.6-cibuild-contrib-unit-test.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ on:
2323
- '*.md'
2424

2525
env:
26-
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py36-ubuntu18.04
26+
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py38-ubuntu20.04
2727
JOBNAME: deeprec-ci-cpu-${{ github.run_id }}
2828

2929
jobs:
@@ -52,4 +52,4 @@ jobs:
5252
- name: Remove Container
5353
if: ${{ always() }}
5454
run:
55-
cibuild/remove_container.sh ${JOBNAME}
55+
cibuild/remove_container.sh ${JOBNAME}

.github/workflows/ubuntu18.04-py3.6-cibuild-core-unit-test.yaml

+2-2
Original file line numberDiff line numberDiff line change
@@ -23,7 +23,7 @@ on:
2323
- '*.md'
2424

2525
env:
26-
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py36-ubuntu18.04
26+
IMAGE: alideeprec/deeprec-cicd:deeprec-dev-cpu-cibuild-py38-ubuntu20.04
2727
JOBNAME: deeprec-ci-cpu-${{ github.run_id }}
2828

2929
jobs:
@@ -52,4 +52,4 @@ jobs:
5252
- name: Remove Container
5353
if: ${{ always() }}
5454
run:
55-
cibuild/remove_container.sh ${JOBNAME}
55+
cibuild/remove_container.sh ${JOBNAME}

0 commit comments

Comments
 (0)