libs

Author: KSemenenko

.github/workflows/ci.yml

@@ -78,6 +78,7 @@ jobs:
         run: |
           mkdir -p artifacts/native/osx-arm64
           cp native/build/macos/libmlxsharp.dylib artifacts/native/osx-arm64/
+          cp native/build/macos/extern/mlx/mlx/backend/metal/kernels/mlx.metallib artifacts/native/osx-arm64/
 
       - name: Upload native artifact
         uses: actions/upload-artifact@v4
@@ -141,6 +142,7 @@ jobs:
         run: |
           mkdir -p src/MLXSharp/runtimes/osx-arm64/native
           cp artifacts/native/osx-arm64/libmlxsharp.dylib src/MLXSharp/runtimes/osx-arm64/native/
+          cp artifacts/native/osx-arm64/mlx.metallib src/MLXSharp/runtimes/osx-arm64/native/
           mkdir -p src/MLXSharp/runtimes/linux-x64/native
           cp artifacts/native/linux-x64/libmlxsharp.so src/MLXSharp/runtimes/linux-x64/native/
 
@@ -152,6 +154,7 @@ jobs:
           TEST_OUTPUT="src/MLXSharp.Tests/bin/Release/net9.0"
           mkdir -p "$TEST_OUTPUT/runtimes/osx-arm64/native"
           cp src/MLXSharp/runtimes/osx-arm64/native/libmlxsharp.dylib "$TEST_OUTPUT/runtimes/osx-arm64/native/"
+          cp src/MLXSharp/runtimes/osx-arm64/native/mlx.metallib "$TEST_OUTPUT/runtimes/osx-arm64/native/"
           ls -la "$TEST_OUTPUT/runtimes/osx-arm64/native/"
 
       - name: Run tests
@@ -171,13 +174,14 @@ jobs:
           mkdir -p artifacts/packages
           mkdir -p artifacts/native
           cp artifacts/native/osx-arm64/libmlxsharp.dylib artifacts/native/
+          cp artifacts/native/osx-arm64/mlx.metallib artifacts/native/
           cp artifacts/native/linux-x64/libmlxsharp.so artifacts/native/
 
       - name: Pack MLXSharp library
-        run: dotnet pack src/MLXSharp/MLXSharp.csproj --configuration Release --output artifacts/packages -p:MLXSharpMacNativeBinary=$GITHUB_WORKSPACE/artifacts/native/osx-arm64/libmlxsharp.dylib -p:MLXSharpLinuxNativeBinary=$GITHUB_WORKSPACE/artifacts/native/linux-x64/libmlxsharp.so
+        run: dotnet pack src/MLXSharp/MLXSharp.csproj --configuration Release --output artifacts/packages -p:MLXSharpMacNativeBinary=$GITHUB_WORKSPACE/artifacts/native/osx-arm64/libmlxsharp.dylib -p:MLXSharpMacMetallibBinary=$GITHUB_WORKSPACE/artifacts/native/osx-arm64/mlx.metallib -p:MLXSharpLinuxNativeBinary=$GITHUB_WORKSPACE/artifacts/native/linux-x64/libmlxsharp.so
 
       - name: Pack MLXSharp.SemanticKernel library
-        run: dotnet pack src/MLXSharp.SemanticKernel/MLXSharp.SemanticKernel.csproj --configuration Release --output artifacts/packages -p:MLXSharpMacNativeBinary=$GITHUB_WORKSPACE/artifacts/native/osx-arm64/libmlxsharp.dylib -p:MLXSharpLinuxNativeBinary=$GITHUB_WORKSPACE/artifacts/native/linux-x64/libmlxsharp.so -p:MLXSharpSkipLinuxNativeValidation=true
+        run: dotnet pack src/MLXSharp.SemanticKernel/MLXSharp.SemanticKernel.csproj --configuration Release --output artifacts/packages -p:MLXSharpMacNativeBinary=$GITHUB_WORKSPACE/artifacts/native/osx-arm64/libmlxsharp.dylib -p:MLXSharpMacMetallibBinary=$GITHUB_WORKSPACE/artifacts/native/osx-arm64/mlx.metallib -p:MLXSharpLinuxNativeBinary=$GITHUB_WORKSPACE/artifacts/native/linux-x64/libmlxsharp.so -p:MLXSharpSkipLinuxNativeValidation=true
 
       - name: Verify package contains native libraries
         run: |
@@ -205,6 +209,12 @@ jobs:
                   echo "  ✗ macOS library missing"
                   package_missing=1
                 fi
+                if unzip -l "${package}" | grep -q "runtimes/osx-arm64/native/mlx.metallib"; then
+                  echo "  ✓ macOS metallib present"
+                else
+                  echo "  ✗ macOS metallib missing"
+                  package_missing=1
+                fi
                 if unzip -l "${package}" | grep -q "runtimes/linux-x64/native/libmlxsharp.so"; then
                   echo "  ✓ Linux library present"
                 else

src/MLXSharp/MLXSharp.csproj

@@ -20,6 +20,8 @@
     <MLXSharpMacNativeDestination>$([System.IO.Path]::Combine('$(MSBuildProjectDirectory)','runtimes','osx-arm64','native','libmlxsharp.dylib'))</MLXSharpMacNativeDestination>
     <MLXSharpSkipMacNativeValidation Condition="'$(MLXSharpSkipMacNativeValidation)' == ''">false</MLXSharpSkipMacNativeValidation>
     <MLXSharpMacNativeDestinationDir>$([System.IO.Path]::GetDirectoryName('$(MLXSharpMacNativeDestination)'))</MLXSharpMacNativeDestinationDir>
+    <MLXSharpMacMetallibBinary Condition="'$(MLXSharpMacMetallibBinary)' == ''">$([System.IO.Path]::Combine('$(MSBuildProjectDirectory)','..','..','native','build','macos','extern','mlx','mlx','backend','metal','kernels','mlx.metallib'))</MLXSharpMacMetallibBinary>
+    <MLXSharpMacMetallibDestination Condition="'$(MLXSharpMacMetallibDestination)' == ''">$([System.IO.Path]::Combine('$(MLXSharpMacNativeDestinationDir)','mlx.metallib'))</MLXSharpMacMetallibDestination>
 
     <MLXSharpLinuxNativeBinary Condition="'$(MLXSharpLinuxNativeBinary)' == ''">$([System.IO.Path]::Combine('$(MSBuildProjectDirectory)','..','..','native','build','linux','libmlxsharp.so'))</MLXSharpLinuxNativeBinary>
     <MLXSharpLinuxNativeDestination>$([System.IO.Path]::Combine('$(MSBuildProjectDirectory)','runtimes','linux-x64','native','libmlxsharp.so'))</MLXSharpLinuxNativeDestination>
@@ -33,6 +35,8 @@
     <ItemGroup>
       <_MacNativeSource Include="$(MLXSharpMacNativeBinary)"
                         Condition="Exists('$(MLXSharpMacNativeBinary)')" />
+      <_MacMetallibSource Include="$(MLXSharpMacMetallibBinary)"
+                          Condition="Exists('$(MLXSharpMacMetallibBinary)')" />
     </ItemGroup>
 
     <MakeDir Directories="$(MLXSharpMacNativeDestinationDir)"
@@ -42,6 +46,10 @@
           DestinationFiles="$(MLXSharpMacNativeDestination)"
           SkipUnchangedFiles="true"
           Condition="'@(_MacNativeSource)' != ''" />
+    <Copy SourceFiles="@(_MacMetallibSource)"
+          DestinationFiles="$(MLXSharpMacMetallibDestination)"
+          SkipUnchangedFiles="true"
+          Condition="'@(_MacMetallibSource)' != ''" />
 
     <ItemGroup>
       <None Include="$(MLXSharpMacNativeDestination)"
@@ -51,10 +59,19 @@
         <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
         <Link>runtimes\osx-arm64\native\%(Filename)%(Extension)</Link>
       </None>
+      <None Include="$(MLXSharpMacMetallibDestination)"
+            Condition="Exists('$(MLXSharpMacMetallibDestination)')">
+        <Pack>true</Pack>
+        <PackagePath>runtimes/osx-arm64/native/%(Filename)%(Extension)</PackagePath>
+        <CopyToOutputDirectory>PreserveNewest</CopyToOutputDirectory>
+        <Link>runtimes\osx-arm64\native\%(Filename)%(Extension)</Link>
+      </None>
     </ItemGroup>
 
     <Error Condition="'$(MLXSharpSkipMacNativeValidation)' != 'true' and !Exists('$(MLXSharpMacNativeDestination)')"
            Text="libmlxsharp.dylib is missing. Build the native library or supply MLXSharpMacNativeBinary before packing." />
+    <Error Condition="'$(MLXSharpSkipMacNativeValidation)' != 'true' and !Exists('$(MLXSharpMacMetallibDestination)')"
+           Text="mlx.metallib is missing. Build the native library or supply MLXSharpMacMetallibBinary before packing." />
   </Target>
 
   <Target Name="EnsureLinuxNativeBinary"