Reduce duration of cpu_backend_gemm_test
PiperOrigin-RevId: 247951547
This commit is contained in:
parent
fdd04bd83d
commit
2b4a08db39
@ -594,7 +594,7 @@ TYPED_TEST(CpuBackendGemmTest, Rectangular) {
|
|||||||
|
|
||||||
TYPED_TEST(CpuBackendGemmTest, HighlyRectangular) {
|
TYPED_TEST(CpuBackendGemmTest, HighlyRectangular) {
|
||||||
std::vector<std::tuple<int, int, int>> shapes;
|
std::vector<std::tuple<int, int, int>> shapes;
|
||||||
for (int size = 1; size <= 10000; size *= 10) {
|
for (int size = 1; size <= 1000; size *= 10) {
|
||||||
shapes.push_back(std::make_tuple(size, 10, 10));
|
shapes.push_back(std::make_tuple(size, 10, 10));
|
||||||
shapes.push_back(std::make_tuple(10, size, 10));
|
shapes.push_back(std::make_tuple(10, size, 10));
|
||||||
shapes.push_back(std::make_tuple(10, 10, size));
|
shapes.push_back(std::make_tuple(10, 10, size));
|
||||||
|
Loading…
x
Reference in New Issue
Block a user