update

Author: NADOOITChristophBa

.DS_Store

@@ -209,6 +209,7 @@ if(WITH_METAL)
     "-framework Metal"
     "-framework Foundation"
     "-framework MetalKit"
+    "-framework MetalPerformanceShaders"
   )
 endif()
 

CMakeLists.txt

@@ -7,6 +7,7 @@ set -e
 # Use your universal binary Python (replace 'ub python3' with your actual command if different)
 export ARCHFLAGS='-arch x86_64 -arch arm64'
 export CMAKE_OSX_ARCHITECTURES='arm64;x86_64'
+
 # Install build tool if needed
 uv pip install --upgrade pip build
 
@@ -16,7 +17,7 @@ rm -rf build/ dist/ ctranslate2.egg-info/
 
 # Build sdist and universal2 wheel
 echo "Building universal2 wheel..."
-uv run -m build --sdist --wheel
+CMAKE_ARGS="-DWITH_METAL=ON" uv run -m build --sdist --wheel
 
 echo "\nBuild complete. Check dist/ for the universal2 wheel."
 ls -lh dist/

python/.DS_Store

@@ -151,7 +151,7 @@ namespace ctranslate2 {
 
                  Arguments:
                    model_path: Path to the CTranslate2 model directory.
-                   device: Device to use (possible values are: cpu, cuda, auto).
+                   device: Device to use (possible values are: cpu, cuda, metal, auto).
                    device_index: Device IDs where to place this generator on.
                    compute_type: Model computation type or a dictionary mapping a device name
                      to the computation type (possible values are: default, auto, int8, int8_float32,

python/build_full.log

@@ -109,6 +109,7 @@ namespace ctranslate2 {
       py::enum_<Device>(m, "Device")
         .value("cpu", Device::CPU)
         .value("cuda", Device::CUDA)
+        .value("metal", Device::METAL)
         ;
 
       py::class_<StorageView>(

python/build_metal.log

@@ -300,7 +300,7 @@ namespace ctranslate2 {
 
                  Arguments:
                    model_path: Path to the CTranslate2 model directory.
-                   device: Device to use (possible values are: cpu, cuda, auto).
+                   device: Device to use (possible values are: cpu, cuda, metal, auto).
                    device_index: Device IDs where to place this generator on.
                    compute_type: Model computation type or a dictionary mapping a device name
                      to the computation type (possible values are: default, auto, int8, int8_float32,