nabu.cuda package

Submodules

• nabu.cuda.convolution module
  • Convolution
    • Convolution.convolve()
• nabu.cuda.kernel module
  • CudaKernel
    • CudaKernel.compile_kernel_source()
    • CudaKernel.prepare()
    • CudaKernel.follow_device_arr()
    • CudaKernel.get_last_kernel_time()
    • CudaKernel.call()
• nabu.cuda.medfilt module
  • MedianFilter
    • MedianFilter.medfilt2()
• nabu.cuda.padding module
  • CudaPadding
    • CudaPadding.pad()
• nabu.cuda.processing module
  • CudaProcessing
    • CudaProcessing.push_context()
    • CudaProcessing.pop_context()
• nabu.cuda.utils module
  • get_cuda_context()
  • count_cuda_devices()
  • get_gpu_memory()
  • is_gpu_usable()
  • detect_cuda_gpus()
  • collect_cuda_gpus()
  • cuarray_format_to_dtype()
  • cuarray_shape_dtype()
  • get_shape_dtype()
  • copy_array()
  • copy_big_gpuarray()
  • replace_array_memory()

Module contents