Jinzhe Zeng
b3a8fdbcb7
resolve gcc_host_compiler_path in a symlink directory
...
Resolves a missing dependency declarations error, when gcc_host_compiler_path is in a symlink directory resolving to other directories.
2022-06-04 19:06:58 -04:00
A. Unique TensorFlower
72177a67c2
Automated visibility attribute cleanup.
...
PiperOrigin-RevId: 452979650
2022-06-04 13:58:37 -07:00
TensorFlower Gardener
b219f97a5e
Merge pull request #56357 from mseth10:master-arm
...
PiperOrigin-RevId: 452979629
2022-06-04 13:54:28 -07:00
A. Unique TensorFlower
8ece8fd02a
Automated visibility attribute cleanup.
...
PiperOrigin-RevId: 452970706
2022-06-04 11:55:39 -07:00
A. Unique TensorFlower
fe29e4e2db
Update TFRT dependency to use revision
...
62a3081341 .
PiperOrigin-RevId: 452961053
2022-06-04 09:39:32 -07:00
Benjamin Kramer
faed056f87
Integrate LLVM at llvm/llvm-project@dd2362a8ba
...
Updates LLVM usage to match
[dd2362a8bab3](https://github.com/llvm/llvm-project/commit/dd2362a8bab3 )
PiperOrigin-RevId: 452960213
2022-06-04 09:28:16 -07:00
Manu Seth
b1c110c7af
update arm dockerfile arg name
2022-06-04 11:38:44 +00:00
A. Unique TensorFlower
5e66183337
compat: Update forward compatibility horizon to 2022-06-04
...
PiperOrigin-RevId: 452920991
2022-06-04 02:08:11 -07:00
A. Unique TensorFlower
caec910673
Update GraphDef version to 1152.
...
PiperOrigin-RevId: 452920983
2022-06-04 02:04:46 -07:00
Weiyi Wang
bceca57613
Introduces RootProfiler which can hold a list of child profilers. TFLite will fire profiling profiling events to root profiler and it will be responsible to distribute the events to all sub profilers.
...
PiperOrigin-RevId: 452904408
2022-06-03 22:48:37 -07:00
Anish Tondwalkar
baa73d9884
Remove remnants of DISC from MHLO
...
DISC is developed in https://github.com/alibaba/BladeDISC .
PiperOrigin-RevId: 452903121
2022-06-03 22:31:30 -07:00
Michael Gester
489c35f703
Support non-replicated TPU ops in V1 island outlining
...
Now in V1 island outlining we support compilation markers without associated
replication markers.
Also, added support for single-op TPU clusters (they were wrongly skipped
before) and improved test coverage.
PiperOrigin-RevId: 452894779
2022-06-03 20:54:58 -07:00
Hyojun Kim
1797c08306
Use tuple arguments selectively
...
Introduce use_tuple_args to CompileTFFunctionToHlo() to use
tuple arguments selectively.
PiperOrigin-RevId: 452884038
2022-06-03 19:02:37 -07:00
A. Unique TensorFlower
a67f5d22fa
Fix some ClangTidy warnings in Grappler's dependency optimizer on a boring Friday afternoon.
...
PiperOrigin-RevId: 452882239
2022-06-03 18:43:03 -07:00
Tongfei Guo
86e99ce367
[XLA]Use recv-done's 0th subsharding for it's parameters.
...
PiperOrigin-RevId: 452881825
2022-06-03 18:38:36 -07:00
Wilsin Gosti
5aa60275b5
[tf.data] Increase test size to address flakiness.
...
PiperOrigin-RevId: 452877178
2022-06-03 17:56:00 -07:00
Katherine Wu
04279434a7
Fix TensorFlow checkpoint and trackable imports.
...
PiperOrigin-RevId: 452876975
2022-06-03 17:53:00 -07:00
A. Unique TensorFlower
8310070633
Fix tsan error: data race for sync_protocol_thread_.
...
PiperOrigin-RevId: 452876317
2022-06-03 17:47:04 -07:00
A. Unique TensorFlower
de1caa02b1
Update TFRT dependency to use revision
...
4b3693ed75 .
PiperOrigin-RevId: 452872357
2022-06-03 17:15:58 -07:00
A. Unique TensorFlower
a7c8141cbb
Integrate LLVM at llvm/llvm-project@9a79b1b04c
...
Updates LLVM usage to match
[9a79b1b04c3a](https://github.com/llvm/llvm-project/commit/9a79b1b04c3a )
PiperOrigin-RevId: 452870674
2022-06-03 17:05:41 -07:00
Sagun Bajra
0416617eca
Address redundant H2D/D2D/D2H transfers for int32 eager operations.
...
PiperOrigin-RevId: 452869929
2022-06-03 17:00:39 -07:00
Shiva Shahrokhi
989ea1677c
Removing the TryAcquireTpuLock() from the profiler, because it should not get the tpu lock. The lock should be acquired when we initialize the libtpu.
...
PiperOrigin-RevId: 452867011
2022-06-03 16:40:49 -07:00
Juanli Shen
aeaf0ea6d8
Add back dependency for metrics exporting
...
PiperOrigin-RevId: 452866344
2022-06-03 16:35:38 -07:00
TensorFlower Gardener
e9f76ac8a0
Merge pull request #55943 from trevor-m:tmorris-static-export
...
PiperOrigin-RevId: 452856806
2022-06-03 15:41:15 -07:00
A. Unique TensorFlower
a1d9f04eb4
Internal change
...
PiperOrigin-RevId: 452851454
2022-06-03 15:09:36 -07:00
A. Unique TensorFlower
1b6c4ba5bf
Bug fix for the naming of report protos.
...
PiperOrigin-RevId: 452848974
2022-06-03 14:59:24 -07:00
Terry Heo
66a2d21d09
Fix selective-build script
...
Make "--debug" option work for easier debugging with Docker container.
$ tensorflow/lite/tools/build_aar_with_docker.sh \
--input_models="${PWD}/tensorflow/lite/testdata/softplus_flex.bin" \
--target_archs=x86,x86_64,arm64-v8a,armeabi-v7a --checkpoint=master \
--cache_dir=/tmp/bazel_cache \
--debug
PiperOrigin-RevId: 452839024
2022-06-03 14:08:12 -07:00
A. Unique TensorFlower
7430626d38
Internal change only.
...
PiperOrigin-RevId: 452838334
2022-06-03 14:04:30 -07:00
TensorFlower Gardener
e2dc7d24a2
Merge pull request #56321 from bjacob:dequantize-softmax
...
PiperOrigin-RevId: 452835802
2022-06-03 13:53:03 -07:00
TensorFlower Gardener
acece254d3
Merge pull request #56338 from mseth10:master-arm
...
PiperOrigin-RevId: 452833864
2022-06-03 13:47:18 -07:00
Wilsin Gosti
b1ff6ee500
Tag "no_tsan" should be "notsan".
...
PiperOrigin-RevId: 452833773
2022-06-03 13:40:32 -07:00
Sagun Bajra
c04f65da3e
Address redundant H2D/D2D/D2H transfers for int32 eager operations.
...
PiperOrigin-RevId: 452832110
2022-06-03 13:35:44 -07:00
Andrew Audibert
1557cc81c7
Internal change
...
PiperOrigin-RevId: 452825617
2022-06-03 13:00:34 -07:00
Yang Chen
66c7abe379
#tf-data-service Rename MultiTrainerCache to CrossTrainerCache.
...
PiperOrigin-RevId: 452823806
2022-06-03 12:55:31 -07:00
A. Unique TensorFlower
c7184f9987
Update TFRT dependency to use revision
...
1b3a945efc .
PiperOrigin-RevId: 452823550
2022-06-03 12:50:12 -07:00
A. Unique TensorFlower
9fe6e93f5d
Integrate LLVM at llvm/llvm-project@1a729bce86
...
Updates LLVM usage to match
[1a729bce8617](https://github.com/llvm/llvm-project/commit/1a729bce8617 )
PiperOrigin-RevId: 452820639
2022-06-03 12:39:25 -07:00
A. Unique TensorFlower
d34a9ea705
Bug fix for the naming of report protos.
...
PiperOrigin-RevId: 452817120
2022-06-03 12:21:08 -07:00
Anish Tondwalkar
b1e30334b6
[NFC] Disable identifier tidying
...
Current identifier naming is a mix of google + MLIR style, so this
check should be disabled until we migrate the codebase entirely in
either direction. Until then, let's try and be consistent at least
within a given function or file.
PiperOrigin-RevId: 452815912
2022-06-03 12:15:41 -07:00
Christian Sigg
67a5b3659d
Fix PropagateStaticShapesToKernelPass to handle non-tail memref arguments.
...
PiperOrigin-RevId: 452815818
2022-06-03 12:11:53 -07:00
Justin Lebar
bbd0b4f23d
[XLA:GPU] Constant-fold broadcasts if they're the input to a gemm.
...
In general we don't want to fold broadcasts of scalars; it just makes the
program "less simple" and makes it require more memory.
But for the special case of A@B+C fusion, we do want to fold them. Otherwise
we end up with situations where we run a fusion just to calculate the constant
addend C.
PiperOrigin-RevId: 452813130
2022-06-03 12:00:23 -07:00
Jean-Baptiste Lespiau
bde18b7101
Remove usages of errors::Code::NAME as it's going away.
...
PiperOrigin-RevId: 452804370
2022-06-03 11:24:42 -07:00
A. Unique TensorFlower
c7d82d7a4d
Add device-specific performance fields to StatType
...
PiperOrigin-RevId: 452802576
2022-06-03 11:19:30 -07:00
Ce Zheng
9cd4c76df7
[XLA] Update spmd handling of scatter to not use operand(1) for indices and operand(2) for updates. NFC.
...
PiperOrigin-RevId: 452801873
2022-06-03 11:11:26 -07:00
Yang Chen
6f2dea2ad6
#tf-data-service Disable tsan in multi-process tests.
...
From the test log, all tests have passed, but the tsan
analyzer can't list all threads in time, probably
because they're in separate processes.
According to 63012133ce/tensorflow/python/distribute/coordinator/BUILD (L110) ,
the multi-process runner does not support tsan.
PiperOrigin-RevId: 452798708
2022-06-03 10:53:33 -07:00
TensorFlower Gardener
575330b791
Merge pull request #56338 from mseth10:master-arm
...
PiperOrigin-RevId: 452796362
2022-06-03 10:44:26 -07:00
A. Unique TensorFlower
765b53507f
Bug fix for the naming of report protos.
...
PiperOrigin-RevId: 452791366
2022-06-03 10:23:36 -07:00
George Karpenkov
63012133ce
[TF] [NFC] Replace absl::optional w/ std::optional in SE and compiler/
...
PiperOrigin-RevId: 452789468
2022-06-03 10:14:17 -07:00
A. Unique TensorFlower
5366015a99
Get env object from coordination service agent directly.
...
PiperOrigin-RevId: 452787142
2022-06-03 10:01:42 -07:00
Wilsin Gosti
1a96957f12
[tf.data] Disable test for tsan due to timing out.
...
PiperOrigin-RevId: 452785407
2022-06-03 09:55:27 -07:00
George Karpenkov
0a723ddac4
[TF] [NFC] Replace absl::optional w/ std::optional in SE and compiler/
...
PiperOrigin-RevId: 452781234
2022-06-03 09:39:15 -07:00