Expand assumes to the other unchecked slice ops

Author: saethlin

library/core/src/slice/index.rs

@@ -293,7 +293,10 @@ unsafe impl<T> SliceIndex<[T]> for usize {
             (this: usize = self, len: usize = slice.len()) => this < len
         );
         // SAFETY: see comments for `get_unchecked` above.
-        unsafe { get_mut_noubcheck(slice, self) }
+        unsafe {
+            crate::hint::assert_unchecked(self < slice.len());
+            get_mut_noubcheck(slice, self)
+        }
     }
 
     #[inline]
@@ -346,7 +349,10 @@ unsafe impl<T> SliceIndex<[T]> for ops::IndexRange {
         // cannot be longer than `isize::MAX`. They also guarantee that
         // `self` is in bounds of `slice` so `self` cannot overflow an `isize`,
         // so the call to `add` is safe.
-        unsafe { get_offset_len_noubcheck(slice, self.start(), self.len()) }
+        unsafe {
+            crate::intrinsics::assume(self.end() <= slice.len());
+            get_offset_len_noubcheck(slice, self.start(), self.len())
+        }
     }
 
     #[inline]
@@ -358,7 +364,10 @@ unsafe impl<T> SliceIndex<[T]> for ops::IndexRange {
         );
 
         // SAFETY: see comments for `get_unchecked` above.
-        unsafe { get_offset_len_mut_noubcheck(slice, self.start(), self.len()) }
+        unsafe {
+            crate::intrinsics::assume(self.end() <= slice.len());
+            get_offset_len_mut_noubcheck(slice, self.start(), self.len())
+        }
     }
 
     #[inline]

tests/codegen/issues/issue-116878.rs

@@ -9,3 +9,27 @@ pub unsafe fn unchecked_slice_no_bounds_check(s: &[u8]) -> u8 {
     // CHECK-NOT: panic_bounds_check
     a + s[0]
 }
+
+// CHECK-LABEL: @unchecked_slice_no_bounds_check_mut
+#[no_mangle]
+pub unsafe fn unchecked_slice_no_bounds_check_mut(s: &mut [u8]) -> u8 {
+    let a = *s.get_unchecked_mut(2);
+    // CHECK-NOT: panic_bounds_check
+    a + s[1]
+}
+
+// CHECK-LABEL: @unchecked_slice_no_bounds_check_range
+#[no_mangle]
+pub unsafe fn unchecked_slice_no_bounds_check_range(s: &[u8]) -> u8 {
+    let _a = &s.get_unchecked(..1);
+    // CHECK-NOT: panic_bounds_check
+    s[0]
+}
+
+// CHECK-LABEL: @unchecked_slice_no_bounds_check_range_mut
+#[no_mangle]
+pub unsafe fn unchecked_slice_no_bounds_check_range_mut(s: &mut [u8]) -> u8 {
+    let _a = &mut s.get_unchecked_mut(..2);
+    // CHECK-NOT: panic_bounds_check
+    s[1]
+}