Info & Profiling Functions

std::tuple<size_t, size_t, size_t> augpy::meminfo(int device_id)

For the device defined by device_id, return the current used, free, and total memory in bytes.

void augpy::enable_profiler()

Enable the Cuda profiler.

void augpy::disable_profiler()

Disable the Cuda profiler.

nvtxRangeId_t augpy::nvtx_range_start(std::string msg)

Tell the Nvidia profiler to start a new nvtx range. Can be used to place marks in profiling output.

Return

range ID to be used with nvtx_range_end

Parameters

• msg: Message attached to the range

void augpy::nvtx_range_end(nvtxRangeId_t end)

Tell the Nvidia profiler to end the given nvtx range.

Parameters

• end: ID of the range to end