utils.gpu_utils

Utils for GPU.

Functions

get_maxThreadBlock	Get the warp size of the current device.
is_cuda_array	Check if var implement the CUDA Array interface.
is_cuda_tensor	Check if var is a CUDA tensor.
is_host_array	Check if var is a host contiguous np array.
nvtx_mark	Decorate to annotate function for profiling.
pin_memory	Create a copy of the array in pinned memory.