Fix exception check.

maleadt · maleadt · commit dd4cc1e11636 · 2020-02-18T13:41:56.000+01:00
diff --git a/src/memory.jl b/src/memory.jl
@@ -68,8 +68,8 @@ function actual_alloc(bytes)::Union{Nothing,CuPtr{Nothing}}
     @assert !haskey(allocated, ptr)
     allocated[ptr] = buf
     return ptr
-  catch ex
-    ex.code == CUDAdrv.ERROR_OUT_OF_MEMORY || rethrow()
+  catch err
+    (isa(err, CuError) && err.code == CUDAdrv.ERROR_OUT_OF_MEMORY) || rethrow()
   end
 
   return nothing
