`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]
 +    );
 +}