From aa7d2917675e882ced4e13b986a9d42a4e3f637b Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Tue, 24 May 2022 14:20:53 -0700 Subject: [PATCH] Replace references to absl::string_view with std::string_view. PiperOrigin-RevId: 450768333 --- jaxlib/cuda/cuda_lu_pivot_kernels.cc | 4 +++- jaxlib/cuda/cuda_prng_kernels.cc | 4 +++- jaxlib/rocm/hip_lu_pivot_kernels.cc | 4 +++- jaxlib/rocm/hip_prng_kernels.cc | 4 +++- 4 files changed, 12 insertions(+), 4 deletions(-) diff --git a/jaxlib/cuda/cuda_lu_pivot_kernels.cc b/jaxlib/cuda/cuda_lu_pivot_kernels.cc index 504a6bd84..b13ba14af 100644 --- a/jaxlib/cuda/cuda_lu_pivot_kernels.cc +++ b/jaxlib/cuda/cuda_lu_pivot_kernels.cc @@ -15,6 +15,8 @@ limitations under the License. #include "jaxlib/cuda/cuda_lu_pivot_kernels.h" +#include + #include "jaxlib/cuda/cuda_gpu_kernel_helpers.h" #include "jaxlib/kernel_helpers.h" #include "tensorflow/compiler/xla/service/custom_call_status.h" @@ -40,7 +42,7 @@ void CudaLuPivotsToPermutation(cudaStream_t stream, void** buffers, XlaCustomCallStatus* status) { auto s = CudaLuPivotsToPermutation_(stream, buffers, opaque, opaque_len); if (!s.ok()) { - absl::string_view message = s.message(); + std::string_view message = s.message(); XlaCustomCallStatusSetFailure(status, message.data(), message.length()); } } diff --git a/jaxlib/cuda/cuda_prng_kernels.cc b/jaxlib/cuda/cuda_prng_kernels.cc index 99c777c3e..ead6be35a 100644 --- a/jaxlib/cuda/cuda_prng_kernels.cc +++ b/jaxlib/cuda/cuda_prng_kernels.cc @@ -15,6 +15,8 @@ limitations under the License. #include "jaxlib/cuda/cuda_prng_kernels.h" +#include + #include "jaxlib/cuda/cuda_gpu_kernel_helpers.h" #include "jaxlib/kernel_helpers.h" #include "tensorflow/compiler/xla/service/custom_call_status.h" @@ -37,7 +39,7 @@ void CudaThreeFry2x32(cudaStream_t stream, void** buffers, const char* opaque, size_t opaque_len, XlaCustomCallStatus* status) { auto s = CudaThreeFry2x32_(stream, buffers, opaque, opaque_len); if (!s.ok()) { - absl::string_view message = s.message(); + std::string_view message = s.message(); XlaCustomCallStatusSetFailure(status, message.data(), message.length()); } } diff --git a/jaxlib/rocm/hip_lu_pivot_kernels.cc b/jaxlib/rocm/hip_lu_pivot_kernels.cc index fc9d819f8..a078dfff3 100644 --- a/jaxlib/rocm/hip_lu_pivot_kernels.cc +++ b/jaxlib/rocm/hip_lu_pivot_kernels.cc @@ -15,6 +15,8 @@ limitations under the License. #include "jaxlib/rocm/hip_lu_pivot_kernels.h" +#include + #include "jaxlib/rocm/hip_gpu_kernel_helpers.h" #include "jaxlib/kernel_helpers.h" #include "tensorflow/compiler/xla/service/custom_call_status.h" @@ -40,7 +42,7 @@ void HipLuPivotsToPermutation(hipStream_t stream, void** buffers, XlaCustomCallStatus* status) { auto s = HipLuPivotsToPermutation_(stream, buffers, opaque, opaque_len); if (!s.ok()) { - absl::string_view message = s.message(); + std::string_view message = s.message(); XlaCustomCallStatusSetFailure(status, message.data(), message.length()); } } diff --git a/jaxlib/rocm/hip_prng_kernels.cc b/jaxlib/rocm/hip_prng_kernels.cc index 60b898326..05992467d 100644 --- a/jaxlib/rocm/hip_prng_kernels.cc +++ b/jaxlib/rocm/hip_prng_kernels.cc @@ -15,6 +15,8 @@ limitations under the License. #include "jaxlib/rocm/hip_prng_kernels.h" +#include + #include "jaxlib/rocm/hip_gpu_kernel_helpers.h" #include "jaxlib/kernel_helpers.h" #include "tensorflow/compiler/xla/service/custom_call_status.h" @@ -37,7 +39,7 @@ void HipThreeFry2x32(hipStream_t stream, void** buffers, const char* opaque, size_t opaque_len, XlaCustomCallStatus* status) { auto s = HipThreeFry2x32_(stream, buffers, opaque, opaque_len); if (!s.ok()) { - absl::string_view message = s.message(); + std::string_view message = s.message(); XlaCustomCallStatusSetFailure(status, message.data(), message.length()); } }