mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fix compilation failure in lapack kernel under msan.

Browse Source 
a_size wasn't defined, but it would only be caught under memory sanitizer.

PiperOrigin-RevId: 480176934
This commit is contained in:
Peter Hawkins 2022-10-10 14:24:17 -07:00 committed by jax authors
parent 2246887f7b
commit a3a2206d49
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jaxlib/lapack_kernels.cc
View File

@ -682,10 +682,9 @@ void RealGees<T>::Kernel(void* out_tuple, void** data, XlaCustomCallStatus*) {
lwork = static_cast<int>(work_query);
T* work = new T[lwork];
size_t a_size = static_cast<int64_t>(n) * static_cast<int64_t>(n) * sizeof(T);
if (a_out != a_in) {
std::memcpy(a_out, a_in,
static_cast<int64_t>(b) * static_cast<int64_t>(n) *
static_cast<int64_t>(n) * sizeof(T));
std::memcpy(a_out, a_in, static_cast<int64_t>(b) * a_size);
}
for (int i = 0; i < b; ++i) {