For the complete documentation index, see llms.txt. Markdown versions of all pages are available by appending .md to any URL (e.g. /max/get-started.md).
Python function
set_virtual_device_target_arch
set_virtual_device_target_arch()
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!