Merge pull request #106 from computational-cell-analytics/adapt-mps-tiling

changed tiling for mps

Author: lufre1

synapse_net/inference/util.py

@@ -452,8 +452,8 @@ def get_default_tiling(is_2d: bool = False) -> Dict[str, Dict[str, int]]:
         print(f"Determined tile size for CUDA: {tiling}")
 
     elif torch.backends.mps.is_available():  # Check for Apple Silicon (MPS)
-        tile = {"x": 512, "y": 512, "z": 64}
-        halo = {"x": 64, "y": 64, "z": 16}
+        tile = {"x": 256, "y": 256, "z": 16}
+        halo = {"x": 16, "y": 16, "z": 4}
         tiling = {"tile": tile, "halo": halo}
         print(f"Determined tile size for MPS: {tiling}")