GpuCompilationContext Class Reference

#include <NvidiaKernel.h>

Inheritance diagram for GpuCompilationContext:

Collaboration diagram for GpuCompilationContext:

Public Member Functions
	GpuCompilationContext ()
void	addDeviceCode (std::unique_ptr< GpuDeviceCompilationContext > &&device_context)
std::pair< void *, void * >	getNativeCode (const size_t device_id) const
std::vector< void * >	getNativeFunctionPointers () const
std::string const &	name (size_t const device_id) const
size_t	getMemSize () const
Public Member Functions inherited from CompilationContext
virtual	~CompilationContext ()
size_t	size () const

Private Attributes
std::vector< std::unique_ptr < GpuDeviceCompilationContext > >	contexts_per_device_

Detailed Description

Definition at line 87 of file NvidiaKernel.h.

Constructor & Destructor Documentation

GpuCompilationContext::GpuCompilationContext ( )

inline

Definition at line 89 of file NvidiaKernel.h.

{}

Member Function Documentation

void GpuCompilationContext::addDeviceCode ( std::unique_ptr< GpuDeviceCompilationContext > &&  device_context )

inline

Definition at line 91 of file NvidiaKernel.h.

References contexts_per_device_.

                                                                                  {
    contexts_per_device_.push_back(std::move(device_context));
  }

size_t GpuCompilationContext::getMemSize ( ) const

inlinevirtual

Implements CompilationContext.

Definition at line 116 of file NvidiaKernel.h.

References contexts_per_device_.

                            {
    return contexts_per_device_.begin()->get()->getModuleSize();
  }

std::pair<void*, void*> GpuCompilationContext::getNativeCode ( const size_t  device_id ) const

inline

Definition at line 95 of file NvidiaKernel.h.

References CHECK_LT, and contexts_per_device_.

                                                                    {
    CHECK_LT(device_id, contexts_per_device_.size());
    auto device_context = contexts_per_device_[device_id].get();
    return std::make_pair<void*, void*>(device_context->kernel(),
                                        device_context->module());
  }

std::vector<void*> GpuCompilationContext::getNativeFunctionPointers ( ) const

inline

Definition at line 102 of file NvidiaKernel.h.

References CHECK, and contexts_per_device_.

                                                     {
    std::vector<void*> fn_ptrs;
    for (auto& device_context : contexts_per_device_) {
      CHECK(device_context);
      fn_ptrs.push_back(device_context->kernel());
    }
    return fn_ptrs;
  }

std::string const& GpuCompilationContext::name ( size_t const  device_id ) const

inline

Definition at line 111 of file NvidiaKernel.h.

References CHECK_LT, and contexts_per_device_.

                                                      {
    CHECK_LT(device_id, contexts_per_device_.size());
    return contexts_per_device_[device_id]->name();
  }

Member Data Documentation

std::vector<std::unique_ptr<GpuDeviceCompilationContext> > GpuCompilationContext::contexts_per_device_

private

Definition at line 121 of file NvidiaKernel.h.

Referenced by addDeviceCode(), getMemSize(), getNativeCode(), getNativeFunctionPointers(), and name().

---

The documentation for this class was generated from the following file:

• /home/jenkins-slave/workspace/core-os-doxygen/QueryEngine/NvidiaKernel.h