cuda.bindings._v2.driver.device_set_graph_mem_attribute#

cuda.bindings._v2.driver.device_set_graph_mem_attribute(int device, int attr, intptr_t value)#

Set asynchronous allocation attributes related to graphs.

Valid attributes are:.

  • CU_GRAPH_MEM_ATTR_USED_MEM_HIGH: High watermark of memory, in bytes, associated with graphs since the last time it was reset. High watermark can only be reset to zero.

  • CU_GRAPH_MEM_ATTR_RESERVED_MEM_HIGH: High watermark of memory, in bytes, currently allocated for use by the CUDA graphs asynchronous allocator.

Parameters:

  • device (int) – Specifies the scope of the query.

  • attr (GraphMemAttribute) – attribute to get.

  • value (intptr_t) – pointer to value to set.

See also

cuDeviceSetGraphMemAttribute