From c11e1d61ffb7e614c8cfaffceb7fce8dd3b1894e Mon Sep 17 00:00:00 2001 From: louisfd Date: Tue, 17 Sep 2024 11:48:07 -0400 Subject: [PATCH] remove outdated test --- .../src/matmul/tests/tiling2d/compute_loop.rs | 38 +------------------ .../tests/matmul/tiling2d/matmul_internal.rs | 5 --- 2 files changed, 1 insertion(+), 42 deletions(-) diff --git a/crates/cubecl-linalg/src/matmul/tests/tiling2d/compute_loop.rs b/crates/cubecl-linalg/src/matmul/tests/tiling2d/compute_loop.rs index ef3280ad..de5c831f 100644 --- a/crates/cubecl-linalg/src/matmul/tests/tiling2d/compute_loop.rs +++ b/crates/cubecl-linalg/src/matmul/tests/tiling2d/compute_loop.rs @@ -3,9 +3,7 @@ use cubecl_core::prelude::*; use crate::matmul::tiling2d::outer_product::tile_outer_product; use crate::matmul::{ - tests::test_utils::{ - assert_equals, create_empty, make_tiling2d_config, range_tensor, range_tensor_transposed, - }, + tests::test_utils::{assert_equals, create_empty, make_tiling2d_config, range_tensor}, tiling2d::{ base::{Coordinates, TILE_SIZE}, compute_loop::compute_loop, @@ -174,37 +172,3 @@ pub fn compute_loop_unit_test(device: &R::Device) { ]; assert_equals::(&client, results, expected); } - -/// Exported test -pub fn compute_loop_unit_offset_test(device: &R::Device) { - let client = R::client(device); - let lhs = range_tensor_transposed::(&client, 8, 4); - let rhs = range_tensor::(&client, 4, 8); - let results = create_empty::(&client, TILE_SIZE, TILE_SIZE); - let cube_dim = CubeDim::new(1, 1, 1); - let cube_count = CubeCount::Static(1, 1, 1); - - let config = make_tiling2d_config(4, 8, 4); - - unsafe { - compute_loop_test::launch_unchecked::( - &R::client(device), - cube_count, - cube_dim, - TensorArg::from_raw_parts(&lhs.handle, &lhs.strides, &lhs.shape, TILE_SIZE as u8), - TensorArg::from_raw_parts(&rhs.handle, &rhs.strides, &rhs.shape, TILE_SIZE as u8), - ScalarArg::new(4), - ScalarArg::new(4), - ArrayArg::from_raw_parts(&results, 16, 1), - 16, - 16, - config, - ); - }; - - let expected = &[ - 1160.0, 1230.0, 1300.0, 1370.0, 1416.0, 1502.0, 1588.0, 1674.0, 1672.0, 1774.0, 1876.0, - 1978.0, 1928.0, 2046.0, 2164.0, 2282.0, - ]; - assert_equals::(&client, results, expected); -} diff --git a/crates/cubecl-linalg/src/tests/matmul/tiling2d/matmul_internal.rs b/crates/cubecl-linalg/src/tests/matmul/tiling2d/matmul_internal.rs index d6e76074..b77a5339 100644 --- a/crates/cubecl-linalg/src/tests/matmul/tiling2d/matmul_internal.rs +++ b/crates/cubecl-linalg/src/tests/matmul/tiling2d/matmul_internal.rs @@ -21,11 +21,6 @@ macro_rules! testgen_tiling2d_internal { tests::tiling2d::compute_loop::compute_loop_unit_test::(&Default::default()) } - #[test] - pub fn tiling2d_matmul_compute_loop_unit_offset_test() { - tests::tiling2d::compute_loop::compute_loop_unit_offset_test::(&Default::default()) - } - #[test] pub fn load_lhs_transposed_unit_test() { tests::tiling2d::load_shared_memory::load_lhs_transposed_unit_test::(