add the missing method (#4363)

Co-authored-by: Olatunji Ruwase <[email protected]>
microsoft · Sep 20, 2023 · 727609d · 727609d
1 parent c581464
commit 727609d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/accelerator/mps_accelerator.py b/accelerator/mps_accelerator.py
@@ -138,6 +138,9 @@ def is_bf16_supported(self):
     def is_fp16_supported(self):
         return False
 
+    def supported_dtypes(self):
+        return [torch.float]
+
     # Misc
     def amp(self):
         return