Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 09:14:35 +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/amd/__init__.py
+2 -2
View File
@@ -1,7 +1,7 @@
from modules.dml.optimizer.optimizer import Optimizer
from driver.atiadlxx import ATIADLxx
from .driver.atiadlxx import ATIADLxx
class AMDOptimizer(Optimizer):
driver: ATIADLxx = ATIADLxx()
def memory_stats(self, index):
def memory_stats(index):
return (AMDOptimizer.driver.iHyperMemorySize, AMDOptimizer.driver.get_dedicated_vram_usage(index))