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

Merge pull request #24778 from cainmagi:fix-pr-23852

Browse Source 
PiperOrigin-RevId: 694565904
This commit is contained in:
jax authors 2024-11-08 11:00:19 -08:00
commit 4d1a1264f0
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
jaxlib/gpu/solver_kernels_ffi.cc
View File

@ -915,7 +915,7 @@ ffi::Error GesvdjImpl(int64_t batch, int64_t rows, int64_t cols,
auto a_data = static_cast<T*>(a.untyped_data());
auto out_data = static_cast<T*>(out->untyped_data());
auto s_data = static_cast<solver::RealType<T>::value*>(s->untyped_data());
auto s_data = static_cast<typename solver::RealType<T>::value*>(s->untyped_data());
auto u_data = static_cast<T*>(u->untyped_data());
auto v_data = static_cast<T*>(v->untyped_data());
auto info_data = info->typed_data();