rust-lang / rust-lang/rust

Missed optimization: bounds check not elided for `i * s < n` when `0 <= i < n / s`

Open
#134,636 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

A-LLVM C-optimization T-compiler
Dominant language
Rust
Stars
119k
Forks
16.1k
PR merge metrics
PR metrics pending

Description

I tried this code:

#[inline(never)]
pub fn step_sum(arr: &[u32]) -> u32
{
    const STEP_SIZE: usize = 8;

    let mut result = 0;

    for step in 0..(arr.len() / STEP_SIZE) {
        result += arr[step * STEP_SIZE];
    }

    result
}

I expected to see this happen: the index is in the bounds, so the check is elided.

Instead, this happened:

.LBB0_4:
        cmp     rcx, rsi
        jae     .LBB0_6
        add     eax, dword ptr [rdi + 4*rcx]
        add     rcx, 8
        dec     rdx
        jne     .LBB0_4
        ret
.LBB0_6:
        push    rax
        lea     rdx, [rip + .L__unnamed_1]
        mov     rdi, rcx
        call    qword ptr [rip + core::panicking::panic_bounds_check::h300eea3d2ac1c8da@GOTPCREL]

Godbolt

Also, that isn't even the strictest scenario where the compiler should optimize this.
Stricter example:

#[inline(never)]
pub fn step_sum(arr: &[u32]) -> u32
{
    const STEP_SIZE: usize = 8;

    let mut result = 0;

    for step in 0..(arr.len().div_ceil(STEP_SIZE)) {
        let offset = (arr.len() - 1) % STEP_SIZE;
        result += arr[step * STEP_SIZE + offset];
    }

    result
}

Contributor guide

Open the contributing guide

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by reproducing both Rust examples with rustc nightly at opt-level=3 and inspect the generated assembly, as shown in the linked Godbolt case. Done means the compiler can prove the displayed indexed accesses are in bounds and elide their bounds checks; no repository file or test is named.

Written by the indexing model from the issue text.

Assessment

Tech stack
rust
Domain
compilers, performance
Issue type
Bug
Difficulty
4/5
Estimated time
3-5 days
Activity status
Stale
Clarity
Mostly clear
Newbie friendliness
35/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.