utils.gpu_utils

utils.gpu_utils#

Utils for GPU.

Functions

get_maxThreadBlock

Get the warp size of the current device.

nvtx_mark

Decorate to annotate function for profiling.