Code

Add support for MPS device#176

Code

Merged

constantinpape merged 11 commits intodevcomputational-cell-analytics/micro-sam:devfrom mpscomputational-cell-analytics/micro-sam:mpsCopy head branch name to clipboardSep 7, 2023

+139-16

Commits

Commits on Sep 4, 2023

• Add support for MPS device

  constantinpape

  committed
  81958a1

Commits on Sep 5, 2023

• Fix typo

  constantinpape

  committed
  41951f0

Commits on Sep 6, 2023

• Make micro-sam compatible with mps torch Macbook backend

  GenevieveBuckley

  committed
  9289e38

• Use MPS fallback to CPU, not all torch/torchvision functions are implemented for Apple Silicon

  GenevieveBuckley

  committed
  6916e53

• Merge pull request #178 from GenevieveBuckley/mps-edits

  Show description for 92e3087
  constantinpape

  authored
  92e3087

Commits on Sep 7, 2023

• Vendor batched_mask_to_box function from segment_anything/util/amg.py so we can fix a bug affecting our code

  GenevieveBuckley

  committed
  3e8fd1b

• Endure input to batched_mask_to_box is boolean array, otherwise result output is incorrect

  GenevieveBuckley

  committed
  bbcc649

• Make batched_mask_to_box compatible with MPS Pytorch backend (Apple Silicon)

  GenevieveBuckley

  committed
  72a30dc

• Only pass boolean masks to batched_mask_to_box function (bug means int masks produce incorrect results)

  GenevieveBuckley

  committed
  2c41d23

• Add tests for vendored batched_mask_to_box function from segment_anything/util/amg.py

  GenevieveBuckley

  committed
  6981148

• Merge pull request #180 from GenevieveBuckley/mps

  Show description for 9e1e3bf
  constantinpape

  authored
  9e1e3bf