Skip to content

Commit 46b6f27

Browse files
disable unsupported head sizes for AMD in test
1 parent 6609974 commit 46b6f27

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

tests/test-backend-ops.cpp

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2173,7 +2173,11 @@ static bool test_backend(ggml_backend_t backend, test_mode mode, const char * op
21732173
test_cases.emplace_back(new test_timestep_embedding());
21742174
test_cases.emplace_back(new test_leaky_relu());
21752175

2176+
#if defined(GGML_USE_HIPBLAS) && defined(__HIP_PLATFORM_AMD__)
2177+
for (int hs : { 64, 128, }) { // other head sizes not implemented
2178+
#else
21762179
for (int hs : { 64, 80, 128, 256, }) {
2180+
#endif // defined(GGML_USE_HIPBLAS) && defined(__HIP_PLATFORM_AMD__)
21772181
for (int nh : { 32, }) {
21782182
for (int kv : { 512, 1024, }) {
21792183
for (int nb : { 1, 2, 4, 8, }) {

0 commit comments

Comments
 (0)