mirror of
https://github.com/deepseek-ai/DeepInfer.git
synced 2026-10-28 23:07:09 +08:00
36 lines
808 B
Python
36 lines
808 B
Python
"""Public hardware-neutral discovery functions."""
|
|
|
|
from deep_cluster.hardware.device import (
|
|
accelerator_kind,
|
|
assign_visible_devices,
|
|
device_numa_node,
|
|
get_actual_num_gpus,
|
|
get_bf16_tflops_limit,
|
|
is_sm_major_10,
|
|
num_gpus,
|
|
physical_device_count,
|
|
physical_device_id,
|
|
set_visible_devices,
|
|
visible_device_count,
|
|
visible_device_env_name,
|
|
visible_device_ids,
|
|
visible_devices,
|
|
)
|
|
|
|
__all__ = [
|
|
"accelerator_kind",
|
|
"assign_visible_devices",
|
|
"device_numa_node",
|
|
"get_actual_num_gpus",
|
|
"get_bf16_tflops_limit",
|
|
"is_sm_major_10",
|
|
"num_gpus",
|
|
"physical_device_count",
|
|
"physical_device_id",
|
|
"set_visible_devices",
|
|
"visible_device_count",
|
|
"visible_device_env_name",
|
|
"visible_device_ids",
|
|
"visible_devices",
|
|
]
|