Spaces:

bilegentile
/

test

Runtime error

test / modules /dml /memory_amd /__init__.py

Upload folder using huggingface_hub

c19ca42 verified over 2 years ago

342 Bytes

from .driver.atiadlxx import ATIADLxx


class AMDMemoryProvider:
    driver: ATIADLxx = ATIADLxx()

    @staticmethod
    def mem_get_info(index):
        usage = AMDMemoryProvider.driver.get_dedicated_vram_usage(index) * (1 << 20)
        return (AMDMemoryProvider.driver.iHyperMemorySize - usage, AMDMemoryProvider.driver.iHyperMemorySize)

	from .driver.atiadlxx import ATIADLxx


	class AMDMemoryProvider:
	driver: ATIADLxx = ATIADLxx()

	@staticmethod
	def mem_get_info(index):
	usage = AMDMemoryProvider.driver.get_dedicated_vram_usage(index) * (1 << 20)
	return (AMDMemoryProvider.driver.iHyperMemorySize - usage, AMDMemoryProvider.driver.iHyperMemorySize)