fortrangoingonforty/armfortas / 91a0dec

Browse files

Test WHERE then-arm unary on second array load vectorizes at O3

Authored by mfwolffe <wolffemf@dukes.jmu.edu>
SHA
91a0dece940dda1854a9f288938cfb04a74af62a
Parents
7a73724
Tree
118b4fd

2 changed files

StatusFile+-
A test_programs/do_loop_vectorize_where_unary_on_b.f90 26 0
A tests/vectorize_where_unary_on_b.rs 71 0
test_programs/do_loop_vectorize_where_unary_on_b.f90added
@@ -0,0 +1,26 @@
1
+! WHERE then-arm with unary on a SECOND array load:
2
+!   where (a > 0) c = -d
3
+!
4
+! a(i) = i - 16; d(i) = real(i, 4); c(i) = 0.
5
+! WHERE a > 0 (lanes 17..32): c ← -d (= -17 .. -32).
6
+! ELSE (default lowering): c stays at 0.
7
+!   c(1)  = 0.0   (mask=false, stays 0)
8
+!   c(16) = 0.0   (a(16)=0, NOT > 0)
9
+!   c(17) = -17.0 (mask=true → -d(17))
10
+!   c(32) = -32.0 (mask=true → -d(32))
11
+!
12
+! CHECK: 0.0000000E0     0.0000000E0    -1.7000000E1    -3.2000000E1
13
+program test_do_loop_vectorize_where_unary_on_b
14
+  implicit none
15
+  integer :: i
16
+  real(4) :: a(32), c(32), d(32)
17
+  do i = 1, 32
18
+    a(i) = real(i - 16, 4)
19
+    d(i) = real(i, 4)
20
+    c(i) = 0.0
21
+  end do
22
+  where (a > 0.0)
23
+    c = -d
24
+  end where
25
+  print *, c(1), c(16), c(17), c(32)
26
+end program test_do_loop_vectorize_where_unary_on_b
tests/vectorize_where_unary_on_b.rsadded
@@ -0,0 +1,71 @@
1
+use std::collections::BTreeSet;
2
+use std::path::PathBuf;
3
+
4
+use armfortas::driver::OptLevel;
5
+use armfortas::testing::{capture_from_path, CaptureRequest, CapturedStage, Stage};
6
+
7
+fn fixture(name: &str) -> PathBuf {
8
+    let path = PathBuf::from("test_programs").join(name);
9
+    assert!(path.exists(), "missing test fixture {}", path.display());
10
+    path
11
+}
12
+
13
+fn capture_text(request: CaptureRequest, stage: Stage) -> String {
14
+    let result = capture_from_path(&request).expect("capture should succeed");
15
+    match result.get(stage) {
16
+        Some(CapturedStage::Text(text)) => text.clone(),
17
+        Some(CapturedStage::Run(_)) => panic!("expected text stage for {}", stage.as_str()),
18
+        None => panic!("missing requested stage {}", stage.as_str()),
19
+    }
20
+}
21
+
22
+fn capture_run_stdout(request: CaptureRequest) -> String {
23
+    let result = capture_from_path(&request).expect("capture should succeed");
24
+    match result.get(Stage::Run) {
25
+        Some(CapturedStage::Run(run)) => run.stdout.clone(),
26
+        _ => panic!("missing run stage"),
27
+    }
28
+}
29
+
30
+#[test]
31
+fn o3_vectorizes_where_with_unary_on_second_array_load() {
32
+    let source = fixture("do_loop_vectorize_where_unary_on_b.f90");
33
+
34
+    let o3_ir = capture_text(
35
+        CaptureRequest {
36
+            input: source.clone(),
37
+            requested: BTreeSet::from([Stage::OptIr]),
38
+            opt_level: OptLevel::O3,
39
+        },
40
+        Stage::OptIr,
41
+    );
42
+    assert_eq!(
43
+        o3_ir.matches("vselect").count(),
44
+        1,
45
+        "expected one vselect:\n{}",
46
+        o3_ir
47
+    );
48
+    assert!(
49
+        o3_ir.contains("vneg"),
50
+        "expected vneg lifted from `c = -d` then arm:\n{}",
51
+        o3_ir
52
+    );
53
+
54
+    let stdout = capture_run_stdout(CaptureRequest {
55
+        input: source,
56
+        requested: BTreeSet::from([Stage::Run]),
57
+        opt_level: OptLevel::O3,
58
+    });
59
+    let trimmed: Vec<&str> = stdout
60
+        .lines()
61
+        .map(|l| l.trim())
62
+        .filter(|l| !l.is_empty())
63
+        .collect();
64
+    assert_eq!(trimmed.len(), 1, "expected one output line:\n{}", stdout);
65
+    assert_eq!(
66
+        trimmed[0],
67
+        "0.0000000E0     0.0000000E0    -1.7000000E1    -3.2000000E1",
68
+        "WHERE-unary-on-b wrong: {:?}",
69
+        trimmed[0]
70
+    );
71
+}