Add supports_float64() to OmniPlatform and clean up MPS (#2488)

Signed-off-by: Xiaodong Ye <yeahdongcn@gmail.com>
This commit is contained in:
R0CKSTAR authored and GitHub committed 2026-04-07 11:15:02 +08:00
1 parent 368de99f08
commit 7a72f34ce4
7 files changed
+16 -12

No files matched your search

+4
View File
@@ -117,6 +117,10 @@ class OmniPlatform(Platform):
def supports_cpu_offload(cls) -> bool:
return True
@classmethod
def supports_float64(cls) -> bool:
return True
@classmethod
def set_device_control_env_var(cls, devices: str | int | None) -> None:
import os