diff options
| author |  Benjamin Kramer <kramerb@google.com> | 2018-10-01 19:42:12 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-01 19:46:29 -0700 |
| commit | 991f06fd50fc73285ce415d57f720994c2b2e861 (patch) | |
| tree | 0b87402386aad22ec958f171bfd57f9c7c3e8571 /tensorflow/compiler/xla/service/llvm_ir | |
| parent | beede8525be5386451bf0098992c37416d1864db (diff) | |
[XLA] Migrate from gtl::FlatSet to absl::flat_hash_set
PiperOrigin-RevId: 215324035
Diffstat (limited to 'tensorflow/compiler/xla/service/llvm_ir')
| -rw-r--r-- | tensorflow/compiler/xla/service/llvm_ir/BUILD | 1 | ||||
| -rw-r--r-- | tensorflow/compiler/xla/service/llvm_ir/alias_analysis.cc | 6 | ||||
| -rw-r--r-- | tensorflow/compiler/xla/service/llvm_ir/alias_analysis.h | 1 |
3 files changed, 3 insertions, 5 deletions
diff --git a/tensorflow/compiler/xla/service/llvm_ir/BUILD b/tensorflow/compiler/xla/service/llvm_ir/BUILD index 3934d2e493..6223a34b12 100644 --- a/tensorflow/compiler/xla/service/llvm_ir/BUILD +++ b/tensorflow/compiler/xla/service/llvm_ir/BUILD @@ -39,6 +39,7 @@ cc_library( "//tensorflow/compiler/xla/service:logical_buffer", "//tensorflow/core:lib", "@com_google_absl//absl/container:flat_hash_map", + "@com_google_absl//absl/container:flat_hash_set", "@com_google_absl//absl/strings", "@llvm//:core", ], diff --git a/tensorflow/compiler/xla/service/llvm_ir/alias_analysis.cc b/tensorflow/compiler/xla/service/llvm_ir/alias_analysis.cc index e5370eca56..643ecd0fba 100644 --- a/tensorflow/compiler/xla/service/llvm_ir/alias_analysis.cc +++ b/tensorflow/compiler/xla/service/llvm_ir/alias_analysis.cc @@ -15,7 +15,7 @@ limitations under the License. #include "tensorflow/compiler/xla/service/llvm_ir/alias_analysis.h" -#include <unordered_set> +#include <map> #include "llvm/IR/MDBuilder.h" #include "tensorflow/compiler/xla/service/llvm_ir/llvm_util.h" @@ -164,9 +164,7 @@ llvm::MDNode* AliasAnalysis::GetNoaliasMetadataForBuffer( add_buffers_to_worklist(operand); } - tensorflow::gtl::FlatSet<BufferAllocation::Slice, - BufferAllocation::Slice::Hasher> - buffers; + std::set<BufferAllocation::Slice> buffers; for (const LogicalBuffer* buffer : worklist) { // Skip buffers which cannot be added to the noalias set. if (!assignment.HasAllocation(*buffer) || diff --git a/tensorflow/compiler/xla/service/llvm_ir/alias_analysis.h b/tensorflow/compiler/xla/service/llvm_ir/alias_analysis.h index 88cde2d3d9..2b46b3c396 100644 --- a/tensorflow/compiler/xla/service/llvm_ir/alias_analysis.h +++ b/tensorflow/compiler/xla/service/llvm_ir/alias_analysis.h @@ -23,7 +23,6 @@ limitations under the License. #include "tensorflow/compiler/xla/service/hlo_instruction.h" #include "tensorflow/compiler/xla/service/llvm_ir/ir_array.h" #include "tensorflow/compiler/xla/types.h" -#include "tensorflow/core/lib/gtl/flatset.h" namespace xla { namespace llvm_ir { |