diff --git a/ci/miri.sh b/ci/miri.sh index 4e02d32..0f5fa38 100755 --- a/ci/miri.sh +++ b/ci/miri.sh @@ -12,7 +12,8 @@ rustup default "$MIRI_NIGHTLY" rustup component add miri cargo miri setup -# disable isolation for num_cpus::get_physical -MIRIFLAGS=-Zmiri-disable-isolation \ +# Disable isolation for num_cpus::get_physical. +# Also add flags for additional checks. +MIRIFLAGS="-Zmiri-disable-isolation -Zmiri-tag-raw-pointers -Zmiri-check-number-validity" \ MMTEST_FAST_TEST=1 \ cargo miri test "$@" diff --git a/src/gemm.rs b/src/gemm.rs index 1d9e2c5..92fbb4c 100644 --- a/src/gemm.rs +++ b/src/gemm.rs @@ -409,8 +409,8 @@ unsafe fn gemm_packed(nc: usize, kc: usize, mc: usize, // NOTE: For the rust kernels, it performs better to simply // always use the masked kernel function! if K::always_masked() || nr_ < nr || mr_ < mr { - masked_kernel::<_, K>(kc, alpha, &*app.ptr(), &*bpp.ptr(), - beta, &mut *c.ptr(), rsc, csc, + masked_kernel::<_, K>(kc, alpha, app.ptr(), bpp.ptr(), + beta, c.ptr(), rsc, csc, mr_, nr_, mask_buf); continue; } else {