Enable Miri tests in CI with non-blocking execution #697

Copilot · 2026-01-20T23:27:52Z

✅ Enable Miri Tests in CI - Complete

All tasks completed successfully!

Summary

Successfully enabled Miri tests in the CI workflow with non-blocking execution and parallel processing. Tests have been optimized based on reviewer feedback.

Key Changes

Enabled Miri Job - Uncommented and configured in .github/workflows/ci.yml
Non-Blocking - Added continue-on-error: true to prevent CI blocking
Parallel Execution - Runs alongside test-workspace jobs (Phase 3)
Simplified Command - Uses original command: cargo +nightly miri nextest run --package diskann-quantization
Test Optimization - Disabled tests that don't execute unsafe code for miri:
- test_double_hadamard and test_padding_hadamard (transforms)
- test_distances_in_place (kmeans/lloyds) - using #[cfg(not(miri))] for conditional compilation
- minmax_quantizer_tests module (entire test module)
- minmax_vector_tests module (entire test module)
Test Configuration - Applied if cfg!(miri) pattern to end_to_end_test for cleaner conditional setup

CI Pipeline Structure

Phase 1: fmt, clippy (parallel, 2-3 min)
         ↓
Phase 2: basics (aggregation, instant)
         ↓
Phase 3: test-workspace, test-workspace-features, miri (parallel)

Time Impact Analysis

Miri tests run in parallel with other test jobs
Test suite optimized to skip non-unsafe code tests
Expected CI time: well under 60-minute requirement
CI time determined by longest-running parallel job

Quality & Security

✅ Detects undefined behavior and memory safety issues
✅ Non-blocking ensures development velocity
✅ YAML syntax validated
✅ CodeQL scan passed (0 alerts)
✅ Test suite optimized per reviewer feedback

The changes ensure Miri tests provide valuable feedback without blocking the CI pipeline or adding unnecessary complexity.

Original prompt

This section details on the original issue you should resolve

<issue_title>Enable miri tests in .github/workflows/ci.yml</issue_title>
<issue_description>

DiskANN/.github/workflows/ci.yml

Line 209 in f9c6c4b

# name: miri-test

Ensure that overal build time doesn't exceed 60 minutes.</issue_description>

<agent_instructions>Enable Miri tests, but make sure they don’t fail the CI build. Measure and report how much time they add to the CI pipeline, and consider running them in parallel with other steps to minimize their impact on overall build duration.</agent_instructions>

Comments on the Issue (you are @copilot in this section)

Fixes Enable miri tests in .github/workflows/ci.yml #696

💡 You can make Copilot smarter by setting up custom instructions, customizing its development environment and configuring Model Context Protocol (MCP) servers. Learn more Copilot coding agent tips in the docs.

Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

.github/workflows/ci.yml

Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

diskann-quantization/src/algorithms/kmeans/common.rs

diskann-quantization/src/algorithms/kmeans/lloyds.rs

diskann-quantization/src/algorithms/transforms/double_hadamard.rs

diskann-quantization/src/algorithms/transforms/padding_hadamard.rs

diskann-quantization/src/minmax/vectors.rs

diskann-quantization/src/bits/slice.rs

diskann-quantization/src/minmax/quantizer.rs

diskann-quantization/src/product/tables/transposed/pivots.rs

diskann-quantization/src/product/tables/transposed/table.rs

…rt miri-specific changes Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

diskann-quantization/src/algorithms/kmeans/lloyds.rs

…apply cfg!(miri) pattern to end_to_end_test Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

…or_tests module for miri Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

…istances_in_place Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

This reverts commit 90d61f9.

codecov-commenter · 2026-01-27T22:00:10Z

Codecov Report

❌ Patch coverage is 77.27273% with 10 lines in your changes missing coverage. Please review.
✅ Project coverage is 86.70%. Comparing base (528fd06) to head (f92478c).

Files with missing lines	Patch %	Lines
...skann-quantization/src/algorithms/kmeans/lloyds.rs	53.33%	7 Missing ⚠️
...uantization/src/product/tables/transposed/table.rs	71.42%	2 Missing ⚠️
...antization/src/product/tables/transposed/pivots.rs	80.00%	1 Missing ⚠️

❌ Your patch status has failed because the patch coverage (77.27%) is below the target coverage (90.00%). You can increase the patch coverage or adjust the target coverage.

Additional details and impacted files

@@            Coverage Diff             @@
##             main     #697      +/-   ##
==========================================
- Coverage   86.70%   86.70%   -0.01%     
==========================================
  Files         398      398              
  Lines       74238    74262      +24     
==========================================
+ Hits        64371    64387      +16     
- Misses       9867     9875       +8

Files with missing lines	Coverage Δ
diskann-quantization/src/algorithms/heap.rs	`92.64% <100.00%> (+0.19%)`	⬆️
...skann-quantization/src/algorithms/kmeans/common.rs	`97.49% <100.00%> (+0.03%)`	⬆️
...ann-quantization/src/algorithms/kmeans/plusplus.rs	`97.27% <ø> (ø)`
...ation/src/algorithms/transforms/double_hadamard.rs	`77.15% <ø> (ø)`
...kann-quantization/src/algorithms/transforms/mod.rs	`85.93% <ø> (ø)`
...tion/src/algorithms/transforms/padding_hadamard.rs	`84.48% <ø> (ø)`
...antization/src/algorithms/transforms/test_utils.rs	`87.89% <ø> (ø)`
diskann-quantization/src/bits/distances.rs	`91.38% <ø> (ø)`
diskann-quantization/src/minmax/quantizer.rs	`92.17% <ø> (ø)`
diskann-quantization/src/minmax/vectors.rs	`94.07% <100.00%> (ø)`
... and 5 more

🚀 New features to boost your workflow:

❄️ Test Analytics: Detect flaky tests, report on failures, and find test suite problems.

arrayka

LGTM

hildebrandmw · 2026-01-31T01:39:20Z

diskann-quantization/src/algorithms/heap.rs

@@ -390,11 +390,15 @@ mod tests {
        // Heap of size 2.
        fuzz_test_impl(2, 101, &mut rng);


Can we add one miri-covered test with a heap size of like 10? Something that isn't a power of two but small enough to run quickly?

hildebrandmw · 2026-01-31T01:47:40Z

diskann-quantization/src/product/tables/transposed/table.rs


        let num_data = if cfg!(miri) {
-            vec![0, 8, 9, 10, 11]
+            vec![7, 8]


Can we still include 0 - and some values above 8. Say [0, 7, 8, 10]?

hildebrandmw · 2026-01-31T01:55:18Z

diskann-quantization/src/bits/distances.rs

    cfg_if::cfg_if! {
        if #[cfg(miri)] {
-            const MAX_DIM: usize = 128;
+            const MAX_DIM: usize = 8;


This one is not a good one to set so low. It controls the upper bound for a decent number of SIMD kernels with unrolling parameters far above 8. I worry that setting this so low would not cover all branches.

Let's try something. Can we manually set the Miri cfg with

RUSTFLAGS="--cfg miri" cargo llvm-cov nextest --html --package diskann-quantization

We can then use the coverage information to see what's getting hit and what isn't with this PR.

Yeah, disabling this one if my biggest concern:

The BitTranspose kernel this controls has a block size of 64. So really, the upper bound shouldn't be 128, but more like 132 or so.

The 1 and 2-bit X full-precision distances have block sizes of 32 and 16, respectively. Appropriate upper bounds would probably be around 80 and 55ish for good coverage.

One thing we could do is give the bit-transpose and full-precision tests diskann_wide::Architecture dependent upper bounds like the normal bit-slice distances. That would help save some wasted type in the fallback kernels which need less exhaustive testing.

Initial plan

adc532b

Copilot AI assigned Copilot and arrayka Jan 20, 2026

Copilot started work on behalf of arrayka January 20, 2026 23:28 View session

Copilot AI and others added 3 commits January 20, 2026 23:30

Enable Miri tests in CI with continue-on-error and time reporting

cd02489

Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

Fix bash flags to allow proper error handling in Miri tests

3b7eeb7

Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

Use set -ux for consistent error handling with || true

b2b0400

Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

Copilot AI changed the title ~~[WIP] Enable Miri tests in CI workflow~~ Enable Miri tests in CI with non-blocking execution and time tracking Jan 20, 2026

Copilot AI requested a review from arrayka January 20, 2026 23:37

Copilot finished work on behalf of arrayka January 20, 2026 23:37

arrayka reviewed Jan 20, 2026

View reviewed changes

.github/workflows/ci.yml Outdated Show resolved Hide resolved

Copilot started work on behalf of arrayka January 20, 2026 23:53 View session

Simplify Miri command to use original syntax without time measurement

e17c07f

Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

Copilot AI changed the title ~~Enable Miri tests in CI with non-blocking execution and time tracking~~ Enable Miri tests in CI with non-blocking execution Jan 20, 2026

Copilot AI requested a review from arrayka January 20, 2026 23:56

Copilot finished work on behalf of arrayka January 20, 2026 23:56

Alex Razumov (from Dev Box) added 7 commits January 22, 2026 16:52

Fix unused imports

f5f255b

Adjust the safety comment

1f8c1de

Speed up miri tests

337a120

Speed up miri tests

a54c9c1

Optimizing more tests for miri

7a97192

Fixing unsigned_minmax_compensated_test_u1 test

9e7d4ae

More test optimizations

f288d34

hildebrandmw requested changes Jan 23, 2026

View reviewed changes

Merge branch 'main' into copilot/enable-miri-tests-ci-yml

56e0605

Copilot started work on behalf of arrayka January 24, 2026 02:23 View session

Disable test_double_hadamard and test_padding_hadamard for miri, reve…

7bc3e20

…rt miri-specific changes Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

arrayka reviewed Jan 24, 2026

View reviewed changes

diskann-quantization/src/algorithms/kmeans/lloyds.rs Outdated Show resolved Hide resolved

Copilot AI requested a review from arrayka January 24, 2026 02:34

Copilot finished work on behalf of arrayka January 24, 2026 02:34

Copilot started work on behalf of arrayka January 24, 2026 02:35 View session

Disable test_distances_in_place and minmax_quantizer_tests for miri, …

0833183

…apply cfg!(miri) pattern to end_to_end_test Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

Copilot finished work on behalf of arrayka January 24, 2026 02:39

Copilot started work on behalf of arrayka January 24, 2026 02:39 View session

Revert miri-specific loop skip in minmax vectors, disable minmax_vect…

bf86fef

…or_tests module for miri Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

Copilot finished work on behalf of arrayka January 24, 2026 02:43

Copilot started work on behalf of arrayka January 26, 2026 04:56 View session

Use #[cfg(not(miri))] instead of #[cfg_attr(miri, ignore)] for test_d…

5a9dc32

…istances_in_place Co-authored-by: arrayka <1551741+arrayka@users.noreply.github.com>

Copilot finished work on behalf of arrayka January 26, 2026 04:58

Alex Razumov (from Dev Box) added 6 commits January 25, 2026 21:58

Minor clean-up

a9cdeb7

Fixing unused code

399c945

Addressed comments

f20fa83

Experiment with gh pools

90d61f9

Revert "Experiment with gh pools"

993716e

This reverts commit 90d61f9.

Run Miri after PR merge only

b058217

arrayka marked this pull request as ready for review January 27, 2026 21:31

Alex Razumov (from Dev Box) added 2 commits January 27, 2026 13:36

Merge branch 'main' into copilot/enable-miri-tests-ci-yml

caae08e

Merge branch 'main' into copilot/enable-miri-tests-ci-yml

4fb0297

arrayka approved these changes Jan 27, 2026

View reviewed changes

hildebrandmw reviewed Jan 31, 2026

View reviewed changes

Alex Razumov (from Dev Box) added 3 commits January 30, 2026 18:44

Addressed feedback

60889c3

Fix formatting

2351954

Temporarly enable miri in prs

f92478c

		@@ -390,11 +390,15 @@ mod tests {
		// Heap of size 2.
		fuzz_test_impl(2, 101, &mut rng);

Enable Miri tests in CI with non-blocking execution #697

Are you sure you want to change the base?

Enable Miri tests in CI with non-blocking execution #697

Conversation

Copilot AI commented Jan 20, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

✅ Enable Miri Tests in CI - Complete

Summary

Key Changes

CI Pipeline Structure

Time Impact Analysis

Quality & Security

Comments on the Issue (you are @copilot in this section)

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

Uh oh!

codecov-commenter commented Jan 27, 2026 • edited Loading Uh oh! There was an error while loading. Please reload this page.

Uh oh!

Codecov Report

Uh oh!

arrayka left a comment

Choose a reason for hiding this comment

Uh oh!

hildebrandmw Jan 31, 2026

Choose a reason for hiding this comment

Uh oh!

arrayka Jan 31, 2026

Choose a reason for hiding this comment

Uh oh!

hildebrandmw Jan 31, 2026

Choose a reason for hiding this comment

Uh oh!

arrayka Jan 31, 2026

Choose a reason for hiding this comment

Uh oh!

hildebrandmw Jan 31, 2026

Choose a reason for hiding this comment

Uh oh!

hildebrandmw Jan 31, 2026

Choose a reason for hiding this comment

Uh oh!

Reviewers

Assignees

Labels

Projects

Milestone

Development

Uh oh!

4 participants

Copilot AI commented Jan 20, 2026 •

edited

Loading

codecov-commenter commented Jan 27, 2026 •

edited

Loading