class TensorStream::OpenCLUtil

Public Class Methods

new(session) click to toggle source

initializes a OpenCL helper class based on a session

# File lib/tensor_stream/opencl/utils.rb, line 5
def initialize(session)
  @session = session
end

Public Instance Methods

get_memory_usage() click to toggle source

Retrieves OpenCL memory usage

# File lib/tensor_stream/opencl/utils.rb, line 11
def get_memory_usage
  cl_buffer_uniq_set = Set.new
  @session.last_session_context[:_cache][:_cl_buffers].inject(0) do |sum, elem|
    cl_buffer_uniq_set.add?(elem[1].cl_buffer.object_id) ? sum + elem[1].cl_buffer.size : sum
  end
end