Skip to main content

AMD_GPU

Create an AMD GPU device instance.

def AMD_GPU(device: AMD_GPUType) -> Device: ...

Create an AMD GPU device instance.

Parameters​

NameTypeDescription
deviceAMD_GPUTypeThe specific model of AMD GPU to instantiate, such as 'MI100', 'MI210', 'MI250', 'MI300X', etc.

Returns​

TypeDescription
DeviceA Device instance configured with the specified AMD GPU type, a quantity of 1, and the 'AMD_GPU' device class.