diff --git a/mlx/backend/metal/device.cpp b/mlx/backend/metal/device.cpp index 24251ca0e9..eba03c4c6e 100644 --- a/mlx/backend/metal/device.cpp +++ b/mlx/backend/metal/device.cpp @@ -17,8 +17,6 @@ namespace fs = std::filesystem; namespace mlx::core::metal { -static Device metal_device_; - namespace { // TODO nicer way to set this or possibly expose as an environment variable @@ -240,6 +238,7 @@ MTL::ComputePipelineState* Device::get_kernel( } Device& device(mlx::core::Device) { + static Device metal_device_; return metal_device_; }