Skip to content

Commit

Permalink
remove outdated test
Browse files Browse the repository at this point in the history
  • Loading branch information
louisfd committed Sep 17, 2024
1 parent fac3cfc commit c11e1d6
Show file tree
Hide file tree
Showing 2 changed files with 1 addition and 42 deletions.
38 changes: 1 addition & 37 deletions crates/cubecl-linalg/src/matmul/tests/tiling2d/compute_loop.rs
Original file line number Diff line number Diff line change
Expand Up @@ -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,
Expand Down Expand Up @@ -174,37 +172,3 @@ pub fn compute_loop_unit_test<R: Runtime>(device: &R::Device) {
];
assert_equals::<R>(&client, results, expected);
}

/// Exported test
pub fn compute_loop_unit_offset_test<R: Runtime>(device: &R::Device) {
let client = R::client(device);
let lhs = range_tensor_transposed::<R>(&client, 8, 4);
let rhs = range_tensor::<R>(&client, 4, 8);
let results = create_empty::<R>(&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::<f32, R>(
&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::<R>(&client, results, expected);
}
Original file line number Diff line number Diff line change
Expand Up @@ -21,11 +21,6 @@ macro_rules! testgen_tiling2d_internal {
tests::tiling2d::compute_loop::compute_loop_unit_test::<TestRuntime>(&Default::default())
}

#[test]
pub fn tiling2d_matmul_compute_loop_unit_offset_test() {
tests::tiling2d::compute_loop::compute_loop_unit_offset_test::<TestRuntime>(&Default::default())
}

#[test]
pub fn load_lhs_transposed_unit_test() {
tests::tiling2d::load_shared_memory::load_lhs_transposed_unit_test::<TestRuntime>(
Expand Down

0 comments on commit c11e1d6

Please sign in to comment.