Disable MLXFastKernel on Linux (needs new implementation)

Author: Joannis

Libraries/MLXLLM/Models/SSM.swift

@@ -18,7 +18,7 @@ public func computeDt(_ dt: MLXArray, _ dtBias: MLXArray, _ timeStepLimit: (Floa
     return MLX.clip(dt, min: timeStepLimit.0, max: timeStepLimit.1)
 }
 
-#if canImport(MLXFast)
+#if os(macOS) || os(iOS) | | os(tvOS) | | os(watchOS) || os(visionOS)
     private func makeSSMKernel() -> MLXFast.MLXFastKernel? {
         let source = """
                 auto n = thread_position_in_grid.z;
@@ -87,7 +87,7 @@ public func computeDt(_ dt: MLXArray, _ dtBias: MLXArray, _ timeStepLimit: (Floa
     }
 #endif
 
-#if canImport(MLXFast)
+#if os(macOS) || os(iOS) | | os(tvOS) | | os(watchOS) || os(visionOS)
     func ssmUpdateKernel(
         hiddenStates: MLXArray,
         ALog: MLXArray,

Libraries/MLXLMCommon/InterpolationUtils.swift

@@ -11,7 +11,7 @@ import Foundation
 import MLX
 
 // MARK: - Kernel Sources
-
+#if os(macOS) || os(iOS) || os(tvOS) || os(watchOS) || os(visionOS)
 private func makeBicubicInterpolationKernel() -> MLXFast.MLXFastKernel {
     let header = """
         // Bicubic kernel function
@@ -452,3 +452,4 @@ public func nearestInterpolate(
 ) -> MLXArray {
     nearestInterpolate(x, scaleFactor: (scaleFactor, scaleFactor))
 }
+#endif