Fix CPU tests (#54)

JuliaFolds · Mar 24, 2021 · af7de5f · af7de5f
1 parent e859777
commit af7de5f
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/test/FoldsCUDATests/src/FoldsCUDATests.jl b/test/FoldsCUDATests/src/FoldsCUDATests.jl
@@ -3,19 +3,19 @@ module FoldsCUDATests
 using GPUArrays
 using Test
 
-function include_tests(dir)
+function include_tests(m = @__MODULE__, dir = @__DIR__)
     for file in readdir(dir)
         if match(r"^test_.*\.jl$", file) !== nothing
-            include(joinpath(dir, file))
+            Base.include(m, joinpath(dir, file))
         end
     end
 end
 
-include_tests(@__DIR__)
+include_tests()
 
 module Generic
 using ..FoldsCUDATests: include_tests
-include_tests(joinpath(@__DIR__, "generic"))
+include_tests(@__MODULE__, joinpath(@__DIR__, "generic"))
 end
 
 should_test_gpu() =