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 deviceidx 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.