fhelium.native.cuda
CUDA device and peer-topology inspection.
CudaDeviceInventory
data View source
python
CudaDeviceInventory = dict[int, CudaDeviceProperties]1
CudaDeviceProperties
data View source
python
CudaDeviceProperties = dict[str, Any]1
CudaSystemInfo
class View source
python
CudaSystemInfo()1
Bases: TypedDict
CUDA device inventory and native peer-topology result.
Attributes
| Name | Type | Default/value |
|---|---|---|
devices | CudaDeviceInventory | |
p2p | dict[str, Any] |
get_cuda_device_properties
function View source
python
def get_cuda_device_properties() -> CudaDeviceInventory: ...1
Return CUDA device properties without peer-bandwidth measurements.
get_cuda_info
function View source
python
def get_cuda_info(test_p2p_bandwidth: bool=True) -> CudaSystemInfo: ...1
Return CUDA device and peer-topology information.
Parameters
test_p2p_bandwidth: Measure peer-to-peer bandwidth when multiple CUDA devices are visible. The measurement allocates and copies test buffers.