Browse Source

fblock, ftensor: Fix test skip return value on missing implementation

Samuel Thibault 3 years ago
parent
commit
577407d420
2 changed files with 15 additions and 14 deletions
  1. 9 8
      examples/filters/fblock.c
  2. 6 6
      examples/filters/ftensor.c

+ 9 - 8
examples/filters/fblock.c

@@ -157,11 +157,8 @@ int main(void)
                 task->cl_arg_size = sizeof(multiplier);
 
                 ret = starpu_task_submit(task);
-                if (ret)
-		{
-                        FPRINTF(stderr, "Error when submitting task\n");
-                        exit(ret);
-                }
+                if (ret == -ENODEV) goto enodev;
+                STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
         }
 
         /* Unpartition the data, unregister it from StarPU and shutdown */
@@ -178,8 +175,12 @@ int main(void)
         FPRINTF(stderr, "OUT Block\n");
         print_block(block, NX, NY, NZ, NX, NX*NY);
 
-	free(block);
+        free(block);
+        
+        starpu_shutdown();
+        return 0;
 
-	starpu_shutdown();
-	return 0;
+enodev:
+        starpu_shutdown();
+        return 77;
 }

+ 6 - 6
examples/filters/ftensor.c

@@ -167,11 +167,8 @@ int main(void)
         task->cl_arg_size = sizeof(multiplier);
 
         ret = starpu_task_submit(task);
-        if (ret)
-        {
-            FPRINTF(stderr, "Error when submitting task\n");
-            exit(ret);
-        }
+        if (ret == -ENODEV) goto enodev;
+        STARPU_CHECK_RETURN_VALUE(ret, "starpu_task_submit");
     }
 
     /* Unpartition the data, unregister it from StarPU and shutdown */
@@ -188,4 +185,7 @@ int main(void)
     starpu_shutdown();
     return 0;
 
-}    
+enodev:
+    starpu_shutdown();
+    return 77;
+}