experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
mm-fix-macos-protobuf-segfault
STT-tensorflow/tensorflow/compiler/xla
History
Stephan Herhut 1a44a1f9dd Drop the implementation of the mlir_gpu plugin prototype in XLA. 
This leaves some passes and the kernel_lowering in place for now. Those are pending removal,, too.

PiperOrigin-RevId: 348621555
Change-Id: I4ff2ed11fb646bf76ceaa780956c6257d89f54ae
2020-12-22 06:05:01 -08:00
..
client [XLA] Avoid reshape to R1 in NormalFloatingPointDistribution 2020-12-15 11:13:11 -08:00
experimental/xla_sharding [XLA:SPMD] Minor fixes and utils for manual sharding 2020-12-01 13:37:36 -08:00
g3doc [TF2XLA] Support asserts as no-ops for tf.function(jit_compile=True) 2020-12-17 18:53:10 -08:00
pjrt Mark Execute methods non-const allow more flexibility, e.g., lazily load programs on first execution. 2020-12-16 12:13:08 -08:00
python Have the C++ path raise a cleaner error. 2020-12-16 07:39:51 -08:00
python_api Python 3 fixes for ConvertLiteralToNumpyArray. 2020-06-26 16:49:42 -07:00
rpc Internal change 2020-10-02 20:33:42 -07:00
service Drop the implementation of the mlir_gpu plugin prototype in XLA. 2020-12-22 06:05:01 -08:00
tests Optimize calls to std::string::find() and friends for a single char. 2020-12-21 08:54:54 -08:00
tools Drop the implementation of the mlir_gpu plugin prototype in XLA. 2020-12-22 06:05:01 -08:00
.clang-format
array2d_test.cc Enable half precision convolution for the CPU and GPU backends. 2018-02-15 10:42:55 -08:00
array2d.h [TF] [XLA] Fix a number of compiler warnings on Mac OS X. 2018-11-19 13:19:32 -08:00
array3d_test.cc Enable half precision convolution for the CPU and GPU backends. 2018-02-15 10:42:55 -08:00
array3d.h [XLA] FP16 Dot support for the CPU and GPU backends. 2018-02-28 12:59:55 -08:00
array4d_test.cc Change headers to directly include absl::Span, and clean up the build 2018-08-30 22:52:05 -07:00
array4d.h Add a default constructor to Array4D 2020-02-07 15:23:01 -08:00
array_test.cc [XLA] Rename all (Mutable)ArraySlice to absl::Span. 2018-08-30 16:07:27 -07:00
array.cc Add specialization of method Array::FillRandom() for complex64 type. 2019-12-13 16:56:06 -08:00
array.h Fix C++14 build. 2020-08-08 13:12:15 -07:00
bit_cast_test.cc Remove absl::bit_casts of non-trivially-copyable Eigen types, and replace them with a new, extensible xla::BitCast call. The latter does everything absl::bit_cast does, but additionally can be specialized to support non-trivial types. 2019-06-14 02:44:19 -07:00
bit_cast.h Move bfloat16 header to tensorflow/core/platform. 2020-08-14 13:44:57 -07:00
BUILD Move llvm dependency from xla/util.h to xla/service/gpu/ir_emission_utils.h 2020-12-14 16:51:06 -08:00
comparison_util.cc
comparison_util.h
cpu_function_runtime.cc [XLA] Move tensorflow/compiler/tf2xla/cpu_function_runtime.h to tensorflow/compiler/xla/cpu_function_runtime.h 2019-04-01 09:27:18 -07:00
cpu_function_runtime.h [XLA:CPU] Reconsider alignment heuristics 2020-04-24 02:51:55 -07:00
debug_options_flags.cc [XLA:GPU] Add --xla_gpu_deterministic_ops to force run-to-run determinism. 2020-12-15 12:51:36 -08:00
debug_options_flags.h
debug_options_parsers_test.cc
debug_options_parsers.h
device_util.h
error_spec.h [XLA] Add ErrorSpec::fewer_infs_ok knob. 2019-01-24 16:53:05 -08:00
executable_run_options.cc [NFC] Moved GlobalDeviceId into its own file as it is also used by the CPU runtime. 2020-11-24 04:21:38 -08:00
executable_run_options.h [NFC] Moved GlobalDeviceId into its own file as it is also used by the CPU runtime. 2020-11-24 04:21:38 -08:00
execution_options_util.cc [XLA] Move xla::legacy_flags into xla namespace. 2018-11-08 12:20:48 -08:00
execution_options_util.h minor spelling tweaks 2019-12-09 18:21:12 +09:00
index_util_test.cc Replace std::vector<std::initializer_list> with std::vector<std::vector>. 2018-11-08 03:19:36 -08:00
index_util.cc xla directory resolutions 2020-07-26 22:14:33 +00:00
index_util.h Replace Shape with a C++ class in XLA. 2018-11-28 16:08:45 -08:00
iterator_util_test.cc
iterator_util.h
layout_test.cc [XLA] Remove unsupported sparse layout 2020-01-02 18:04:52 -08:00
layout_util_test.cc [XLA] Remove unsupported sparse layout 2020-01-02 18:04:52 -08:00
layout_util.cc Merge pull request from tg-at-google:wsign-compare-semi-final-xla 2020-07-29 19:10:26 -07:00
layout_util.h [XLA] Remove unsupported sparse layout 2020-01-02 18:04:52 -08:00
layout.cc This CL optimizes C++11 range-based for loops where the variable is copied in each iteration but it would suffice to obtain it by const reference. This is only applied to loop variables of types that are expensive to copy which means they are not trivially copyable or have a non-trivial copy constructor or destructor. 2020-04-06 17:55:23 -07:00
layout.h [XLA] Remove unsupported sparse layout 2020-01-02 18:04:52 -08:00
literal_comparison.cc fix typos in compiler dir 2020-10-29 01:08:30 +03:00
literal_comparison.h Near comparison now works for non-fp types. Callback uses ShapeIndex parameter. 2019-06-05 16:01:58 -07:00
literal_test.cc [Resubmit] Dynamic literal support 2020-07-17 18:19:30 -07:00
literal_util.cc xla directory resolutions 2020-07-26 22:14:33 +00:00
literal_util.h Introduce a few new literal conversion functions for various floating point types 2020-01-27 07:47:30 -08:00
literal.cc Add several fixes for s390x architecture and other Big-Endian systems 2020-10-28 11:14:47 -04:00
literal.h [Resubmit] Dynamic literal support 2020-07-17 18:19:30 -07:00
map_util.h [XLA] Switch to absl::StrFormat. 2018-08-27 14:55:29 -07:00
metric_table_report.cc
metric_table_report.h
overflow_util.h [XLA] Simplify ({0, ..., n-1} + n) % n to {0, ..., n-1}. 2019-07-07 22:32:37 -07:00
packed_literal_reader.cc Replace Layout and Tile protos with C++ classes in XLA. 2018-12-11 21:00:21 -08:00
packed_literal_reader.h Global de-std::unique_ptr cleanup for xla::Literal. 2018-09-10 12:38:19 -07:00
parse_flags_from_env_test.cc Handle the cases where stdout may have CRLF suffix on windows in xla parse_flags_from_env_test 2020-03-12 14:33:13 -07:00
parse_flags_from_env.cc [XLA] Use LOG(QFATAL) rather than LOG(FATAL) in XLA_FLAGS argument parsing. 2019-11-27 10:27:55 -08:00
parse_flags_from_env.h minor spelling tweaks 2019-12-09 18:21:12 +09:00
primitive_util_test.cc
primitive_util.cc
primitive_util.h
protobuf_util.cc
protobuf_util.h
README.md
refcounting_hash_map_test.cc Added RefCountingHashMap::GetOrTryCreateIfAbsent to allow for factory methods that may fail. 2020-11-26 06:10:04 -08:00
refcounting_hash_map.h [XLA] Fix race condition in RefcountingHashMap that could lead to a use-after-free segfault. 2020-12-15 12:11:48 -08:00
reference_util_test.cc Qualify calls to some functions from <cmath>. 2019-03-19 16:29:16 -07:00
reference_util.cc [XLA] Add an optional preferred_element_type argument to Dot/Conv builder methods to enable generating HLOs that has wider accumulation type than default shape inference result. 2020-11-20 13:16:56 -08:00
reference_util.h [XLA] Do extensive testing of int32 matmuls 2019-09-16 08:52:24 -07:00
service_interface.h [XLA] Split ExecuteGraph into Compile and Execute. 2018-11-07 19:20:01 -08:00
shape_layout.cc
shape_layout.h
shape_test.cc
shape_tree_test.cc Internal tests cleanup. 2020-10-28 07:40:36 -07:00
shape_tree.h Split ShapeTreeIterator into two classes: ShapeTreeIterator and ShapeTreeLeafIterator 2020-09-15 14:29:58 -07:00
shape_util_test.cc Speed up Shape creation by avoiding unnecessary validations. 2020-11-22 20:40:07 -08:00
shape_util.cc [XLA:GPU] Migrate convolution thunk emission to MLIR 2020-12-14 20:55:56 -08:00
shape_util.h [XLA:GPU] Migrate convolution thunk emission to MLIR 2020-12-14 20:55:56 -08:00
shape.cc
shape.h [libtpu] Reimplement XLA_Shape in C API for performance. 2020-10-19 14:17:22 -07:00
status_macros_test.cc minor spelling tweaks 2019-12-09 18:21:12 +09:00
status_macros.cc [TF:XLA] Add error message about the non-strict XLA jit. 2019-02-07 02:40:43 -08:00
status_macros.h Multiple blocks of code in StreamExecutor can be simplified using the 2019-05-22 12:00:46 -07:00
status.h [XLA] s/tensorflow::Status/Status/. 2018-05-11 16:07:35 -07:00
statusor.h
test_helpers.h Remove unnecessary forward declaration of util::Status. 2019-04-19 16:58:17 -07:00
test.h [XLA] Fix build break in OSS by adding missing dependencies. 2019-08-12 14:44:12 -07:00
text_literal_reader_test.cc Global de-std::unique_ptr cleanup for xla::Literal. 2018-09-10 12:38:19 -07:00
text_literal_reader.cc Change Shape parsing from regexp matcher to parser. 2018-12-14 08:37:21 -08:00
text_literal_reader.h Global de-std::unique_ptr cleanup for xla::Literal. 2018-09-10 12:38:19 -07:00
text_literal_writer_test.cc Use Env::LocalTempFilename for a temp filename. 2020-02-19 17:30:20 -08:00
text_literal_writer.cc
text_literal_writer.h
types.h Merge pull request from ROCmSoftwarePlatform:google-upstream-pr-merge_to_gpu_executable 2019-07-18 16:39:15 -07:00
union_find.h XLA Parallel reduce. 2020-08-27 15:21:33 -07:00
util_test.cc [XLA] Fix a bug in SplitF64ToF32 2020-11-23 11:27:37 -08:00
util.cc [XLA] Fix a bug in SplitF64ToF32 2020-11-23 11:27:37 -08:00
util.h Move llvm dependency from xla/util.h to xla/service/gpu/ir_emission_utils.h 2020-12-14 16:51:06 -08:00
window_util_test.cc
window_util.cc
window_util.h
xla_data.proto [XLA] Add creation_pass_id and dummy op_names to OpMetadata. 2020-12-11 17:48:13 -08:00
xla.bzl
xla.proto [XLA:GPU] Add --xla_gpu_deterministic_ops to force run-to-run determinism. 2020-12-15 12:51:36 -08:00

README.md

XLA (Accelerated Linear Algebra) is a domain-specific compiler for linear algebra that optimizes TensorFlow computations. See the documentation.