Skip to content

Commit b551b3f

Browse files
Reverted ROCM7.2 guards
1 parent 93c118b commit b551b3f

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

tests/cpp/operator/test_cublaslt_gemm.cu

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -228,6 +228,14 @@ void performTest(const TestParams& params) {
228228

229229
#ifdef __HIP_PLATFORM_AMD__
230230

231+
#if HIP_VERSION < 70200000
232+
if (prop.major == 9 && prop.minor == 5 &&
233+
params.transa && !params.transb &&
234+
params.m == 2304 && params.k == 768 && params.n == 4096) {
235+
GTEST_SKIP() << "Skip TN 2304x768x4096 on gfx950 for ROCm < 7.2";
236+
}
237+
#endif
238+
231239
// Enable FP8 GEMM + GELU fusion tests only on MI300 (gfx942) with ROCm > 7.0.
232240
// hipBLASLt currently supports this config only
233241
bool fp8_gelu_fusion_config = false;

0 commit comments

Comments
 (0)