Commit Graph

128542 Commits

Author SHA1 Message Date
Mihai Maruseac
c39f19c52a Use rolling release for Bazel.
Since TF is in the progress of migrating to `cc_shared_library`, we are stress testing Bazel and uncovering a large number of bugs. We are in the state where some TF builds are broken with cascading failures and there is a long wait until a new LTS Bazel release. So, we're switching to rolling releases in an attempt to fix the builds.

However, if this causes more failures than it resolves, we will need to return back to stable releases.

PiperOrigin-RevId: 445259819
2022-04-28 15:14:55 -07:00
Jaesung Chung
23cca349e4 Add requantization case into MHLO uniform_quantize op
This is needed for converting between quantized int tensors.

PiperOrigin-RevId: 445254328
2022-04-28 14:51:43 -07:00
TensorFlower Gardener
b917181c29 Merge pull request #55730 from graphcore:awf/issue-55530
PiperOrigin-RevId: 445252025
2022-04-28 14:41:18 -07:00
Chia-hung Duan
cce6f6484e [TFG] Add ViewOpGraphPass to tfg-transform-opt
PiperOrigin-RevId: 445242875
2022-04-28 14:06:06 -07:00
Laura Pak
e2137a01a5 Update vulkan_headers from ec2db85225ab410bc6829251bef6c578aaed5868 to 32c07c0c5334aea069e518206d75e002ccd85389.
PiperOrigin-RevId: 445232402
2022-04-28 13:26:20 -07:00
Mihai Maruseac
4f483ebaa0 Update curl from curl-7.81.0 to curl-7.82.0.
PiperOrigin-RevId: 445230515
2022-04-28 13:20:48 -07:00
Alan Liu
cff267650c Fix tf.raw_ops.DeleteSessionTensor vulnerability with invalid handle.
Check that `handle` input is actually a scalar before treating it as such.

PiperOrigin-RevId: 445228994
2022-04-28 13:12:18 -07:00
Xinyi Wang
339d5de981 Reduce repetitive logging.
PiperOrigin-RevId: 445226529
2022-04-28 13:06:57 -07:00
Nitin Srinivasan
c97b2b4c9f Add twine and setuptools to build dependencies
PiperOrigin-RevId: 445226135
2022-04-28 13:02:28 -07:00
Alan Liu
48305e8ffe Fix tf.raw_ops.GetSessionTensor vulnerability with invalid handle.
Check that input is actually a scalar before treating it as such.

PiperOrigin-RevId: 445218701
2022-04-28 12:29:12 -07:00
Alan Liu
13d38a07ce Fix tf.raw_ops.UnsortedSegmentJoin vulnerability with invalid num_segments.
Check that input is actually a scalar before treating it as such.

PiperOrigin-RevId: 445206880
2022-04-28 11:51:59 -07:00
A. Unique TensorFlower
fa57990ccc [Preemption Checkpoint] Initialize interfaces.
PiperOrigin-RevId: 445205889
2022-04-28 11:46:56 -07:00
Haoyu Zhang
6b5b833952 Fix data race in coordination service.
PiperOrigin-RevId: 445204177
2022-04-28 11:41:23 -07:00
A. Unique TensorFlower
8eb62dc9cc [grappler][tfg] Enable control-flow sink
Enables control-flow sinking in the TFG grappler optimizer by default.

PiperOrigin-RevId: 445204039
2022-04-28 11:34:01 -07:00
Alan Liu
098e7762d9 Fix tf.raw_ops.QuantizeAndDequantizeV4Grad vulnerability with invalid input_min or input_max.
Check that argument is actually a scalar before treating it as such.

PiperOrigin-RevId: 445198280
2022-04-28 11:18:27 -07:00
Eugene Burmako
e505acc640 Add initial support for int4 to MHLO
This CL adds very simple support for int4 to the MHLO dialect.

It doesn't worry too much about integrations (with frameworks or compilers)
or semantics (e.g. verifiers is a known improvement area for the dialect)
and simply aims to expand the representation power of MHLO to unblock experiments.

Figuring out integrations and semantics is also on the roadmap, but there is
consensus that we should unblock forward progress right away and figure out
the rest in parallel.

PiperOrigin-RevId: 445198171
2022-04-28 11:13:16 -07:00
Yang Chen
e4fea3281c #tf-data-service Data service dispatcher accepts a server builder option.
This allows users to customize the server, e.g.: adding an HTTP server.

PiperOrigin-RevId: 445197293
2022-04-28 11:10:02 -07:00
Alan Liu
290bb05c80 Fix tf.raw_ops.TensorSummaryV2 vulnerability with invalid serialized_summary_metadata.
Check that input is actually a scalar before treating it as such.

PiperOrigin-RevId: 445197183
2022-04-28 11:05:57 -07:00
A. Unique TensorFlower
263ad6ad21 Enables control-flow sinking in the TFG grappler optimizer by default.
PiperOrigin-RevId: 445191200
2022-04-28 10:45:18 -07:00
TensorFlower Gardener
80186fbcde Merge pull request #55770 from Yulv-git:typos5
PiperOrigin-RevId: 445174285
2022-04-28 09:41:45 -07:00
A. Unique TensorFlower
a23bced2b3 Update TFRT dependency to use revision
b3211a9e36.

PiperOrigin-RevId: 445167839
2022-04-28 09:16:05 -07:00
Antonio Sanchez
43be2eb15c Update Eigen to commit:b02c384ef4e8eba7b8bdef16f9dc6f8f4d6a6b2b
CHANGELOG
=========
b02c384ef - Add fused multiply functions for PowerPC - pmsub, pnmadd and pnmsub
3de96caea - Fix HouseholderSequence.h
f845a8bb1 - Fix cwise NaN propagation for scalar input.
a4bb513b9 - Update HouseholderSequence.h
fc1d88841 - Remove AVX512VL dependency in trsm
07db964bd - Restrict new AVX512 trsm to AVX512VL, rename files for consistency.
67eeba6e7 - Avoidable heap allocation in applyHouseholderToTheLeft
3342fc7e4 - Allow all tests to pass with `EIGEN_TEST_NO_EXPLICIT_VECTORIZATION`
efb08e0bb - Revert "Fix ambiguous DiagonalMatrix constructors."
53eec53d2 - Fix Power GEMV order of operations in predux for MMA.
a81bba962 - Fix ambiguous DiagonalMatrix constructors.
f7b31f864 - Revert "Replace call to FixedDimensions() with a singleton instance of"
f3ba220c5 - Remove EIGEN_EMPTY_STRUCT_CTOR
5ed7a86ae - Fix MSVC+CUDA issues.
734ed1efa - Fix ODR issues in lapacke_helpers.
2c45a3846 - Fix some max size expressions.
edc822666 - Fix navbar scroll with toc.
df87d40e3 - constexpr reshape helper
403fa3340 - Performance improvements in GEMM for Power
e1df3636b - More constexpr helpers
64909b82b - static const class members turned into constexpr
2c0ef43b4 - Added Scaling function overload for vector rvalue reference
ba2cb835a - Add back std::remove* aliases - third-party libraries rely on these.
0c859cf35 - Consider inf/nan in scalar test_isApprox.
1ddd3e29c - fixed order of arguments in blas syrk
2c5644280 - Don't include .cpp in lapack.
73b2c13bf - Disable f16c scalar conversions for MSVC.
9bc9992dd - Eliminate trace unused warning.
e22d58e81 - Add is_constant_evaluated, update alignment checks
f0a91838a - Enable Aarch64 CI
b9d2900e8 - added a missing typename and fixed a unused typedef warning
0611f7fff - Add missing explicit reinterprets
cd3c81c3b - Add a NNLS solver to unsupported - issue #655
0699fa06f - Split general_matrix_vector_product interface for Power into two macros - one ColMajor and RowMajor.
19a6a827c - Optimize visitor traversal in case of RowMajor.
f2a3e03e9 - Fix usages of wrong namespace
4451823fb - Fix ODR violation in trsm.
9a14d91a9 - Fix AVX512 builds with MSVC.
7b10795e3 - Change EIGEN_ALTIVEC_ENABLE_MMA_DYNAMIC_DISPATCH and EIGEN_ALTIVEC_DISABLE_MMA flags to be like TensorFlow's...
3ca1228d4 - Work around MSVC compiler bug dropping `const`.
40eb34bc5 - Fix RowMajorBit <-> RowMajor mixup.
c06298346 - Completed a missing parenthesis in tutorial.
9deaa1912 - Work around g++-10 docker issue for geo_orthomethods_4.
e34db1239 - Fix missing pound
591906477 - Fix up PowerPC MMA flags so it builds by default.
518fc321c - AVX512 Optimizations for Triangular Solve
01b5bc48c - Disable schur non-convergence test.
421cbf086 - Replace Eigen type metaprogramming with corresponding std types and make use of alias templates
514f90c9f - Remove workarounds for bad GCC-4 warnings
9ad566148 - Revert "Fix up PowerPC MMA flags so it builds by default."
65eeedf96 - Fix up PowerPC MMA flags so it builds by default.
cb1e8228e - Convert bit calculation to constexpr, avoid casts.
baf9a985e - Fix swap test for size 1 inputs.
788240885 - Temporarily disable aarch64 CI.
2a6be5492 - Fix construct_at compilation breakage on ROCm.
a3b64625e - Remove ComputeCpp-specific code from SYCL Vptr
9296bb4b9 - Fix edge-case in zeta for large inputs.
cd2ba9d03 - Add construct_at, destroy_at wrappers. Use throughout.
dfa517678 - make SparseSolverBase and IterativeSolverBase move constructable
9883108f3 - Remove copy_bool workaround for gcc 4.3
3a9d404d7 - Add support for Apple's Accelerate sparse matrix solvers

PiperOrigin-RevId: 445164768
2022-04-28 09:01:32 -07:00
A. Unique TensorFlower
22fe6f0153 Enables control-flow sinking in the TFG grappler optimizer by default.
PiperOrigin-RevId: 445161805
2022-04-28 08:49:17 -07:00
A. Unique TensorFlower
da55206082 remove outdated legacy source, related to cl/329802973 (which deprecated these traceMes).
PiperOrigin-RevId: 445159833
2022-04-28 08:44:23 -07:00
TensorFlower Gardener
19025b6fea Merge pull request #55767 from Yulv-git:typos2
PiperOrigin-RevId: 445158708
2022-04-28 08:37:20 -07:00
A. Unique TensorFlower
a9b211c225 Remove unneeded TraceMe
PiperOrigin-RevId: 445151516
2022-04-28 08:00:07 -07:00
Dan Moldovan
97ff06ed8b Disable broken test.
PiperOrigin-RevId: 445150110
2022-04-28 07:54:37 -07:00
A. Unique TensorFlower
4d05ed17d1 Update TFRT dependency to use revision
610fd8e6ee.

PiperOrigin-RevId: 445133384
2022-04-28 06:23:44 -07:00
A. Unique TensorFlower
8ff7048424 Disable broken test until fixed.
PiperOrigin-RevId: 445125819
2022-04-28 05:39:33 -07:00
Adrian Kuegel
fb491acc74 Add another regression test for concatenate with zero-sized dimension.
PiperOrigin-RevId: 445115361
2022-04-28 04:34:23 -07:00
A. Unique TensorFlower
3e37a3b7b5 Update TFRT dependency to use revision
b25854f14f.

PiperOrigin-RevId: 445106307
2022-04-28 03:32:13 -07:00
Michael Forster
c306ee3531 Integrate LLVM at llvm/llvm-project@18b9c46370
Updates LLVM usage to match
[18b9c4637099](https://github.com/llvm/llvm-project/commit/18b9c4637099)

PiperOrigin-RevId: 445104770
2022-04-28 03:24:10 -07:00
Chris Jones
4d6eb2716a [XLA] Cleanup transpose_folding_test.
PiperOrigin-RevId: 445102223
2022-04-28 03:13:59 -07:00
A. Unique TensorFlower
37d5ac2d11 compat: Update forward compatibility horizon to 2022-04-28
PiperOrigin-RevId: 445091299
2022-04-28 02:16:54 -07:00
A. Unique TensorFlower
9e216e4c27 Update GraphDef version to 1115.
PiperOrigin-RevId: 445091298
2022-04-28 02:12:06 -07:00
Adrian Kuegel
4af2fda630 Do not create the specialization pipeline if specialization is disabled.
PiperOrigin-RevId: 445083392
2022-04-28 01:20:38 -07:00
Adrian Kuegel
9e3ce05db4 Add a large regression test.
Currently it times out somewhere in the second pipeline
(JitRTCompilationPipeline).

PiperOrigin-RevId: 445079653
2022-04-28 00:54:13 -07:00
A. Unique TensorFlower
6adcab4f6e Update TFRT dependency to use revision
a8f841ebc5.

PiperOrigin-RevId: 445073882
2022-04-28 00:17:41 -07:00
Marat Dukhan
019d401dff [XNNPACK] Support FP16 inference for AVERAGE_POOL_2D & SOFTMAX operators
PiperOrigin-RevId: 445072897
2022-04-28 00:09:08 -07:00
Jaesung Chung
b955483967 Add uniform_quantize and uniform_dequantize operators into MHLO
PiperOrigin-RevId: 445063051
2022-04-27 22:54:06 -07:00
A. Unique TensorFlower
dde12a6904 Update TFRT dependency to use revision
aae8198e0f.

PiperOrigin-RevId: 445053759
2022-04-27 21:46:05 -07:00
Antonio Sanchez
237822b59f Fix tf.compat.v1.placeholder_with_default vulnerability with quantized types.
When iterating through the tensor to extract shape values, an underlying missing kernel
(`StridedSlice` for quantized types) causes an error, which then results in a `nullptr`
being passed to `ParseDimensionValue()`, causing a segfault.

The `nullptr` check allows the missing kernel error to propagate.
Adding the missing kernel registrations allows the shape values
to be extracted successfully.

PiperOrigin-RevId: 445045957
2022-04-27 20:55:48 -07:00
TensorFlower Gardener
527a612a87 Merge pull request #55774 from Yulv-git:typos8
PiperOrigin-RevId: 445043621
2022-04-27 20:39:36 -07:00
Ce Zheng
f29cb6826a [XLA] Make WhileLoopConstantSinking sink broadcasts of constants as well.
PiperOrigin-RevId: 445039088
2022-04-27 20:10:47 -07:00
A. Unique TensorFlower
a1b030da6c Update TFRT dependency to use revision
eaf4cbbb72.

PiperOrigin-RevId: 445031028
2022-04-27 19:26:21 -07:00
Eugene Burmako
207d50d253 Add a static limit to folding binary ops to avoid constant size explosions
This is similar to a fix to a similar problem that we encountered in Dec 2021
when we introduced a static limit for folding scatters. There's a ticket to
follow up on this with a general solution.

PiperOrigin-RevId: 445030081
2022-04-27 19:21:00 -07:00
A. Unique TensorFlower
c059951680 Update TFRT dependency to use revision
43a5658b24.

PiperOrigin-RevId: 445026773
2022-04-27 19:02:04 -07:00
Yuqi Li
c8890049ff install certain version of opencv-python-headless to fix import cv2 failure.
PiperOrigin-RevId: 445023890
2022-04-27 18:45:09 -07:00
Anish Tondwalkar
4c0a855f43 [NFC] Consistent local variable naming conventions in mhlo exporter
camelCase -> snake_case in the few functions that were added later using different conventions

PiperOrigin-RevId: 445023721
2022-04-27 18:41:26 -07:00
Fabien Hertschuh
258142fc0a Explicitly depend on the tensorflow estimator target.
PiperOrigin-RevId: 445005394
2022-04-27 17:03:52 -07:00