automatic/modules/dml/optimizer/amd/__init__.py

8 lines
294 B
Python

from modules.dml.optimizer.optimizer import Optimizer
from .driver.atiadlxx import ATIADLxx
class AMDOptimizer(Optimizer):
driver: ATIADLxx = ATIADLxx()
def memory_stats(index):
return (AMDOptimizer.driver.iHyperMemorySize, AMDOptimizer.driver.get_dedicated_vram_usage(index))