Skip to content

Commit

Permalink
Simplify scan0 test.
Browse files Browse the repository at this point in the history
  • Loading branch information
nhey committed Aug 9, 2023
1 parent 7f4142b commit 22954ec
Showing 1 changed file with 46 additions and 43 deletions.
89 changes: 46 additions & 43 deletions tests/dependence-analysis/scan0.fut
Original file line number Diff line number Diff line change
@@ -1,58 +1,61 @@
-- ==
-- structure { Screma/BinOp 1 }
-- structure { BinOp 1 }

def main (X: []i32) (Y: []i32) (n: i64) =
let free_ne = i32.i64 n * 2
let A = (zip X Y)
def main (A: [](i32,i32)) (n: i64) =
let r =
loop A for _i < n do
scan (\(a,b) (x,y) -> (x+a,y+b)) (free_ne,0) A
loop A for i < n do
scan (\(a,b) (x,y) -> if i == 0 then (a,b) else (a+x,b+y)) (0,0) A
in map (.0) r

-- Before changes:
-- entry("main",
-- {X: []i32,
-- Y: []i32,
-- {A: opaque "arr_(i32, i32)_1d",
-- n: i64},
-- {*[]i32})
-- entry_main (d₁_5271 : i64,
-- X_5272 : [d₁_5271]i32,
-- Y_5273 : [d₁_5271]i32,
-- n_5274 : i64)
-- : {*[d₁_5271]i32} = {
-- let {i64_res_5332 : i32} =
-- sext i64 n_5274 to i32
-- let {free_ne_5333 : i32} =
-- mul32(2i32, i64_res_5332)
-- let {r_5341 : [d₁_5271]i32,
-- r_5342 : [d₁_5271]i32} =
-- loop {A_5344 : [d₁_5271]i32,
-- A_5345 : [d₁_5271]i32} = {X_5272, Y_5273}
-- for _i_5343:i64 < n_5274 do {
-- let {defunc_0_scan_res_5359 : [d₁_5271]i32,
-- defunc_0_scan_res_5360 : [d₁_5271]i32} =
-- scanomap(d₁_5271,
-- {A_5344, A_5345},
-- {\ {eta_p_5348 : i32,
-- eta_p_5349 : i32,
-- eta_p_5350 : i32,
-- eta_p_5351 : i32}
-- entry_main (d₀_5256 : i64,
-- A_5257 : [d₀_5256]i32,
-- A_5258 : [d₀_5256]i32,
-- n_5259 : i64)
-- : {*[d₀_5256]i32} = {
-- let {r_5311 : [d₀_5256]i32,
-- r_5312 : [d₀_5256]i32} =
-- loop {A_5314 : [d₀_5256]i32,
-- A_5315 : [d₀_5256]i32} = {A_5257, A_5258}
-- for i_5313:i64 < n_5259 do {
-- let {cond_5322 : bool} =
-- eq_i64(i_5313, 0i64)
-- let {defunc_0_scan_res_5332 : [d₀_5256]i32,
-- defunc_0_scan_res_5333 : [d₀_5256]i32} =
-- scanomap(d₀_5256,
-- {A_5314, A_5315},
-- {\ {eta_p_5318 : i32,
-- eta_p_5319 : i32,
-- eta_p_5320 : i32,
-- eta_p_5321 : i32}
-- : {i32,
-- i32} ->
-- let {tmp_5352 : i32} =
-- add32(eta_p_5348, eta_p_5350)
-- let {tmp_5353 : i32} =
-- add32(eta_p_5349, eta_p_5351)
-- in {tmp_5352, tmp_5353},
-- {free_ne_5333, 0i32}},
-- \ {x_5354 : i32,
-- x_5355 : i32}
-- let {lifted_lambda_res_5323 : i32,
-- lifted_lambda_res_5324 : i32} =
-- if cond_5322
-- then {eta_p_5318, eta_p_5319} else {
-- let {tmp_5325 : i32} =
-- add32(eta_p_5318, eta_p_5320)
-- let {tmp_5326 : i32} =
-- add32(eta_p_5319, eta_p_5321)
-- in {tmp_5325, tmp_5326}
-- }
-- : {i32,
-- i32}
-- in {lifted_lambda_res_5323, lifted_lambda_res_5324},
-- {0i32, 0i32}},
-- \ {x_5327 : i32,
-- x_5328 : i32}
-- : {i32,
-- i32} ->
-- {x_5354, x_5355})
-- in {defunc_0_scan_res_5359, defunc_0_scan_res_5360}
-- {x_5327, x_5328})
-- in {defunc_0_scan_res_5332, defunc_0_scan_res_5333}
-- }
-- let {defunc_0_map_res_5356 : [d₁_5271]i32} =
-- copy(r_5341)
-- in {defunc_0_map_res_5356}
-- let {defunc_0_map_res_5329 : [d₀_5256]i32} =
-- copy(r_5311)
-- in {defunc_0_map_res_5329}
-- }

0 comments on commit 22954ec

Please sign in to comment.