Skip to content
Snippets Groups Projects
Commit aa1cf408 authored by prrathi's avatar prrathi
Browse files

tmp change matmul test

parent 7e45a3c3
No related branches found
No related tags found
1 merge request!115GPU backend
......@@ -10,11 +10,13 @@ juno_build::juno!("matmul");
fn main() {
async_std::task::block_on(async {
const I: usize = 256;
const J: usize = 64;
const K: usize = 128;
let mut a: Box<[i32]> = (0..I * J).map(|_| random::<i32>() % 100).collect();
let mut b: Box<[i32]> = (0..J * K).map(|_| random::<i32>() % 100).collect();
const I: usize = 4;
const J: usize = 2;
const K: usize = 8;
// let mut a: Box<[i32]> = (0..I * J).map(|_| random::<i32>() % 100).collect();
// let mut b: Box<[i32]> = (0..J * K).map(|_| random::<i32>() % 100).collect();
let mut a: Box<[i32]> = (0..I * J).map(|i| (i as i32) % 100).collect();
let mut b: Box<[i32]> = (0..J * K).map(|i| (i as i32) % 100).collect();
let mut correct_c: Box<[i32]> = (0..I * K).map(|_| 0).collect();
for i in 0..I {
for k in 0..K {
......@@ -41,7 +43,7 @@ fn main() {
let c = r.run(I as u64, J as u64, K as u64, a, b).await;
let mut c_cpu: Box<[i32]> = vec![0; correct_c.len()].into_boxed_slice();
c.to_cpu_ref(&mut c_cpu);
assert_eq!(c_cpu.as_ref(), correct_c.as_ref());
assert_eq!(c_cpu.as_ref(), &*correct_c);
}
});
}
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment