Skip to content

Commit 388bde0

Browse files
committed
don't log gpu mem on systems without gpu
1 parent 9de1478 commit 388bde0

File tree

1 file changed

+14
-8
lines changed

1 file changed

+14
-8
lines changed

test/runtests.jl

Lines changed: 14 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -9,6 +9,14 @@ function jltest(a, b)
99
y*10f0
1010
end
1111

12+
function log_gpu_mem()
13+
if :cudanative in supported_backends()
14+
info("GPUMem: ", CUDAdrv.Mem.used() / 10^6)
15+
gc()
16+
info(" gc: ", CUDAdrv.Mem.used() / 10^6)
17+
end
18+
end
19+
1220
macro allbackends(title, backendname::Symbol, block)
1321
quote
1422
for backend in supported_backends()
@@ -18,9 +26,7 @@ macro allbackends(title, backendname::Symbol, block)
1826
$(esc(backendname)) = backend
1927
$(esc(block))
2028
end
21-
info("GPUMem: ", CUDAdrv.Mem.used() / 10^6)
22-
gc()
23-
info(" gc: ", CUDAdrv.Mem.used() / 10^6)
29+
log_gpu_mem()
2430
end
2531
end
2632
end
@@ -36,25 +42,25 @@ for backend in supported_backends()
3642
@testset "$backend" begin
3743
include("$(backend).jl")
3844
end
39-
gc()
45+
log_gpu_mem()
4046
end
4147
end
4248

4349
@testset "BLAS" begin
4450
include("blas.jl")
4551
end
46-
gc()
52+
log_gpu_mem()
4753

4854
@testset "Shared" begin
4955
include("shared.jl")
5056
end
51-
gc()
57+
log_gpu_mem()
5258
@testset "Array/Vector Operations" begin
5359
include("indexing.jl")
5460
include("vector.jl")
5561
end
56-
gc()
62+
log_gpu_mem()
5763
@testset "FFT" begin
5864
include("fft.jl")
5965
end
60-
gc()
66+
log_gpu_mem()

0 commit comments

Comments
 (0)