tensorflow/third_party
Shaogang Wang 733d71db88
Some checks are pending
ARM CI / build (3.10) (push) Waiting to run
Creates a GitHub Issue when a PR Rolled back via Commit to Master / create-issue-on-pr-rollback (push) Waiting to run
Scorecards supply-chain security / Scorecards analysis (push) Waiting to run
PR #19528: [XLA:GPU] use separte command buffer cmd flag for conditional and loop
Imported from GitHub PR https://github.com/openxla/xla/pull/19528

Observed in saxml workload that sharing the same command buffer cmd type (CONDITIONALS) for WHILE and CONDITIONAL command over kill the lowering opportunities.

Many cases could allow CONDITIONAL instruction to lower into command buffer, while WHILE is not possible.

This PR uses separate command buffer cmd type flag for CONDITIONAL and WHILE instructions when user specifies the type to lowering.
Copybara import of the project:

--
4d62fb512995e2fc6e9077a1b3251a6754c866ca by Shawn Wang <shawnw@nvidia.com>:

use separte command buffer cmd flag for conditional and loop

Merging this change closes #19528

PiperOrigin-RevId: 698729891
2024-11-21 04:47:24 -08:00
..
absl Fix windows-specific issues for pywrap rules 2024-11-09 21:31:30 -08:00
android
benchmark
boringssl
clang_toolchain
compute_library Rollback of PR #76831 2024-10-06 10:32:50 -07:00
cpuinfo
dlpack
ducc
eigen3
farmhash
fft2d
flatbuffers
FP16
gemmlowp
git
gloo
googleapis
gpus Move tsl/platform/{cloud,default,windows} to xla/tsl/platform 2024-11-20 18:15:47 -08:00
grpc
hexagon
highwayhash
hwloc
icu
implib_so
jpeg
kissfft
libprotobuf_mutator
llvm Integrate LLVM at llvm/llvm-project@b03a747fc0 2024-11-19 09:43:58 -08:00
llvm_openmp
mkl
mkl_dnn
mpi
mpitrampoline
nanobind PR #17819: Added use_enabled_free_threading flag to build nanobind with NB_FREE_THREADED=1 2024-10-07 13:57:14 -07:00
nasm
nccl Move tsl/platform/{cloud,default,windows} to xla/tsl/platform 2024-11-20 18:15:47 -08:00
nvtx
opencl_headers
ortools
pasta
protobuf Introduce pywrap bazel rules and migrate Tensorflow to it 2024-10-10 00:12:03 -07:00
py Refactor TF wheel build rule, common python rules and flag names. 2024-11-19 09:55:09 -08:00
pybind11_abseil
pybind11_bazel
pybind11_protobuf Introduce pywrap bazel rules and migrate Tensorflow to it 2024-10-10 00:12:03 -07:00
python_runtime
remote_config
robin_map
ruy
shardy Integrate LLVM at llvm/llvm-project@b03a747fc0 2024-11-19 09:43:58 -08:00
sobol_data
spirv_llvm_translator
stablehlo [StableHLO] Add VhloDialect to the dependent dialects of vhlo-to-version 2024-11-18 12:00:31 -08:00
systemlibs
tensorrt
tf_runtime Update TFRT dependency to use revision 2024-10-10 20:24:04 -07:00
triton Merge sparsity_layout.patch into sparse_dot.patch 2024-11-20 08:17:56 -08:00
uv
vulkan_headers
xla PR #19528: [XLA:GPU] use separte command buffer cmd flag for conditional and loop 2024-11-21 04:47:24 -08:00
__init__.py
arm_neon_2_x86_sse.BUILD
BUILD
codegen.BUILD
coremltools.BUILD
cub.BUILD
cudnn_frontend_header_fix.patch PR #18839: [NVIDIA] Update Cudnn Frontend to 1.8.0 2024-11-04 09:32:58 -08:00
cudnn_frontend.BUILD
curl.BUILD buildifier formatting 2024-11-13 15:34:50 +01:00
cutlass.BUILD
cython.BUILD
gif_fix_image_counter.patch
gif_fix_strtok_r.patch
gif.BUILD
linenoise.BUILD
net_zstd.BUILD
nlohmann_json.BUILD
nsync.patch
nvtx.BUILD
png_fix_rpi.patch
png.BUILD
pprof.BUILD Move tsl/platform/{cloud,default,windows} to xla/tsl/platform 2024-11-20 18:15:47 -08:00
pybind11.BUILD
repo.bzl
riegeli_fix.patch Move tsl/platform/{cloud,default,windows} to xla/tsl/platform 2024-11-20 18:15:47 -08:00
six.BUILD
snappy.BUILD
sqlite.BUILD
tf_toolchains.BUILD
tflite_mobilenet_float.BUILD
tflite_mobilenet_quant.BUILD
tflite_mobilenet.BUILD
tflite_ovic_testdata.BUILD
tflite_smartreply.BUILD
zlib.BUILD