Commit Graph

107793 Commits

Author SHA1 Message Date
Mihai Maruseac
e0b8bbee7a
Merge pull request #48729 from geetachavan1/cherrypicks_RKGNE
[Cherrypicke:2.5]Disable pad_op_test for macos. Test times out during 2.5 release
2021-04-23 17:13:00 -07:00
Mihai Maruseac
1390a49a13
Merge pull request #48728 from geetachavan1/cherrypicks_P0S4H
[CherryPick 2.5] 3.9 compatibility fix for AST unparsing
2021-04-23 17:12:20 -07:00
Geeta Chavan
4ea9a27a1c Disable pad_op_test for macos. Test times out during 2.5 release
PiperOrigin-RevId: 370178263
Change-Id: I57989213810709d73d9e503b40b0f871e9006a05
2021-04-23 16:35:14 -07:00
Dan Moldovan
e162761030 [CherryPick 2.5] 3.9 compatibility fix for AST unparsing 2021-04-23 15:45:38 -07:00
Mihai Maruseac
eb3b2914c6
Merge pull request #48443 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_47491
[r2.5 port][ROCm] Port PR 47491 to r2.5
2021-04-23 08:24:22 -07:00
TensorFlower Gardener
142d5a7b88 Merge pull request #47491 from ROCmSoftwarePlatform:google_upstream_rocm_hiprand_rocrand_210301
PiperOrigin-RevId: 366329409
Change-Id: I3ac36219fab7735fc46e893f395263a2ed6335da
2021-04-22 16:10:57 -07:00
Deven Desai
dcf483cea6 Updating rocm_configure.bzl to do the right thing based on whether or not the softlinks to <rocm_path>/include/[hiprand|rocrand] exist. Removing the workaroung to handle the same in Dockerfile.rocm, since that is no longer required (as a consequence of the change in rocm_configure.bzl 2021-04-22 16:08:18 -07:00
Deven Desai
00252bc0c1 Updates to account for the removal of $ROCM_PATH/include/hiprand,rocrand dirs
The following dirs (which are soft-links really) will removed in a (near) future ROCm release. This commit updates `rocm_configure.bzl` to account for that change by explicitly copying over the header files (for `local_config_rocm` repo) from the following directories instead
* $ROCM_PATH/hiprand/include
* $ROCM_PATH/rocrand/include

Unfortunately making the above change also breaks the build, if the directories that will be going away, are still present. This is because we end up with two `make_copy_dir_rule`s in `rocm_configure.bzl` that try to write to the same destination dirs

```
ERROR: Traceback (most recent call last):
	File "/root/.cache/bazel/_bazel_root/efb88f6336d9c4a18216fb94287b8d97/external/local_config_rocm/rocm/BUILD", line 1588, column 8, in <toplevel>
		genrule(
Error in genrule: generated file 'rocm/include/hiprand/hiprand.h' in rule 'hiprand-include' conflicts with existing generated file from rule 'rocm-include', defined at /root/.cache/bazel/_bazel_root/efb88f6336d9c4a18216fb94287b8d97/external/local_config_rocm/rocm/BUILD:175:8
```

In order to get around this error, this commit is also updating the `Dockerfile.rocm` to explicitly delete those two dirs from the `$ROCM_PATH/include` dir, immediately after installing ROCm
2021-04-22 16:05:35 -07:00
Mihai Maruseac
bfda315296
Merge pull request #48543 from alenik01/mkl_aarch64_update
[r2.5] Update of oneDNN version for mkl_aarch64 build target
2021-04-22 15:47:31 -07:00
Mihai Maruseac
2e3959b348
Merge pull request #48569 from geetachavan1/cherrypicks_2NJI2
[Cherrypick:2.5]Copy execution plan in graph partitioner. Also adds code to model_builder_test that fails without this fix.
2021-04-22 15:29:59 -07:00
Mihai Maruseac
f09d694a57
Merge pull request #48447 from ROCmSoftwarePlatform/google-upstream-rocm-amp-r2.5
[r2.5 port][ROCm] Port PR#47650 and PR#48441 to r2.5
2021-04-22 15:29:34 -07:00
Mihai Maruseac
c87875ede5
Merge pull request #48445 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_48346
[r2.5 port][ROCm] Port PR 48346 to r2.5
2021-04-22 15:29:18 -07:00
Mihai Maruseac
d25f3bc1e8
Merge pull request #48444 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_47980
[r2.5 port][ROCm] Port PR 47980 to r2.5
2021-04-22 15:29:10 -07:00
Mihai Maruseac
467b9d837c
Merge pull request #48493 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_48419
[r2.5 port][ROCm] Port PR 48419 to r2.5
2021-04-22 15:28:18 -07:00
Mihai Maruseac
30afb3b823
Merge pull request #48495 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_48336
[r2.5 port][ROCm] Port PR 48336 to r2.5
2021-04-22 15:27:55 -07:00
Mihai Maruseac
851691035f
Merge pull request #48442 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_47646
[r2.5 port][ROCm] Port PR 47646 to r2.5
2021-04-22 15:27:35 -07:00
Mihai Maruseac
8e0516e743
Merge pull request #48507 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_48187
[r2.5 port][ROCm] Port PR 48187 to r2.5
2021-04-22 15:26:58 -07:00
Mihai Maruseac
95d88654e0
Merge pull request #48440 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_47937
[r2.5 port][ROCm] Port PR 47937 to r2.5
2021-04-22 15:26:44 -07:00
Mihai Maruseac
6dda2edc18
Merge pull request #48628 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_47851
[r2.5 port][ROCm] Port PR 47851 to r2.5
2021-04-22 15:26:23 -07:00
Mihai Maruseac
952d3b8748
Merge pull request #48618 from geetachavan1/cherrypicks_GAEEN
[Cherrypick:2.5]Switch absl to `lts_2021_03_24` LTS branch
2021-04-22 15:26:09 -07:00
Mihai Maruseac
790bbcbf98
Merge pull request #48622 from wchargin/cherrypicks_VFBWH
2.5.0-rc1 cherry-pick request: Update tensorboard dependency to 2.5.x
2021-04-22 15:26:03 -07:00
Mihai Maruseac
3aba8ab14d
Merge pull request #48645 from ROCmSoftwarePlatform/google_upstream_r25_port_pr_47508
[r2.5 port][ROCm] Port PR 47508 to r2.5
2021-04-22 15:25:51 -07:00
Mihai Maruseac
b95a352416
Merge pull request #48651 from penpornk/cherrypicks_WWU73
r2.5-rc2 cherry-pick request: [Intel MKL] Log a warning message when int8 is not supported
2021-04-22 15:25:39 -07:00
Mihai Maruseac
f4b96ff8b3
Merge pull request #48653 from geetachavan1/cherrypicks_GFW62
[CherryPick:r2.5] Use ast instead of astunparse for python 3.9+.
2021-04-22 15:25:27 -07:00
Mihai Maruseac
02ccebd433
Merge pull request #48655 from geetachavan1/cherrypicks_N2EJ7
[CherryPick:r2.5]Allow ProfilerFactory to return a nullptr ProfilerInterface
2021-04-22 15:25:03 -07:00
Mihai Maruseac
21846186dd
Merge pull request #48707 from geetachavan1/cherrypicks_N8Y87
[CHerryPick:r2.5] Keras Training API: Make sure to update the stateful metrics in the Progbar once the MetricsContainer is built so the metrics are not averaged over an epoch.
2021-04-22 15:24:46 -07:00
Mihai Maruseac
ddb7698b88
Merge pull request #48709 from tensorflow-jenkins/version-numbers-2.5.0rc2-30521
Update version numbers for TensorFlow 2.5.0-rc2
2021-04-22 15:24:28 -07:00
TensorFlow Release Automation
17ca02ce03 Update version numbers to 2.5.0-rc2 2021-04-22 14:32:53 -07:00
Rick Chao
4e80602256 Keras Training API: Make sure to update the stateful metrics in the Progbar once the MetricsContainer is built so the metrics are not averaged over an epoch.
Fixes https://github.com/tensorflow/tensorflow/issues/47099

Unit test element courtesy of the reporter of https://github.com/tensorflow/tensorflow/issues/47099.

PiperOrigin-RevId: 369572207
Change-Id: I308fcccac5164ab92a70e74d27ecd0f0cbba275a
2021-04-22 12:17:38 -07:00
Jose Baiocchi
fb4e94e3fd Allow ProfilerFactory to return a nullptr ProfilerInterface
PiperOrigin-RevId: 368888650
Change-Id: I6d76435f8dfb376c5c8108d778efbf11d2e2b939
2021-04-20 17:05:26 -07:00
Jun Xu
7b58a54805 [CherryPick:r2.5] Use ast instead of astunparse for python 3.9+. 2021-04-20 13:44:57 -07:00
xiaohong1031
74ba5d23d2 fix warning message 2021-04-20 13:17:11 -07:00
xiaohong1031
30fdffd2ae coding style fix and minor changes of warning message per review suggestion 2021-04-20 13:17:11 -07:00
xiaohong1031
ab6c547c65 Add a warning message for build/setting which does not support INT8 2021-04-20 13:17:10 -07:00
TensorFlower Gardener
a90a1dcfff Merge pull request #47508 from ROCmSoftwarePlatform:google_upstream_rocm_roctracer_210302
PiperOrigin-RevId: 368391195
Change-Id: I970f9db1f08c9e82d16178204bab649081d1d65b
2021-04-20 02:55:33 +00:00
Deven Desai
457df8e17a Addressing PR feedback 2021-04-20 02:51:33 +00:00
Deven Desai
6a74de1f86 Addressing PR feedback - removing forward class decl in header 2021-04-20 02:51:22 +00:00
Deven Desai
ecdbd93e09 Addressing PR feedback - adding unique_ptr 2021-04-20 02:51:11 +00:00
Deven Desai
1b91b19ab9 Addressing PR feedback 2021-04-20 02:50:58 +00:00
Deven Desai
880f1ca785 Replacing 2020 with 2021 in the header copyright 2021-04-20 02:50:47 +00:00
Deven Desai
2813bf5105 Replacing uint32/int32/uint64/int64 with their _t counterparts 2021-04-20 02:50:36 +00:00
Deven Desai
4f503631bd Addressing PR feedback 2021-04-20 02:50:24 +00:00
Deven Desai
8c94fd0426 Re-enabling profiler related unit-tests / sub-tests, that were disabled on the ROCm platform 2021-04-20 02:50:13 +00:00
Deven Desai
b50574c455 Adding device tracing support for ROCm 2021-04-20 02:50:02 +00:00
Deven Desai
4c93486ea2 renaming device_tracer.cc to device_tracer_rocm.cc 2021-04-20 02:49:51 +00:00
Deven Desai
bd7e773e95 Add roctracer_wrapper library in ROCm stream executor 2021-04-20 02:49:40 +00:00
Deven Desai
505751314d Adding roctracer to local_config_rocm repo 2021-04-20 02:49:26 +00:00
Deven Desai
64d578a77d Addressing PR feedback 2021-04-20 01:54:18 +00:00
Deven Desai
8d5eeb6189 Addressing PR feedback comments 2021-04-20 01:54:18 +00:00
Eugene Kuznetsov
5a336a4150 Switching to rocblas_gemm_ex for MFMA-enabled architectures 2021-04-20 01:54:18 +00:00