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