Skip to content

Commit 0e8ee73

Browse files
authored
fix alt shape tests (#903)
1 parent a85222c commit 0e8ee73

File tree

1 file changed

+5
-17
lines changed
  • crates/cubecl-matmul/src/tests/layered/macros/common/problem

1 file changed

+5
-17
lines changed

crates/cubecl-matmul/src/tests/layered/macros/common/problem/problem_size.rs

Lines changed: 5 additions & 17 deletions
Original file line numberDiff line numberDiff line change
@@ -3,7 +3,7 @@ macro_rules! testgen_matmul_problem_size {
33
($kind: ident, $algorithm: ty, $precision: ty, $selection: expr, $layouts: expr) => {
44
use $crate::components::MatmulProblem;
55

6-
#[cfg(not(feature = "matmul_tests_vecmat"))]
6+
#[cfg(feature = "matmul_tests_vecmat")]
77
mod g256x256x256 {
88
use super::*;
99
$crate::testgen_matmul_launch!(
@@ -23,10 +23,7 @@ macro_rules! testgen_matmul_problem_size {
2323
);
2424
}
2525

26-
#[cfg(all(
27-
feature = "matmul_tests_alt_shapes",
28-
not(feature = "matmul_tests_vecmat")
29-
))]
26+
#[cfg(feature = "matmul_tests_alt_shapes")]
3027
mod g100x100x100 {
3128
use super::*;
3229
$crate::testgen_matmul_launch!(
@@ -47,10 +44,7 @@ macro_rules! testgen_matmul_problem_size {
4744
}
4845

4946
// line_size_lhs != line_size_rhs
50-
#[cfg(all(
51-
feature = "matmul_tests_alt_shapes",
52-
not(feature = "matmul_tests_vecmat")
53-
))]
47+
#[cfg(feature = "matmul_tests_alt_shapes")]
5448
mod g100x99x100 {
5549
use super::*;
5650
$crate::testgen_matmul_launch!(
@@ -71,10 +65,7 @@ macro_rules! testgen_matmul_problem_size {
7165
}
7266

7367
// line_size_lhs != line_size_out
74-
#[cfg(all(
75-
feature = "matmul_tests_alt_shapes",
76-
not(feature = "matmul_tests_vecmat")
77-
))]
68+
#[cfg(feature = "matmul_tests_alt_shapes")]
7869
mod g100x100x99 {
7970
use super::*;
8071
$crate::testgen_matmul_launch!(
@@ -94,10 +85,7 @@ macro_rules! testgen_matmul_problem_size {
9485
);
9586
}
9687

97-
#[cfg(all(
98-
feature = "matmul_tests_alt_shapes",
99-
not(feature = "matmul_tests_vecmat")
100-
))]
88+
#[cfg(feature = "matmul_tests_alt_shapes")]
10189
mod g23x1x17 {
10290
use super::*;
10391
$crate::testgen_matmul_launch!(

0 commit comments

Comments
 (0)