Commit Graph

130325 Commits

Author SHA1 Message Date
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