arrayfire.cuda module

Functions specific to CUDA backend.

This module provides interoperability with other CUDA libraries.

arrayfire.cuda.get_native_id(idx)

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)

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)

Set native (unsorted) CUDA device ID

Parameters

idxint.

Specifies the (unsorted) native index of the device.