Module: Cumo::CUDA

Defined in:
lib/cumo/cuda.rb,
lib/cumo/cuda/cudnn.rb,
lib/cumo/cuda/event.rb,
lib/cumo/cuda/device.rb,
lib/cumo/cuda/module.rb,
lib/cumo/cuda/stream.rb,
lib/cumo/cuda/compiler.rb,
lib/cumo/cuda/function.rb,
lib/cumo/cuda/link_state.rb,
lib/cumo/cuda/user_kernel.rb,
lib/cumo/cuda/compile_error.rb,
lib/cumo/cuda/nvrtc_program.rb,
lib/cumo/cuda/pinned_memory.rb,
lib/cumo/cuda/reduction_kernel.rb,
lib/cumo/cuda/elementwise_kernel.rb,
ext/cumo/cuda/cudnn.c,
ext/cumo/cuda/nvrtc.c,
ext/cumo/cuda/cublas.c,
ext/cumo/cuda/driver.c,
ext/cumo/cuda/runtime.c,
ext/cumo/cuda/memory_pool.cpp

Defined Under Namespace

Modules: CUDNN, Cublas, Driver, MemoryPool, NVRTC, Runtime, UserKernel Classes: CUDNNError, CompileError, Compiler, CublasError, Device, DriverError, ElementwiseKernel, Event, Function, LinkState, Module, NVRTCError, NVRTCProgram, OutOfMemoryError, PinnedMemory, ReductionKernel, RuntimeError, Stream

Constant Summary collapse

Cudnn =

alias

mCUDNN

Class Method Summary collapse

Class Method Details

.get_elapsed_time(start, stop) ⇒ Object

The milliseconds between two recorded events.



65
66
67
# File 'lib/cumo/cuda/event.rb', line 65

def self.get_elapsed_time(start, stop)
  Runtime.cudaEventElapsedTime(start.handle, stop.handle)
end