Contents

module TensorStream
  class OpenCLUtil
    ##
    # initializes a OpenCL helper class based on a session
    def initialize(session)
      @session = session
    end

    ##
    # Retrieves OpenCL memory usage
    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
  end
  module CLEventHelpers
    def build_event_wait_list(inputs)
      if inputs.is_a?(Array)
        inputs.flatten.compact.map(&:op).compact.uniq
      else
        inputs.op ? [inputs.op] : []
      end
    end
  end
end

Version data entries

2 entries across 2 versions & 1 rubygems

Version	Path
tensor_stream-opencl-0.3.2	lib/tensor_stream/opencl/utils.rb
tensor_stream-opencl-0.3.1	lib/tensor_stream/opencl/utils.rb