Skip to content

Commit 03fd63f

Browse files
committed
fix oom ret
1 parent 96a6c7f commit 03fd63f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/test_gemm_oom.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -314,7 +314,7 @@ static int test_gemm_int8_fp16s_oom(int M, int N, int K, int transA, int transB,
314314
float epsilon = 0.001;
315315

316316
int ret = test_layer_oom_opt("Gemm", pd, weights, opt, a, 1, epsilon);
317-
if (ret != 0)
317+
if (ret != 233 && ret != 0)
318318
{
319319
fprintf(stderr, "test_gemm_int8_fp16s_oom failed M=%d N=%d K=%d transA=%d transB=%d output_elemtype=%d output_transpose=%d constantA=%d constantB=%d output_N1M=%d\n", M, N, K, transA, transB, output_elemtype, output_transpose, constantA, constantB, output_N1M);
320320
return ret;

0 commit comments

Comments
 (0)