Skip to content

Commit

Permalink
Further reduce large gemm tests
Browse files Browse the repository at this point in the history
I suspect that we're running into more oom issues because of tharnsB
and transC and the like. So, move large gemm tests into their own file
where there'll only be one or two tests that run, thus ensuring we
don't overload the machines.
  • Loading branch information
krzysz00 committed Nov 3, 2023
1 parent fcfb27a commit d9e58c2
Show file tree
Hide file tree
Showing 2 changed files with 21 additions and 0 deletions.
1 change: 1 addition & 0 deletions mlir/test/e2e/CMakeLists.txt
Original file line number Diff line number Diff line change
Expand Up @@ -47,6 +47,7 @@ if (ROCK_E2E_TEST_ENABLED)
if (ROCK_E2E_TEST_SUITES STREQUAL "" OR ROCK_E2E_TEST_SUITES STREQUAL "part3")
list(APPEND CONFIGS
Resnext101Config
LargeGemm
PaddedGemmConfig
)
endif()
Expand Down
20 changes: 20 additions & 0 deletions mlir/test/e2e/LargeGemm.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,20 @@
directory = "LargeGemm"
prefix = "rocmlir-gen"
suffix = "--operation gemm --arch %arch %pv %random_data %rocmlir_gen_flags | rocmlir-driver -c | mlir-cpu-runner -O2 --shared-libs=%linalg_test_lib_dir/libmlir_rocm_runtime%shlibext,%conv_validation_wrapper_library_dir/libconv-validation-wrappers%shlibext,%linalg_test_lib_dir/libmlir_runner_utils%shlibext,%linalg_test_lib_dir/libmlir_float16_utils%shlibext --entry-point-result=void | FileCheck %s --check-prefix="

# Note: Just f32 for ensuring lower machine load
[[axis]]
name = "data type"
values = ["f32"]
prefix = "-t "

## Gemm variants
[[suite]]
name = "large_gemm"

# Large gemm, input padding
[[suite.test]]
config = "-g 1 -m 1 -k 32768 -n 32769"
# Large gemm, no padding.
[[suite.test]]
config = "-g 1 -m 64 -k 32768 -n 32768"

0 comments on commit d9e58c2

Please sign in to comment.