Cleanup

stemann · Dec 15, 2024 · 10c7a60 · 10c7a60
1 parent 074a558
commit 10c7a60
Showing 1 changed file with 1 addition and 9 deletions.
diff --git a/src/device.jl b/src/device.jl
@@ -7,9 +7,6 @@ mutable struct Device
     mlx_device::Wrapper.mlx_device
 
     function Device(mlx_device::Wrapper.mlx_device)
-        if mlx_device.ctx == C_NULL # TODO true for Wrapper.mlx_device_new()
-            throw(MLXException("Cannot create device from a null device"))
-        end
         this = new(mlx_device)
         finalizer(d -> Wrapper.mlx_device_free(d.mlx_device), this)
         return this
@@ -19,12 +16,7 @@ end
 function Device(; device_type = DeviceTypeCPU, index::Int = 0)
     mlx_device_type = Wrapper.mlx_device_type(UInt32(device_type))
     mlx_device = Wrapper.mlx_device_new_type(mlx_device_type, index)
-    try
-        return Device(mlx_device)
-    catch
-        Wrapper.mlx_device_free(mlx_device)
-        rethrow()
-    end
+    return Device(mlx_device)
 end
 
 function Base.getproperty(device::Device, name::Symbol)