mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-15 19:36:06 +00:00
Code Issues Packages Projects Releases Wiki Activity

Replace references to absl::string_view with std::string_view.

Browse Source 
PiperOrigin-RevId: 450768333
This commit is contained in:
Peter Hawkins 2022-05-24 14:20:53 -07:00 committed by jax authors
parent d849f49519
commit aa7d291767
4 changed files with 12 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
jaxlib/cuda/cuda_lu_pivot_kernels.cc
View File

@ -15,6 +15,8 @@ limitations under the License.
#include "jaxlib/cuda/cuda_lu_pivot_kernels.h"
#include <string_view>
#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());
}
}

4
jaxlib/cuda/cuda_prng_kernels.cc
View File

@ -15,6 +15,8 @@ limitations under the License.
#include "jaxlib/cuda/cuda_prng_kernels.h"
#include <string_view>
#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());
}
}

4
jaxlib/rocm/hip_lu_pivot_kernels.cc
View File

@ -15,6 +15,8 @@ limitations under the License.
#include "jaxlib/rocm/hip_lu_pivot_kernels.h"
#include <string_view>
#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());
}
}

4
jaxlib/rocm/hip_prng_kernels.cc
View File

@ -15,6 +15,8 @@ limitations under the License.
#include "jaxlib/rocm/hip_prng_kernels.h"
#include <string_view>
#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());
}
}