Refactor

pxl-th · pxl-th · commit 5fbf12f39cad · 2023-04-10T14:44:41.000+03:00
diff --git a/test/runtests.jl b/test/runtests.jl
@@ -26,6 +26,8 @@ macro conditional_testset(name, skip_tests, expr)
     end)
 end
 
+cpu(x) = adapt(CPU(), x)
+
 include("upsample.jl")
 
 function nnlib_testsuite(Backend; skip_tests = Set{String}())
diff --git a/test/upsample.jl b/test/upsample.jl
@@ -1,5 +1,4 @@
 function upsample_testsuite(Backend)
-    cpu(x) = adapt(CPU(), x)
     device(x) = adapt(Backend(), x)
     gradtest_fn = KernelAbstractions.isgpu(Backend()) ? gputest : gradtest
     T = Float32 # TODO test against all supported eltypes for each backend.

Original file line number	Diff line number	Diff line change
`@@ -1,5 +1,4 @@`
`1`	`1`	`function upsample_testsuite(Backend)`
`2`		`- cpu(x) = adapt(CPU(), x)`
`3`	`2`	`device(x) = adapt(Backend(), x)`
`4`	`3`	`gradtest_fn = KernelAbstractions.isgpu(Backend()) ? gputest : gradtest`
`5`	`4`	`T = Float32 # TODO test against all supported eltypes for each backend.`