Implement torch.dml.

VERY UNSTABLE & NOT TESTED.
2026-09-20 01:31:13 +02:00 · 2023-04-26 12:21:44 +09:00
parent 8ea2fe15e7
commit 09ae33cdf7
19 changed files with 347 additions and 97 deletions
@@ -0,0 +1,8 @@
+from abc import *
+from typing import *
+
+class Optimizer(metaclass=ABCMeta):
+    driver: Any = None
+    @abstractmethod
+    def memory_stats(self, index: int) -> Tuple[int, int]:
+        pass