Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-17 16:24:33 +02:00
Code Issues Packages Projects Releases Wiki Activity

Implement memory estimation for AMDGPUs.

Browse Source 
Stable.
This commit is contained in:
Seunghoon Lee
2023-04-26 17:44:32 +09:00
parent df0e89be48
commit d2d5011bd3
10 changed files with 47 additions and 53 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/dml/optimizer/optimizer.py
+1 -1
View File
@@ -4,5 +4,5 @@ from typing import *
class Optimizer(metaclass=ABCMeta):
driver: Any = None
@abstractmethod
def memory_stats(self, index: int) -> Tuple[int, int]:
def memory_stats(index: int) -> Tuple[int, int]:
pass