From 727609df4adf55b862f753261493229aa802966c Mon Sep 17 00:00:00 2001
From: Cheng Li <pistasable@gmail.com>
Date: Wed, 20 Sep 2023 08:04:35 -0700
Subject: [PATCH] add the missing method (#4363)

Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
---
 accelerator/mps_accelerator.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/accelerator/mps_accelerator.py b/accelerator/mps_accelerator.py
index 68e3b6df05c9..8007a50c4bb6 100644
--- 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