`879adc0`

Test WHERE with array+array body vectorizes at O3

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

SHA: 879adc09943876d05bbda8f5d7fd6ef11eb62679
Parents: f4d73bc
Tree: b960784

2 changed files

Status	File	+	-
A	`test_programs/do_loop_vectorize_where_2arr.f90`	25	0
A	`tests/vectorize_where_2arr.rs`	78	0

test_programs/do_loop_vectorize_where_2arr.f90added

 +! WHERE with an array+array body: `where (a > 0) c = a + b`. Both
 +! `a` and `b` are arrays loaded per iteration; the matcher must
 +! recognize the second array load (in the then_block) as a
 +! vector load and feed both into a VAdd that drives vselect's
 +! true arm.
 +!
 +! a(i) = i - 16 (range -15..16); b(i) = 100; c(i) = i.
 +! Where a > 0 (lanes 17..32): c ← a + b = (i-16) + 100 = i + 84.
 +!   c(1) = 1, c(16) = 16, c(17) = 101, c(32) = 116.
 +!
 +! CHECK: 1.0000000E0     1.6000000E1     1.0100000E2     1.1600000E2
 +program test_do_loop_vectorize_where_2arr
 +  implicit none
 +  integer :: i
 +  real(4) :: a(32), b(32), c(32)
 +  do i = 1, 32
 +    a(i) = real(i - 16, 4)
 +    b(i) = 100.0
 +    c(i) = real(i, 4)
 +  end do
 +  where (a > 0.0)
 +    c = a + b
 +  end where
 +  print *, c(1), c(16), c(17), c(32)
 +end program test_do_loop_vectorize_where_2arr

tests/vectorize_where_2arr.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_with_array_plus_array_body() {
 +    let source = fixture("do_loop_vectorize_where_2arr.f90");
++
 +    let o3_ir = capture_text(
 +        CaptureRequest {
 +            input: source.clone(),
 +            requested: BTreeSet::from([Stage::OptIr]),
 +            opt_level: OptLevel::O3,
 +        },
 +        Stage::OptIr,
 +    );
 +    // The array+array WHERE body must lower to two vloads (a and b)
 +    // feeding a vadd that drives vselect's true arm.
 +    assert_eq!(
 +        o3_ir.matches("vselect").count(),
 +        1,
 +        "expected exactly one vselect:\n{}",
 +        o3_ir
 +    );
 +    assert!(
 +        o3_ir.matches("vload").count() >= 3,
 +        "expected >=3 vload (a, dest, b):\n{}",
 +        o3_ir
 +    );
 +    assert!(
 +        o3_ir.contains("vadd"),
 +        "expected vadd lifted from a+b in WHERE body:\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.0000000E0     1.6000000E1     1.0100000E2     1.1600000E2",
 +        "array+array WHERE wrong: {:?}",
 +        trimmed[0]
 +    );
 +}