OutOfGPUMemoryError definition

Author: michel2323

lib/level-zero/oneL0.jl

@@ -15,11 +15,11 @@ else
     using oneAPI_Level_Zero_Loader_jll
 end
 
-include("common.jl")
 include("utils.jl")
 include("pointer.jl")
 
 # core API
+include("common.jl")
 include("libze.jl")
 
 # level zero's structure types are often assumed to be zero-initialized (`= {}` in C).
@@ -88,6 +88,39 @@ end
 include("error.jl")
 include("driver.jl")
 include("device.jl")
+
+# Define OutOfGPUMemoryError after device.jl to ensure ZeDevice is available
+export OutOfGPUMemoryError
+
+"""
+    OutOfGPUMemoryError(sz::Integer=0, dev::ZeDevice)
+
+An operation allocated too much GPU memory.
+"""
+struct OutOfGPUMemoryError <: Exception
+  sz::Int
+  dev::Union{ZeDevice, Nothing}
+
+  function OutOfGPUMemoryError(sz::Integer=0, dev::Union{ZeDevice, Nothing}=nothing)
+    new(sz, dev)
+  end
+end
+
+function Base.showerror(io::IO, err::OutOfGPUMemoryError)
+    print(io, "Out of GPU memory")
+    if err.sz > 0
+      print(io, " trying to allocate $(Base.format_bytes(err.sz))")
+    end
+    if err.dev !== nothing
+        print(" on device $(properties(err.dev).name)")
+        if length(memory_properties(err.dev)) == 1
+            # XXX: how to handle multiple memories?
+            print(" with $(Base.format_bytes(only(memory_properties(err.dev)).totalSize))")
+        end
+    end
+    return io
+end
+
 include("context.jl")
 include("cmdqueue.jl")
 include("cmdlist.jl")
@@ -131,7 +164,7 @@ function __init__()
         zeInit(0)
         functional[] = true
     catch err
-        @warn "Failed to initialize oneAPI"
+        @error "Failed to initialize oneAPI" exception=(err,catch_backtrace())
         functional[] = false
         return
     end

src/pool.jl

@@ -1,32 +1,3 @@
-export OutOfGPUMemoryError
-
-"""
-    OutOfGPUMemoryError()
-
-An operation allocated too much GPU memory.
-"""
-struct OutOfGPUMemoryError <: Exception
-  sz::Int
-  dev::ZeDevice
-
-  function OutOfGPUMemoryError(sz::Integer=0, dev::ZeDevice=device())
-    new(sz, dev)
-  end
-end
-
-function Base.showerror(io::IO, err::OutOfGPUMemoryError)
-    print(io, "Out of GPU memory")
-    if err.sz > 0
-      print(io, " trying to allocate $(Base.format_bytes(err.sz))")
-    end
-    print(" on device $(properties(err.dev).name)")
-    if length(memory_properties(err.dev)) == 1
-        # XXX: how to handle multiple memories?
-        print(" with $(Base.format_bytes(only(memory_properties(err.dev)).totalSize))")
-    end
-    return io
-end
-
 function allocate(::Type{oneL0.DeviceBuffer}, ctx, dev, bytes::Int, alignment::Int)
     bytes == 0 && return oneL0.DeviceBuffer(ZE_NULL, bytes, ctx, dev)