make sure failed testing returns failure status

ahrefs · Jun 8, 2024 · 1bf33cc · 1bf33cc
1 parent 637c1b6
commit 1bf33cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test_gpt2.cu b/test_gpt2.cu
@@ -320,5 +320,5 @@ int main(int argc, char *argv[]) {
     free(expected_grads_memory);
     free(grads_memory_cpu);
     free(grads_memory_cpu_float);
-    return 0;
+    return allok ? EXIT_SUCCESS : EXIT_FAILURE;
 }