that the corresponding FlatBuffer C API type can also be used instead of
the FlatBuffer C++ API type for the TFLiteSettings FlatBuffer parameter.

PiperOrigin-RevId: 565008668

From the original code at commit 03d304b, we are only supposed to scale the reduction
tile size of dimX by the unroll factor for column reductions, so the check when creating
`ReductionCodegenInfo` is only valid for column reductions.

PiperOrigin-RevId: 564991695

Add a comparison operator for fragments, improve encapsulation, fix comments.

PiperOrigin-RevId: 564990063

LaunchDimensions are being decoupled from codegen. Shared memory requirements
are only known after codegen.

Currently, this feature is only used for Triton fusions. All other
fusions that use shared memory allocate it within the kernel.

PiperOrigin-RevId: 564981685

PiperOrigin-RevId: 564978233

PiperOrigin-RevId: 564978209

The upcoming CUDA-12 upgrades requires TensorRT 8.6 and this version
has a new set of headers which requires an update to the bazel configure script.

I also had to change `find_cuda_config.py` because previously it has only
been reporting the major version of TensorRT back to the configure script. But
we will also need the minor version to distinguish between TensorRT 8.5 and below,
and TensorRT 8.6+.

PiperOrigin-RevId: 564927510

PiperOrigin-RevId: 564914384

PiperOrigin-RevId: 564909053

GCC is missing some intrinsics, expand them by hand.

PiperOrigin-RevId: 564899448

PiperOrigin-RevId: 564893903

Create metrics:
1) '/pjrt/compiler/is_compiling_computation' to record if pjrt compiler is compiling computations.
2) '/pjrt/compiler/is_compiling_module' to record if pjrt compiler is compiling modules.

PiperOrigin-RevId: 564891869

PiperOrigin-RevId: 564886962

PiperOrigin-RevId: 564881319

This is for internal error logging only.

PiperOrigin-RevId: 564878979

PiperOrigin-RevId: 564871149

PiperOrigin-RevId: 564864652

For an 8x8 uint32_t transpose, we had:
- 4x `vinsertf128 ymm, ymm, xmm`
- 4x `vperm2f128`.

These are very expensive instructions because they cross the 128-bit lane boundary.

Now, we have `8x vinsertf128` but crucially, the inserted operand now comes from memory. This is important because modern X86 HW can easily broadcast on load which means that `vinsertf128` turns into a blend instead of a shuffle.

We use the same trick for handling matrices which are smaller than the vector width to accelerate the transpose. We still require a cross lane step but we cut down all the other shuffles in half compared to SSE2.

While we are here, don't claim to support kernels which don't exist. This makes the transpose system choose unoptimized implementations.

PiperOrigin-RevId: 564860657

This check can be removed since tf2xla can run ops with non-const input even if CompileTimeConstant attribute is set with the help of valueinference.

PiperOrigin-RevId: 564851049

PiperOrigin-RevId: 564845580

Updates LLVM usage to match
[c1796be93fe5](https://github.com/llvm/llvm-project/commit/c1796be93fe5)

PiperOrigin-RevId: 564842806

PiperOrigin-RevId: 564840225

PiperOrigin-RevId: 564825677

Without this additional bazelrc file TSL will be built with the system compiler which happens to be GCC.

I also had to disable a warning that was raised by clang. It puzzles me a bit that this is not needed for the Tensorflow build which definitely uses Clang.

PiperOrigin-RevId: 564822820

I've seen this file take over 5 minutes to build. Shard it by type.

PiperOrigin-RevId: 564820851

PiperOrigin-RevId: 564813085

This is fixing a UB issue which occurs with newer version of Clang (17+).
The fix is also upstreamed through https://github.com/NVIDIA/nccl/pull/916.

In addition I'm changing the handling of `enqueue.cc` which needs to be compiled
in cuda mode under clang. The previous solution with just passing in the `-x cuda` option fails with CUDA 12+.

I'm also correcting the version number that we set in the patch - not sure if this version is reported in some logs, but if it is, it should be correct.

PiperOrigin-RevId: 564811002

Add CopyToMemorySpace to the PjRtBuffer API. This CL does not implement any instance of the method, but adds the ability to do so in followup CLs.

PiperOrigin-RevId: 564807735

PiperOrigin-RevId: 564807274

PiperOrigin-RevId: 564804399

Remove `uint64` -> `double` -> `uint64` conversions from `NowMicros()` and their users.

PiperOrigin-RevId: 564802323

This broke some tests outside of tflite and darwinn. Temporarily adding back these patterns until the test issues are resolved.

PiperOrigin-RevId: 564800351

PiperOrigin-RevId: 564799226

At the moment, StableHLO does not have the spec and the interpreter for `broadcast`. `broadcast` is a special case of `broadcast_in_dim`, and this pass makes `broadcast -> broadcast_in_dim` conversion.

PiperOrigin-RevId: 564796578

PiperOrigin-RevId: 564789520

PiperOrigin-RevId: 564767887

This was causing a missing symbol error for certain build targets.

PiperOrigin-RevId: 564766163

Imported from GitHub PR https://github.com/openxla/xla/pull/5512

Some build tests were failing due to unresolved symbols so add in their dependencies

This is a redo of https://github.com/tensorflow/tensorflow/pull/61489 which was not merged in time before the code was moved.
Copybara import of the project:

--
a1f80543ccb7b8cccad2e4ce4f33f18b68c3baf1 by Andrew Goodbody <andrew.goodbody@linaro.org>:

Fix link errors in build tests

Some build tests were failing due to unresolved symbols
so add in their dependencies

Merging this change closes #5512

PiperOrigin-RevId: 564758391

PiperOrigin-RevId: 564749546

PiperOrigin-RevId: 564748979