Python function

set_virtual_device_api

set_virtual_device_api()

max.driver.set_virtual_device_api(api: str) → None

source

Sets the target API for virtual devices in compile-only mode.

This specifies which GPU API (e.g., “cuda”, “hip”, “metal”) virtual devices will use for compilation. Must be called before creating virtual devices via set_virtual_device_count().

Parameters:

api (str) – The target API string (e.g., “cuda” for NVIDIA, “hip” for AMD, “metal” for Apple).

Was this page helpful?

Thank you! We'll create more content like this.

Thank you for helping us improve!

😔 What went wrong?

Some code doesn’t work

It includes inaccurate information

It's missing information I need

It was difficult to understand

Other

Submit