Skip to content

Commit 95f09e8

Browse files
authored
Rebase community TensorFlow. (#1213)
* Rebased community TensorFlow. * Fix rebase error of tf_community. * Fix BUILD and docker env. * Fix build error. * Disable remote cache for testing. * Update CUDA version of CI. * Migrate T4 to A10 for CI. * Clean CI. * Refine code. * Update document. * Update paper authors. * Fix rebase errors. * Add missed tf250 requirement file. * Update torch-pre version. * Update torch-pre version. * Update torch-pre dependency version. * Update torch-pre-cpu version. * Fix dependency error of tensorflow-blade. * Fixdeps for AArch64. * Update bazel deps of tensorflow_blade. * Fix transform-dialect errors. * Update tf250 CI file. * Temporarily disabled some UTs. * Temporarily disable remote cache for testing. * Enable CI remote cache and add python version for tf250_cpu. * Update scaled_dot_product_efficient_attention prototype. * Change pdl-type to transform-type. * Disable the buggy AArch64 schedules, and add DCE, CSE, LICM Transform-Dialect ops. * Update README. * Update README. * Update README. * Update tensorflow blade workspace.
1 parent c74ee95 commit 95f09e8

File tree

257 files changed

+4007
-3933
lines changed

Some content is hidden

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

257 files changed

+4007
-3933
lines changed

.bazelversion

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1 +1 @@
1-
5.1.1
1+
6.1.0

.github/workflows/d2_trt_bench.yml

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,7 @@ jobs:
1515
with:
1616
name: torch-trt-and-d2-benchmark
1717
base_image: nvcr.io/nvidia/pytorch:22.01-py3
18-
device: gpu-t4
18+
device: gpu-a10
1919
dockerfile: docker/cronjobs/Dockerfile.torch.ngc
2020
extra_run_args: --gpus all
2121
exec_command: bash -c "cd pytorch_blade && bash ./benchmark/run_benchmark.sh"

.github/workflows/pytorch110_aarch64.yml

Lines changed: 0 additions & 72 deletions
This file was deleted.

.github/workflows/pytorch112_aarch64.yml

Lines changed: 0 additions & 72 deletions
This file was deleted.

.github/workflows/pytorch112_cpu.yml

Lines changed: 0 additions & 28 deletions
This file was deleted.

.github/workflows/pytorch112_gpu.yml

Lines changed: 0 additions & 29 deletions
This file was deleted.

.github/workflows/pytorch112_neural_engine.yml

Lines changed: 0 additions & 29 deletions
This file was deleted.

.github/workflows/pytorch113_aarch64.yml

Lines changed: 0 additions & 72 deletions
This file was deleted.

.github/workflows/pytorch171_gpu.yml

Lines changed: 0 additions & 27 deletions
This file was deleted.

.github/workflows/pytorch181_cpu.yml

Lines changed: 0 additions & 26 deletions
This file was deleted.

0 commit comments

Comments
 (0)