Python function

set_virtual_device_target_arch

set_virtual_device_target_arch()

max.driver.set_virtual_device_target_arch(arch: str) → None

source

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

This specifies the GPU architecture (e.g., “sm_80”, “sm_90”) that virtual devices will target when compiling code. Must be called before creating virtual devices via set_virtual_device_count().

Parameters:

arch (str) – The target GPU architecture string (e.g., “sm_80” for Ampere/A100, “sm_90” for Hopper/H100).

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