`71d02aa`

Test WHERE/ELSEWHERE array-load else vectorizes at O3

Authored by mfwolffe <wolffemf@dukes.jmu.edu> 1 week ago

SHA: 71d02aab73790841e70ccfa44a056186c4c834cb
Parents: 9173691
Tree: de3cb38

2 changed files

Status	File	+	-
A	`test_programs/do_loop_vectorize_where_elsewhere_arr.f90`	30	0
A	`tests/vectorize_where_elsewhere_arr.rs`	74	0

test_programs/do_loop_vectorize_where_elsewhere_arr.f90added

++! WHERE / ELSEWHERE both arms loading from arrays:
++!   where (a > 0) c = a
++!   elsewhere    c = b
++! lowers to vselect(mask, vload_a, vload_b) → vstore_c.
++!
++! a(i) = i - 16 (range -15..16); b(i) = 1000.0 + i; c(i) = 0.
++! WHERE a > 0 (lanes 17..32): c ← a (= 1..16).
++! ELSEWHERE                    : c ← b (= 1001..1016 for lanes 1..16).
++!   c(1)  = 1001 (mask=false → b(1))
++!   c(16) = 1016 (a(16)=0, NOT > 0 → b(16))
++!   c(17) = 1.0  (mask=true → a = 1)
++!   c(32) = 16.0 (mask=true → a = 16)
++!
++! CHECK: 1.0010000E3     1.0160000E3     1.0000000E0     1.6000000E1
++program test_do_loop_vectorize_where_elsewhere_arr
++  implicit none
++  integer :: i
++  real(4) :: a(32), b(32), c(32)
++  do i = 1, 32
++    a(i) = real(i - 16, 4)
++    b(i) = real(1000 + i, 4)
++    c(i) = 0.0
++  end do
++  where (a > 0.0)
++    c = a
++  elsewhere
++    c = b
++  end where
++  print *, c(1), c(16), c(17), c(32)
++end program test_do_loop_vectorize_where_elsewhere_arr

tests/vectorize_where_elsewhere_arr.rsadded

++use std::collections::BTreeSet;
++use std::path::PathBuf;
++
++use armfortas::driver::OptLevel;
++use armfortas::testing::{capture_from_path, CaptureRequest, CapturedStage, Stage};
++
++fn fixture(name: &str) -> PathBuf {
++    let path = PathBuf::from("test_programs").join(name);
++    assert!(path.exists(), "missing test fixture {}", path.display());
++    path
++}
++
++fn capture_text(request: CaptureRequest, stage: Stage) -> String {
++    let result = capture_from_path(&request).expect("capture should succeed");
++    match result.get(stage) {
++        Some(CapturedStage::Text(text)) => text.clone(),
++        Some(CapturedStage::Run(_)) => panic!("expected text stage for {}", stage.as_str()),
++        None => panic!("missing requested stage {}", stage.as_str()),
++    }
++}
++
++fn capture_run_stdout(request: CaptureRequest) -> String {
++    let result = capture_from_path(&request).expect("capture should succeed");
++    match result.get(Stage::Run) {
++        Some(CapturedStage::Run(run)) => run.stdout.clone(),
++        _ => panic!("missing run stage"),
++    }
++}
++
++#[test]
++fn o3_vectorizes_where_elsewhere_with_array_load_else() {
++    let source = fixture("do_loop_vectorize_where_elsewhere_arr.f90");
++
++    let o3_ir = capture_text(
++        CaptureRequest {
++            input: source.clone(),
++            requested: BTreeSet::from([Stage::OptIr]),
++            opt_level: OptLevel::O3,
++        },
++        Stage::OptIr,
++    );
++    // The two-arm `c = a; elsewhere; c = d` should produce a vselect
++    // whose true arm is vload(a) and false arm is vload(d) — at
++    // least 2 vloads in the WHERE body block.
++    assert_eq!(
++        o3_ir.matches("vselect").count(),
++        1,
++        "expected exactly one vselect:\n{}",
++        o3_ir
++    );
++    assert!(
++        o3_ir.matches("vload").count() >= 2,
++        "expected ≥2 vloads (a and d):\n{}",
++        o3_ir
++    );
++
++    let stdout = capture_run_stdout(CaptureRequest {
++        input: source,
++        requested: BTreeSet::from([Stage::Run]),
++        opt_level: OptLevel::O3,
++    });
++    let trimmed: Vec<&str> = stdout
++        .lines()
++        .map(|l| l.trim())
++        .filter(|l| !l.is_empty())
++        .collect();
++    assert_eq!(trimmed.len(), 1, "expected one output line:\n{}", stdout);
++    assert_eq!(
++        trimmed[0],
++        "1.0010000E3     1.0160000E3     1.0000000E0     1.6000000E1",
++        "WHERE/ELSEWHERE-arr wrong: {:?}",
++        trimmed[0]
++    );
++}