From d209062416b2d6d967ae361f8d60cca0b922fab2 Mon Sep 17 00:00:00 2001 From: Manfei <41607353+ManfeiBai@users.noreply.github.com> Date: Thu, 7 Mar 2024 11:39:50 -0800 Subject: [PATCH] [OpenXLA|TSL] Replace ::tsl::gtl::InlinedVector with ::absl::InlinedVector convolution_helper.h (#6656) --- torch_xla/csrc/BUILD | 1 - torch_xla/csrc/convolution_helper.h | 4 ++-- torch_xla/csrc/ir.h | 4 ++-- 3 files changed, 4 insertions(+), 5 deletions(-) diff --git a/torch_xla/csrc/BUILD b/torch_xla/csrc/BUILD index f5bc1147863..7f78c534af4 100644 --- a/torch_xla/csrc/BUILD +++ b/torch_xla/csrc/BUILD @@ -149,7 +149,6 @@ ptxla_cc_library( "@xla//xla/client/lib:svd", "@xla//xla/stream_executor:dnn", "@tsl//tsl/platform:errors", - "@tsl//tsl/lib/gtl:inlined_vector", "@tsl//tsl/profiler/lib:traceme", "@tsl//tsl/profiler/lib:traceme_encode", "@tsl//tsl/platform:tensor_float_32_utils", diff --git a/torch_xla/csrc/convolution_helper.h b/torch_xla/csrc/convolution_helper.h index cc2f4620473..bd833b7b651 100644 --- a/torch_xla/csrc/convolution_helper.h +++ b/torch_xla/csrc/convolution_helper.h @@ -4,7 +4,7 @@ #include #include -#include "tsl/lib/gtl/inlined_vector.h" +#include "absl/container/inlined_vector.h" #include "xla/client/xla_builder.h" #include "xla/xla_data.pb.h" @@ -97,7 +97,7 @@ struct ConvBackpropSpatialDimension { // Computed dimensions for a backwards convolution. struct ConvBackpropDimensions { // Information about each spatial dimension. - ::tsl::gtl::InlinedVector spatial_dims; + ::absl::InlinedVector spatial_dims; // Batch size. int64_t batch_size; diff --git a/torch_xla/csrc/ir.h b/torch_xla/csrc/ir.h index ac6ae9ed428..0a98c1122ce 100644 --- a/torch_xla/csrc/ir.h +++ b/torch_xla/csrc/ir.h @@ -16,10 +16,10 @@ #include #include +#include "absl/container/inlined_vector.h" #include "absl/hash/hash.h" #include "absl/types/span.h" #include "torch_xla/csrc/runtime/types.h" -#include "tsl/lib/gtl/inlined_vector.h" #include "xla/client/xla_builder.h" namespace torch_xla { @@ -29,7 +29,7 @@ static const uint32_t default_hash_seed = (uint32_t)0x5a2d296e9; class XlaNode; class LoweringContext; -using XlaOpVector = tsl::gtl::InlinedVector; +using XlaOpVector = absl::InlinedVector; template using OutputMap =