arrayfire.cuda module
Functions specific to CUDA backend.
This module provides interoperability with other CUDA libraries.
-
arrayfire.cuda.get_native_id(idx)[source]
Get native (unsorted) CUDA device ID
- Parameters:
- idxint.
Specifies the (sorted) index of the device.
- Returns:
- native_idxinteger denoting the native cuda id.
-
arrayfire.cuda.get_stream(idx)[source]
Get the CUDA stream used for the device idx by ArrayFire.
- Parameters:
- idxint.
Specifies the index of the device.
- Returns:
- streaminteger denoting the stream id.
-
arrayfire.cuda.set_native_id(idx)[source]
Set native (unsorted) CUDA device ID
- Parameters:
- idxint.
Specifies the (unsorted) native index of the device.