Update XLA visibility for Enzyme
PiperOrigin-RevId: 614073447
Showing
- tensorflow/core/tpu/kernels/xla/host_compute_ops.cc 3 additions, 2 deletionstensorflow/core/tpu/kernels/xla/host_compute_ops.cc
- tensorflow/python/data/experimental/kernel_tests/service/distributed_save_load_test.py 26 additions, 0 deletions...mental/kernel_tests/service/distributed_save_load_test.py
- tensorflow/python/data/ops/load_op.py 13 additions, 4 deletionstensorflow/python/data/ops/load_op.py
- third_party/xla/xla/backends/interpreter/executor.h 1 addition, 2 deletionsthird_party/xla/xla/backends/interpreter/executor.h
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding.cc 254 additions, 101 deletions...y/xla/xla/hlo/experimental/auto_sharding/auto_sharding.cc
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding.h 10 additions, 5 deletions...ty/xla/xla/hlo/experimental/auto_sharding/auto_sharding.h
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_cost_graph.h 106 additions, 40 deletions...hlo/experimental/auto_sharding/auto_sharding_cost_graph.h
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_dot_handler.cc 10 additions, 5 deletions...o/experimental/auto_sharding/auto_sharding_dot_handler.cc
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_strategy.cc 18 additions, 9 deletions.../hlo/experimental/auto_sharding/auto_sharding_strategy.cc
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_strategy.h 31 additions, 14 deletions...a/hlo/experimental/auto_sharding/auto_sharding_strategy.h
- third_party/xla/xla/hlo/experimental/auto_sharding/auto_sharding_util.cc 2 additions, 1 deletion.../xla/hlo/experimental/auto_sharding/auto_sharding_util.cc
- third_party/xla/xla/mlir/backends/cpu/transforms/BUILD 1 addition, 1 deletionthird_party/xla/xla/mlir/backends/cpu/transforms/BUILD
- third_party/xla/xla/mlir/memref/BUILD 1 addition, 0 deletionsthird_party/xla/xla/mlir/memref/BUILD
- third_party/xla/xla/mlir/runtime/BUILD 1 addition, 0 deletionsthird_party/xla/xla/mlir/runtime/BUILD
- third_party/xla/xla/python/types.cc 2 additions, 2 deletionsthird_party/xla/xla/python/types.cc
- third_party/xla/xla/service/collective_ops_utils.h 21 additions, 0 deletionsthird_party/xla/xla/service/collective_ops_utils.h
- third_party/xla/xla/service/gpu/nccl_p2p_thunk_common.cc 57 additions, 0 deletionsthird_party/xla/xla/service/gpu/nccl_p2p_thunk_common.cc
- third_party/xla/xla/service/gpu/nccl_p2p_thunk_common.h 22 additions, 0 deletionsthird_party/xla/xla/service/gpu/nccl_p2p_thunk_common.h
- third_party/xla/xla/service/gpu/nccl_recv_thunk.cc 42 additions, 15 deletionsthird_party/xla/xla/service/gpu/nccl_recv_thunk.cc
- third_party/xla/xla/service/gpu/nccl_recv_thunk.h 2 additions, 0 deletionsthird_party/xla/xla/service/gpu/nccl_recv_thunk.h
Please register or sign in to comment