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

Add some msan suppressions to the LAPACK symmetric eigendecomposition FFI call.

Browse Source 
This fixes some msan false positives in our CI, since we do not msan-instrument Fortran code.

PiperOrigin-RevId: 669385248
This commit is contained in:
Peter Hawkins 2024-08-30 11:12:15 -07:00 committed by jax authors
parent a55a94de24
commit 1ab3119d43
1 changed files with 5 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
jaxlib/cpu/lapack_kernels.cc
View File

@ -1007,6 +1007,11 @@ ffi::Error EigenvalueDecompositionSymmetric<dtype>::Kernel(
fn(&mode_v, &uplo_v, &x_cols_v, x_out_data, &x_leading_dim_v,
eigenvalues_data, work_data.get(), &work_size_v, iwork_data.get(),
&iwork_size_v, info_data);
ABSL_ANNOTATE_MEMORY_IS_INITIALIZED(x_out_data,
sizeof(*x_out_data) * x_cols * x_cols);
ABSL_ANNOTATE_MEMORY_IS_INITIALIZED(eigenvalues_data,
sizeof(*eigenvalues_data) * x_cols);
ABSL_ANNOTATE_MEMORY_IS_INITIALIZED(info_data, sizeof(lapack_int));
x_out_data += x_out_step;
eigenvalues_data += eigenvalues_step;
++info_data;