@@ -28,7 +28,9 @@ using common::half;
 
 af_err af_set_backend(const af_backend bknd) {
     try {
-        if (bknd != getBackend()) { return AF_ERR_ARG; }
+        if (bknd != getBackend() && bknd != AF_BACKEND_DEFAULT) {
+            return AF_ERR_ARG;
+        }
     }
     CATCHALL;
Original file line number	Diff line number	Diff line change
`@@ -28,7 +28,9 @@ using common::half;`
`28`	`28`
`29`	`29`	`af_err af_set_backend(const af_backend bknd) {`
`30`	`30`	`try {`
`31`		`- if (bknd != getBackend()) { return AF_ERR_ARG; }`
	`31`	`+ if (bknd != getBackend() && bknd != AF_BACKEND_DEFAULT) {`
	`32`	`+ return AF_ERR_ARG;`
	`33`	`+ }`
`32`	`34`	`}`
`33`	`35`	`CATCHALL;`
`34`	`36`