diff --git a/dace_tests/src/polybench.rs b/dace_tests/src/polybench.rs index ee4c76a..76373e4 100644 --- a/dace_tests/src/polybench.rs +++ b/dace_tests/src/polybench.rs @@ -328,6 +328,7 @@ fn _2mm(NI: usize, NJ: usize, NK: usize, NL: usize) -> Rc { Node::new_node(Stmt::Block(vec![ini_loop_ref1, ini_loop_ref2])) } +//created by: Dylan McKellips pub fn cholesky (n: usize) -> Rc { let ubound = n as i32; @@ -338,7 +339,7 @@ pub fn cholesky (n: usize) -> Rc { // create A[i * N + j] /= A[j * N + j]; let s_ref_aij2 = Node::new_ref("a", vec![n,n], |ijk| vec![ijk[0] as usize, ijk[1] as usize]); - let s_ref_ajj = Node::new_ref("a", vec![n], |ijk| vec![ijk[1]]); + let s_ref_ajj = Node::new_ref("a", vec![n], |ijk| vec![ijk[1] as usize]); //create A[i * N + i] -= A[i * N + k] * A[i * N + k]; let s_ref_aii1 = Node::new_ref("a", vec![n], |ijk| vec![ijk[0] as usize]); @@ -353,7 +354,7 @@ pub fn cholesky (n: usize) -> Rc { Node::extend_loop_body(&k1_loop_ref, &s_ref_aij1); Node::extend_loop_body(&k1_loop_ref, &s_ref_aij1); - let j_loop_ref = Node::new_single_loop_dyn_ub("j", move |i| i[0]); + let j_loop_ref = Node::new_single_loop_dyn_ub("j", 0, move |i| i[0]); Node::extend_loop_body(&j_loop_ref, &k1_loop_ref); Node::extend_loop_body(&j_loop_ref, &s_ref_ajj); Node::extend_loop_body(&j_loop_ref, &s_ref_aij2);